Upload model_manage.py (#3)

- Upload model_manage.py (3634f37f18c771844860fd89535eb6544fef8db7)


Co-authored-by: Nguyen Nguyen Anh <AnhLedger@users.noreply.huggingface.co>

chat/model_manage.py

@@ -167,3 +167,35 @@ def full_chain_single_question(input_prompt, db_instance):
     except Exception as e:
         # print(e)
         return temp_answer, "Error occured: " + str(e)
+    
+
+def format_chat_history_from_web(chat_history: list):
+    temp_chat = []
+    for message in chat_history:
+        temp_chat.append(
+            {
+                "role": message["role"],
+                "parts": [message["content"]]
+            }
+        )
+    return temp_chat
+
+def full_chain_history_question(chat_history: list, db_instance):
+    try:
+        temp_chat = format_chat_history_from_web(chat_history)
+        first_prompt = extract_keyword_prompt(temp_chat[-1]["parts"][0])
+        temp_answer = model.generate_content(first_prompt).text
+
+        args = json.loads(utils.trimming(temp_answer))
+        contexts, results = response(args, db_instance)
+        if not results:
+            # print(contexts)
+            return "Random question, direct return", contexts
+        else:
+            QA_Prompt = make_answer_prompt(temp_chat[-1]["parts"][0], contexts)
+            temp_chat[-1]["parts"] = QA_Prompt
+            answer = model.generate_content(temp_chat).text
+            return temp_answer, answer
+    except Exception as e:
+        # print(e)
+        return temp_answer, "Error occured: " + str(e)