weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

参照chat，给knowledge_base_chat/search_engine_chat添加chatgpt代理设置

This commit is contained in:
liunux4odoo 2023-08-24 15:10:51 +08:00
parent ea4b5ebd16
commit 63ca393731
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/chat/knowledge_base_chat.py
View File

@ -52,7 +52,8 @@ def knowledge_base_chat(query: str = Body(..., description="用户输入", examp
callbacks=[callback], callbacks=[callback],
openai_api_key=llm_model_dict[LLM_MODEL]["api_key"], openai_api_key=llm_model_dict[LLM_MODEL]["api_key"],
openai_api_base=llm_model_dict[LLM_MODEL]["api_base_url"], 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) docs = search_docs(query, knowledge_base_name, top_k, score_threshold)
context = "\n".join([doc.page_content for doc in docs]) context = "\n".join([doc.page_content for doc in docs])

3
server/chat/search_engine_chat.py
View File

@ -87,7 +87,8 @@ def search_engine_chat(query: str = Body(..., description="用户输入", exampl
callbacks=[callback], callbacks=[callback],
openai_api_key=llm_model_dict[LLM_MODEL]["api_key"], openai_api_key=llm_model_dict[LLM_MODEL]["api_key"],
openai_api_base=llm_model_dict[LLM_MODEL]["api_base_url"], 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) docs = lookup_search_engine(query, search_engine_name, top_k)