|
@@ -3,8 +3,7 @@ from typing_extensions import TypedDict
|
|
|
from langgraph.graph import StateGraph, START, END
|
|
|
from langgraph.graph.message import add_messages
|
|
|
import os
|
|
|
-from langchain.chat_models import init_chat_model
|
|
|
-from IPython.display import Image, display
|
|
|
+from langchain_openai import ChatOpenAI
|
|
|
from .tools import evaluation_extraction_tool
|
|
|
|
|
|
from langgraph.prebuilt import ToolNode, tools_condition
|
|
@@ -77,7 +76,8 @@ def execute_agent_with_api(user_input: str):
|
|
|
# 如果graph或llm_with_tools未初始化,先初始化
|
|
|
if graph is None or llm_with_tools is None:
|
|
|
try:
|
|
|
- llm = init_chat_model("openai:gpt-4.1")
|
|
|
+ # 使用新版本的 ChatOpenAI
|
|
|
+ llm = ChatOpenAI(model="gpt-4")
|
|
|
tools = [evaluation_extraction_tool]
|
|
|
llm_with_tools = llm.bind_tools(tools=tools)
|
|
|
|