feat: fix dataset retrieve agent llm not support error (#656)

2023-07-27 15:45:52 +08:00 · 2023-07-27 15:45:52 +08:00 · ba3dc8cae0
commit ba3dc8cae0
parent ae7c0380dc
3 changed files with 12 additions and 2 deletions
--- a/api/core/agent/agent/multi_dataset_router_agent.py
+++ b/api/core/agent/agent/multi_dataset_router_agent.py
@ -73,7 +73,6 @@ class MultiDatasetRouterAgent(OpenAIFunctionsAgent):
            ),
            **kwargs: Any,
    ) -> BaseSingleActionAgent:
-        llm.model_name = 'gpt-3.5-turbo'
        return super().from_llm_and_tools(
            llm=llm,
            tools=tools,