Update run.py

run.py

@@ -96,20 +96,41 @@ collection = client.get_collection(name="chromaTS", embedding_function=sentence_
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
 
+#def format_prompt(message, history):
+  #print("HISTORY")
+  #print(history)
+  #prompt = "" #"<s>"
+  #c=1
+  #for user_prompt, bot_response in history:
+  #  if c<2:
+  #      prompt += f"[INST] {user_prompt} [/INST]"
+  #      prompt += f" {bot_response}</s> "
+  #      c=c+1
+  #prompt += f"[INST] {message} [/INST]"
+  #print("Final P")
+  #print(prompt)
+  #return prompt
+
 def format_prompt(message, history):
-  print("HISTORY")
-  print(history)
-  prompt = "" #"<s>"
-  c=1
-  for user_prompt, bot_response in history:
-    if c<2:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response}</s> "
-        c=c+1
-  prompt += f"[INST] {message} [/INST]"
-  print("Final P")
-  print(prompt)
-  return prompt
+    print("HISTORY")
+    print(history)
+    
+    # Initialize prompt with an empty string
+    prompt = ""
+    
+    # Check if history is not empty
+    if history:
+        # Extract the most recent user prompt and bot response
+        user_prompt, bot_response = history[-1]
+        prompt += f"[INST] {user_prompt} [/INST] {bot_response}</s> "
+    
+    # Add the new message to the prompt
+    prompt += f"[INST] {message} [/INST]"
+    
+    print("Final P")
+    print(prompt)
+    
+    return prompt
 
 def response(
     prompt, history,temperature=0.9, max_new_tokens=500, top_p=0.95, repetition_penalty=1.0,