From 1c0be9caf21fb139a7a425b334c1e7f1f9e1e824 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E5=9F=B9=E6=96=87?= Date: Wed, 13 Mar 2024 13:04:30 +0800 Subject: [PATCH 1/2] [fix issue#29] api error --- examples/webui/dialogue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/webui/dialogue.py b/examples/webui/dialogue.py index 36a4e18..86e20c5 100644 --- a/examples/webui/dialogue.py +++ b/examples/webui/dialogue.py @@ -474,7 +474,7 @@ def dialogue_page(api: ApiRequest): prompt, search_engine, se_top_k, history, embed_model=EMBEDDING_MODEL, embed_model_path=embedding_model_dict[EMBEDDING_MODEL], model_device=EMBEDDING_DEVICE, embed_engine=EMBEDDING_ENGINE, llm_model=LLM_MODEL, - pi_key=llm_model_dict[LLM_MODEL]["api_key"], + api_key=llm_model_dict[LLM_MODEL]["api_key"], api_base_url=llm_model_dict[LLM_MODEL]["api_base_url"],) ): if error_msg := check_error_msg(d): # check whether error occured From 74cab0fd4cd3d1f950090e5180d5aa58705e3d3e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E5=9F=B9=E6=96=87?= Date: Wed, 13 Mar 2024 14:18:18 +0800 Subject: [PATCH 2/2] [fix issue#29]EMBEDDING_MODEL error --- examples/webui/document.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/webui/document.py b/examples/webui/document.py index 4f41c97..104cb36 100644 --- a/examples/webui/document.py +++ b/examples/webui/document.py @@ -357,7 +357,7 @@ def knowledge_page( empty.progress(0.0, "") for d in api.recreate_vector_store( kb, vs_type=default_vs_type, embed_model=embedding_model, embedding_device=EMBEDDING_DEVICE, - embed_model_path=embedding_model_dict["embedding_model"], embed_engine=EMBEDDING_ENGINE, + embed_model_path=embedding_model_dict[EMBEDDING_MODEL], embed_engine=EMBEDDING_ENGINE, api_key=llm_model_dict[LLM_MODEL]["api_key"], api_base_url=llm_model_dict[LLM_MODEL]["api_base_url"], ):