Update classification_chain.py

classification_chain.py

@@ -2,8 +2,6 @@
 import os
 from langchain.chains import LLMChain
 from langchain_groq import ChatGroq
-
-# We'll import the classification_prompt from prompts.py
 from prompts import classification_prompt
 
 def get_classification_chain() -> LLMChain:
@@ -22,3 +20,13 @@ def get_classification_chain() -> LLMChain:
         prompt=classification_prompt
     )
     return classification_chain
+
+def classify_with_history(query: str, chat_history: list) -> str:
+    """
+    Classifies a user query based on the context of previous conversation (chat_history).
+    """
+    # Add the history into the query context if needed (depending on the type of model)
+    context = "\n".join([f"User: {msg['content']}" for msg in chat_history]) + "\nUser: " + query
+    # Update the prompt with both the context and the query
+    classification_result = get_classification_chain().run({"query": context})
+    return classification_result