更新版本

glide-the · Feb 10, 2025 · 3851633 · 3851633
1 parent 04513bb
commit 3851633
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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 <[email protected]>"]
 readme = "README.md"

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