RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-08 07:53:29 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

更新依赖

Browse Source
This commit is contained in:
zR 2024-01-20 22:18:22 +08:00
parent e5acf3d2e6
commit dc07cba1d4
5 changed files with 5 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -10,7 +10,7 @@ sentence_transformers==2.2.2
langchain==0.0.354 langchain==0.0.354
langchain-experimental==0.0.47 langchain-experimental==0.0.47
pydantic==1.10.13 pydantic==1.10.13
fschat==0.2.34 fschat==0.2.35
openai~=1.7.1 openai~=1.7.1
fastapi~=0.108.0 fastapi~=0.108.0
sse_starlette==1.8.2 sse_starlette==1.8.2

2
requirements_api.txt
View File

@ -8,7 +8,7 @@ sentence_transformers==2.2.2
langchain==0.0.354 langchain==0.0.354
langchain-experimental==0.0.47 langchain-experimental==0.0.47
pydantic==1.10.13 pydantic==1.10.13
fschat==0.2.34 fschat==0.2.35
openai~=1.7.1 openai~=1.7.1
fastapi~=0.108.0 fastapi~=0.108.0
sse_starlette==1.8.2 sse_starlette==1.8.2

2
requirements_lite.txt
View File

@ -3,7 +3,7 @@
langchain==0.0.354 langchain==0.0.354
langchain-experimental==0.0.47 langchain-experimental==0.0.47
pydantic==1.10.13 pydantic==1.10.13
fschat==0.2.34 fschat==0.2.35
openai~=1.7.1 openai~=1.7.1
fastapi~=0.108.0 fastapi~=0.108.0
sse_starlette==1.8.2 sse_starlette==1.8.2

10
server/knowledge_base/kb_service/zilliz_kb_service.py
View File

@ -16,13 +16,10 @@ class ZillizKBService(KBService):
from pymilvus import Collection from pymilvus import Collection
return Collection(zilliz_name) return Collection(zilliz_name)
# def save_vector_store(self):
# if self.zilliz.col:
# self.zilliz.col.flush()
def get_doc_by_ids(self, ids: List[str]) -> List[Document]: def get_doc_by_ids(self, ids: List[str]) -> List[Document]:
result = [] result = []
if self.zilliz.col: if self.zilliz.col:
ids = [int(id) for id in ids] # for milvus if needed
data_list = self.zilliz.col.query(expr=f'pk in {ids}', output_fields=["*"]) data_list = self.zilliz.col.query(expr=f'pk in {ids}', output_fields=["*"])
for data in data_list: for data in data_list:
text = data.pop("text") text = data.pop("text")
@ -50,8 +47,7 @@ class ZillizKBService(KBService):
def _load_zilliz(self): def _load_zilliz(self):
zilliz_args = kbs_config.get("zilliz") zilliz_args = kbs_config.get("zilliz")
self.zilliz = Zilliz(embedding_function=EmbeddingsFunAdapter(self.embed_model), self.zilliz = Zilliz(embedding_function=EmbeddingsFunAdapter(self.embed_model),
collection_name=self.kb_name, connection_args=zilliz_args) collection_name=self.kb_name, connection_args=zilliz_args)
def do_init(self): def do_init(self):
self._load_zilliz() self._load_zilliz()
@ -95,9 +91,7 @@ class ZillizKBService(KBService):
if __name__ == '__main__': if __name__ == '__main__':
from server.db.base import Base, engine from server.db.base import Base, engine
Base.metadata.create_all(bind=engine) Base.metadata.create_all(bind=engine)
zillizService = ZillizKBService("test") zillizService = ZillizKBService("test")

4
server/utils.py
View File

@ -503,16 +503,12 @@ def set_httpx_config(
no_proxy.append(host) no_proxy.append(host)
os.environ["NO_PROXY"] = ",".join(no_proxy) os.environ["NO_PROXY"] = ",".join(no_proxy)
# TODO: 简单的清除系统代理不是个好的选择，影响太多。似乎修改代理服务器的bypass列表更好。
# patch requests to use custom proxies instead of system settings
def _get_proxies(): def _get_proxies():
return proxies return proxies
import urllib.request import urllib.request
urllib.request.getproxies = _get_proxies urllib.request.getproxies = _get_proxies
# 自动检查torch可用的设备。分布式部署时，不运行LLM的机器上可以不装torch
def detect_device() -> Literal["cuda", "mps", "cpu"]: def detect_device() -> Literal["cuda", "mps", "cpu"]:
try: try: