15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-25 17:22:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor: 修改依赖包的方法 

(cherry picked from commit 3cee9bc495)
This commit is contained in:
wxg0103 2024-11-01 17:33:28 +08:00 committed by shaohuzhang1
parent f5af097e6d
commit 110d8fd398
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 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 @@ class XFChatSparkLLM(MaxKBBaseModel, ChatSparkLLM):
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,

2
pyproject.toml
View File

@ -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"