refactor: 修改依赖包的方法

1Panel-dev · Nov 1, 2024 · 3cee9bc · 3cee9bc
1 parent 251e464
commit 3cee9bc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/apps/setting/models_provider/impl/xf_model_provider/model/llm.py b/apps/setting/models_provider/impl/xf_model_provider/model/llm.py
@@ -9,7 +9,7 @@
 from typing import List, Optional, Any, Iterator, Dict
 
 from langchain_community.chat_models.sparkllm import \
-    ChatSparkLLM, _convert_message_to_dict, _convert_delta_to_message_chunk
+    ChatSparkLLM, convert_message_to_dict, _convert_delta_to_message_chunk
 from langchain_core.callbacks import CallbackManagerForLLMRun
 from langchain_core.messages import BaseMessage, AIMessageChunk
 from langchain_core.outputs import ChatGenerationChunk
@@ -56,7 +56,7 @@ def _stream(
         default_chunk_class = AIMessageChunk
 
         self.client.arun(
-            [_convert_message_to_dict(m) for m in messages],
+            [convert_message_to_dict(m) for m in messages],
             self.spark_user_id,
             self.model_kwargs,
             True,

diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ djangorestframework = "^3.15.2"
 drf-yasg = "1.21.7"
 django-filter = "23.2"
 langchain = "0.2.16"
-langchain_community = "0.2.4"
+langchain_community = "0.2.17"
 langchain-huggingface = "^0.0.3"
 psycopg2-binary = "2.9.7"
 jieba = "^0.42.1"