diff --git a/chatchat-server/chatchat/server/agent/agent_factory/agents_registry.py b/chatchat-server/chatchat/server/agent/agent_factory/agents_registry.py index 7c9a5c0a..7de96aaa 100644 --- a/chatchat-server/chatchat/server/agent/agent_factory/agents_registry.py +++ b/chatchat-server/chatchat/server/agent/agent_factory/agents_registry.py @@ -25,7 +25,10 @@ def agents_registry( 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_executor = AgentExecutor(agent=agent, tools=tools, verbose=verbose, callbacks=callbacks) + + return agent_executor elif "qwen" in llm.model_name.lower(): return create_structured_qwen_chat_agent(llm=llm, tools=tools, callbacks=callbacks) else: @@ -35,6 +38,6 @@ def agents_registry( prompt = hub.pull("hwchase17/structured-chat-agent") # default prompt agent = create_structured_chat_agent(llm=llm, tools=tools, prompt=prompt) - agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=verbose, callbacks=callbacks) + agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=verbose, callbacks=callbacks) - return agent_executor + return agent_executor diff --git a/chatchat-server/chatchat/server/utils.py b/chatchat-server/chatchat/server/utils.py index 8efcd0f8..ac441157 100644 --- a/chatchat-server/chatchat/server/utils.py +++ b/chatchat-server/chatchat/server/utils.py @@ -246,7 +246,10 @@ def get_OpenAIClient( construct an openai Client for specified platform or model ''' if platform_name is None: - platform_name = get_model_info(model_name=model_name, platform_name=platform_name)["platform_name"] + platform_info = get_model_info(model_name=model_name, platform_name=platform_name) + if platform_info is None: + raise RuntimeError(f"cannot find configured platform for model: {model_name}") + platform_name = platform_info.get("platform_name") platform_info = get_config_platforms().get(platform_name) assert platform_info, f"cannot find configured platform: {platform_name}" params = {