15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor: 修改依赖包的方法
Some checks failed
sync2gitee / repo-sync (push) Has been cancelled
Details
Typos Check / Spell Check with Typos (push) Has been cancelled
Details

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