used chain in classification chain.py

classification_chain.py

@@ -2,10 +2,11 @@
 import os
 from langchain.chains import LLMChain
 from langchain_groq import ChatGroq
-
-# We'll import the classification_prompt from prompts.py
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.prompts import ChatPromptTemplate
 from prompts import classification_prompt
 
+
 def get_classification_chain() -> LLMChain:
     """
     Builds the classification chain (LLMChain) using ChatGroq and the classification prompt.
@@ -15,10 +16,8 @@ def get_classification_chain() -> LLMChain:
         model="Gemma2-9b-It",
         groq_api_key=os.environ["GROQ_API_KEY"]  # must be set in environment
     )
+    prompt=
 
     # Build an LLMChain
-    classification_chain = LLMChain(
-        llm=chat_groq_model,
-        prompt=classification_prompt
-    )
+    classification_chain = classification_prompt|chat_groq_model|output_parser
     return classification_chain