zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 修改依赖包的方法

Browse Source
This commit is contained in:
wxg0103 2024-11-01 17:33:28 +08:00
parent 251e4645b7
commit 3cee9bc495
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/setting/models_provider/impl/xf_model_provider/model/llm.py
View File

@ -9,7 +9,7 @@
from typing import List, Optional, Any, Iterator, Dict from typing import List, Optional, Any, Iterator, Dict
from langchain_community.chat_models.sparkllm import \ 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.callbacks import CallbackManagerForLLMRun
from langchain_core.messages import BaseMessage, AIMessageChunk from langchain_core.messages import BaseMessage, AIMessageChunk
from langchain_core.outputs import ChatGenerationChunk from langchain_core.outputs import ChatGenerationChunk
@ -56,7 +56,7 @@ class XFChatSparkLLM(MaxKBBaseModel, ChatSparkLLM):
default_chunk_class = AIMessageChunk default_chunk_class = AIMessageChunk
self.client.arun( 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.spark_user_id,
self.model_kwargs, self.model_kwargs,
True, True,

2
pyproject.toml
View File

@ -12,7 +12,7 @@ djangorestframework = "^3.15.2"
drf-yasg = "1.21.7" drf-yasg = "1.21.7"
django-filter = "23.2" django-filter = "23.2"
langchain = "0.2.16" langchain = "0.2.16"
langchain_community = "0.2.4" langchain_community = "0.2.17"
langchain-huggingface = "^0.0.3" langchain-huggingface = "^0.0.3"
psycopg2-binary = "2.9.7" psycopg2-binary = "2.9.7"
jieba = "^0.42.1" jieba = "^0.42.1"