Update src/rag_system.py

src/rag_system.py

@@ -70,7 +70,7 @@ C:{context}
 A:"""
             
             message = HumanMessage(content=prompt)
-            response = self.llm([message])
+            response = self.llm.invoke([message])
             answer = response.content
             
             # Cache response
@@ -96,7 +96,7 @@ A:"""
 {text}"""
             
             message = HumanMessage(content=prompt)
-            response = self.llm([message])
+            response = self.llm.invoke([message])
             summary = response.content
             
             # Cache
@@ -120,7 +120,8 @@ A:"""
             qa_prompt += f"Context:{context}\nAnswers:"
             
             message = HumanMessage(content=qa_prompt)
-            response = self.llm([message])
+            response = self.llm.invoke([message])
+
             
             # Parse responses
             answers = response.content.split('\n')