From 63ca393731e98cd4a8e665c13e1427b3c886fa32 Mon Sep 17 00:00:00 2001 From: liunux4odoo Date: Thu, 24 Aug 2023 15:10:51 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8F=82=E7=85=A7chat=EF=BC=8C=E7=BB=99knowled?= =?UTF-8?q?ge=5Fbase=5Fchat/search=5Fengine=5Fchat=E6=B7=BB=E5=8A=A0chatgp?= =?UTF-8?q?t=E4=BB=A3=E7=90=86=E8=AE=BE=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- server/chat/knowledge_base_chat.py | 3 ++- server/chat/search_engine_chat.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/server/chat/knowledge_base_chat.py b/server/chat/knowledge_base_chat.py index 4914beb..2774569 100644 --- a/server/chat/knowledge_base_chat.py +++ b/server/chat/knowledge_base_chat.py @@ -52,7 +52,8 @@ def knowledge_base_chat(query: str = Body(..., description="用户输入", examp callbacks=[callback], openai_api_key=llm_model_dict[LLM_MODEL]["api_key"], openai_api_base=llm_model_dict[LLM_MODEL]["api_base_url"], - model_name=LLM_MODEL + model_name=LLM_MODEL, + openai_proxy=llm_model_dict[LLM_MODEL].get("openai_proxy") ) docs = search_docs(query, knowledge_base_name, top_k, score_threshold) context = "\n".join([doc.page_content for doc in docs]) diff --git a/server/chat/search_engine_chat.py b/server/chat/search_engine_chat.py index 638c3be..f99cafc 100644 --- a/server/chat/search_engine_chat.py +++ b/server/chat/search_engine_chat.py @@ -87,7 +87,8 @@ def search_engine_chat(query: str = Body(..., description="用户输入", exampl callbacks=[callback], openai_api_key=llm_model_dict[LLM_MODEL]["api_key"], openai_api_base=llm_model_dict[LLM_MODEL]["api_base_url"], - model_name=LLM_MODEL + model_name=LLM_MODEL, + openai_proxy=llm_model_dict[LLM_MODEL].get("openai_proxy") ) docs = lookup_search_engine(query, search_engine_name, top_k)