enterprise-kb-rag/2.2_LangChain基础知识/30_agents_openai_tools.py

from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parents[1]))

from _chapter2_config import get_chat_model, print_dependency_hint


try:
    from langchain.agents import create_agent
    from langchain_community.agent_toolkits.load_tools import load_tools
    from langchain_core.messages import HumanMessage
    from langgraph.checkpoint.memory import MemorySaver
except ModuleNotFoundError as exc:
    print_dependency_hint(exc)
    raise SystemExit(1)


def main() -> None:
    try:
        model = get_chat_model(temperature=0)
    except ModuleNotFoundError as exc:
        print_dependency_hint(exc)
        raise SystemExit(1)

    memory = MemorySaver()
    tools = load_tools(["llm-math"], llm=model)
    agent_executor = create_agent(model, tools, checkpointer=memory)

    config = {"configurable": {"thread_id": "abc123"}}
    for chunk in agent_executor.stream(
        {"messages": [HumanMessage(content="计算一下300的25%是多少？")]},
        config,
    ):
        print(chunk)
        print("----")


if __name__ == "__main__":
    main()