Update bin_public/app/llama_func.py

bin_public/app/llama_func.py

@@ -124,7 +124,6 @@ def chat_ai(
         question,
         context,
         chatbot,
-        reply_language,
 ):
     os.environ["OPENAI_API_KEY"] = api_key
 
@@ -139,7 +138,6 @@ def chat_ai(
         SIM_K,
         1.0,
         context,
-        reply_language,
     )
     if response is None:
         status_text = "查询失败，请换个问法试试"
@@ -163,7 +161,6 @@ def ask_ai(
         sim_k=5,
         temprature=0,
         prefix_messages=[],
-        reply_language="中文",
 ):
     os.environ["OPENAI_API_KEY"] = api_key
 
@@ -178,13 +175,13 @@ def ask_ai(
     )
 
     response = None  # Initialize response variable to avoid UnboundLocalError
-    qa_prompt = QuestionAnswerPrompt(prompt_tmpl.replace("{reply_language}", reply_language))
-    rf_prompt = RefinePrompt(refine_tmpl.replace("{reply_language}", reply_language))
+    #qa_prompt = QuestionAnswerPrompt(prompt_tmpl.replace("{reply_language}", reply_language))
+    #rf_prompt = RefinePrompt(refine_tmpl.replace("{reply_language}", reply_language))
     response = index.query(
         question,
         similarity_top_k=sim_k,
-        text_qa_template=qa_prompt,
-        refine_template=rf_prompt,
+        #text_qa_template=qa_prompt,
+        #refine_template=rf_prompt,
         response_mode="compact",
     )