diff --git a/src/dreamsboard/pyproject.toml b/src/dreamsboard/pyproject.toml index 81654fc..15672b0 100644 --- a/src/dreamsboard/pyproject.toml +++ b/src/dreamsboard/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "dreamsboard" -version = "0.3.6" +version = "0.3.7" description = "" authors = ["glide-the <2533736852@qq.com>"] readme = "README.md" diff --git a/test_task/glm/main.py b/test_task/glm/main.py index bd2c897..87d3d0b 100644 --- a/test_task/glm/main.py +++ b/test_task/glm/main.py @@ -37,9 +37,9 @@ logging.config.dictConfig(logging_conf) # type: ignore llm = ChatOpenAI( - openai_api_base=os.environ.get("API_BASE"), - model=os.environ.get("API_MODEL"), - openai_api_key=os.environ.get("API_KEY"), + openai_api_base=os.environ.get("ZHIPUAI_API_BASE"), + model=os.environ.get("ZHIPUAI_API_MODEL"), + openai_api_key=os.environ.get("ZHIPUAI_API_KEY"), verbose=True, temperature=0.1, top_p=0.9, @@ -58,7 +58,7 @@ tools= [ { "type": "web_search", "web_search": {"enable": False ,"search_result": False }}] else: tools = [] -llm_with_tools = llm +llm_with_tools = llm.bind( tools=[_get_assistants_tool(tool) for tool in tools] ) kor_dreams_task_step_llm_with_tools = guiji_llm.bind( tools=[_get_assistants_tool(tool) for tool in tools] )