更新版本

Author: glide-the

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"

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] )