RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-01-19 21:37:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

减少错误的显示

Browse Source
This commit is contained in:
zR 2024-01-19 21:49:53 +08:00 committed by liunux4odoo
parent fa4ccdb4be
commit 2c08fce6b0
6 changed files with 22 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/loom.yaml.example
View File

@ -18,6 +18,8 @@ openai_plugins_load_folder:
plugins:
- openai:
name: "openai"
- zhipuai:
name: "zhipuai"
- imitater:
name: "imitater"

20
openai_plugins/openai/profile_endpoint.py
View File

@ -43,11 +43,11 @@ class OpenAIProfileEndpointAdapter(ProfileEndpointAdapter):
model_description="gpt-4-1106-preview",
providers=["model", "embedding"],
model_extra_info="{}"))
list_worker.append(LLMWorkerInfo(worker_id="text-embedding-ada-002",
model_name="text-embedding-ada-002",
model_description="text-embedding-ada-002",
providers=["embedding"],
model_extra_info="{}"))
# list_worker.append(LLMWorkerInfo(worker_id="text-embedding-ada-002",
# model_name="text-embedding-ada-002",
# model_description="text-embedding-ada-002",
# providers=["embedding"],
# model_extra_info="{}"))
return list_worker
def list_llm_models(self) -> List[LLMWorkerInfo]:
@ -68,11 +68,11 @@ class OpenAIProfileEndpointAdapter(ProfileEndpointAdapter):
model_description="gpt-4-1106-preview",
providers=["model", "embedding"],
model_extra_info="{}"))
list_worker.append(LLMWorkerInfo(worker_id="text-embedding-ada-002",
model_name="text-embedding-ada-002",
model_description="text-embedding-ada-002",
providers=["embedding"],
model_extra_info="{}"))
# list_worker.append(LLMWorkerInfo(worker_id="text-embedding-ada-002",
# model_name="text-embedding-ada-002",
# model_description="text-embedding-ada-002",
# providers=["embedding"],
# model_extra_info="{}"))
return list_worker
def get_model_config(self, model_name) -> LLMWorkerInfo:

1
openai_plugins/zhipuai/controller.py
View File

@ -5,7 +5,6 @@ import sys
import logging
logger = logging.getLogger(__name__)
# 为了能使用插件中的函数，需要将当前目录加入到sys.path中
root_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(root_dir)

2
openai_plugins/zhipuai/install.py
View File

@ -9,8 +9,6 @@ import locale
logger = logging.getLogger(__name__)
root_dir = os.path.dirname(os.path.abspath(__file__))
sys.path.append(root_dir)
# Perform install
processed_install = set()
pip_list = None

0
openai_plugins/zhipuai/zhipuai_plugins.json → openai_plugins/zhipuai/openai_plugins.json
View File

20
openai_plugins/zhipuai/profile_endpoint.py
View File

@ -37,11 +37,11 @@ class ZhipuAIProfileEndpointAdapter(ProfileEndpointAdapter):
model_description="glm-3-turbo",
providers=["model", "embedding"],
model_extra_info="{}"))
list_worker.append(LLMWorkerInfo(worker_id="embedding-2",
model_name="embedding-2",
model_description="embedding-2",
providers=["embedding"],
model_extra_info="{}"))
# list_worker.append(LLMWorkerInfo(worker_id="embedding-2",
# model_name="embedding-2",
# model_description="embedding-2",
# providers=["embedding"],
# model_extra_info="{}"))
return list_worker
def list_llm_models(self) -> List[LLMWorkerInfo]:
@ -57,11 +57,11 @@ class ZhipuAIProfileEndpointAdapter(ProfileEndpointAdapter):
model_description="glm-3-turbo",
providers=["model", "embedding"],
model_extra_info="{}"))
list_worker.append(LLMWorkerInfo(worker_id="embedding-2",
model_name="embedding-2",
model_description="embedding-2",
providers=["embedding"],
model_extra_info="{}"))
# list_worker.append(LLMWorkerInfo(worker_id="embedding-2",
# model_name="embedding-2",
# model_description="embedding-2",
# providers=["embedding"],
# model_extra_info="{}"))
return list_worker
def get_model_config(self, model_name) -> LLMWorkerInfo: