from langchain_core.messages import HumanMessage, AIMessageChunk, AIMessage

from agent import agent, checkpointer


async def search_recipes(prompt: str, thread_id: str):
    """调用agent搜索食谱"""
    print(f"[用户]: {prompt}, thread_id: {thread_id}")
    try:
        message = HumanMessage(content=prompt)

        # 流式调用Agent
        for chunk, metadata in agent.stream(
                {"messages": [message]},
                {"configurable": {"thread_id": thread_id}},
                stream_mode="messages"
        ):
            if isinstance(chunk, AIMessageChunk) and chunk.content:
                yield chunk.content

    except Exception as e:
        print(f"\n[错误]: {str(e)}")
        yield "信息检索失败，试试看手动输入食物列表？"


# 清空会话
def clear_messages(thread_id: str):
    """清空会话"""
    print(f"清空历史消息，thread_id: {thread_id}")
    checkpointer.delete_thread(thread_id)


# 查询会话历史
def get_messages(thread_id: str) -> list[dict[str, str]]:
    """获取会话历史"""
    print(f"获取历史消息，thread_id: {thread_id}")

    # 根据 thread_id 查询 checkpoint
    checkpoint = checkpointer.get({"configurable": {"thread_id": thread_id}})

    # 如果不存在，返回空列表
    if not checkpoint:
        return []

    # 安全获取 messages
    channel_values = checkpoint.get("channel_values")
    if not channel_values:
        return []

    messages = channel_values.get("messages", [])
    if not messages:
        return []

    # 转换消息格式
    result = []
    for msg in messages:
        if not msg.content:
            continue

        if isinstance(msg, HumanMessage):
            result.append({"role": "user", "content": msg.content})
        elif isinstance(msg, AIMessage):
            result.append({"role": "assistant", "content": msg.content})

    return result