diff --git a/server/agent/agent_factory/agents_registry.py b/server/agent/agent_factory/agents_registry.py index be77bd2f..9fae0969 100644 --- a/server/agent/agent_factory/agents_registry.py +++ b/server/agent/agent_factory/agents_registry.py @@ -9,6 +9,7 @@ from langchain_core.prompts import ChatPromptTemplate from langchain_core.tools import BaseTool from server.agent.agent_factory import ( create_structured_qwen_chat_agent) +from server.agent.agent_factory.glm3_agent import create_structured_glm3_chat_agent def agents_registry( @@ -23,8 +24,8 @@ def agents_registry( # Write any optimized method here. if "glm3" in llm.model_name.lower(): # An optimized method of langchain Agent that uses the glm3 series model - # agent = create_structured_glm3_chat_agent(llm=llm, tools=tools) - pass + agent = create_structured_glm3_chat_agent(llm=llm, tools=tools) + # pass elif "qwen" in llm.model_name.lower(): agent = create_structured_qwen_chat_agent(llm=llm, tools=tools) else: diff --git a/server/pydantic_types.py b/server/pydantic_types.py index 78e754e6..af532414 100644 --- a/server/pydantic_types.py +++ b/server/pydantic_types.py @@ -1,6 +1,7 @@ from langchain_core.pydantic_v1 import * from pydantic.fields import FieldInfo from pydantic.schema import model_schema +from pydantic.typing import typing # from pydantic.v1 import * # from pydantic.v1.fields import FieldInfo