Update rag.py

rag.py

@@ -34,7 +34,7 @@ class TicketResolver:
 
         return issue_resolution_pairs, resolutions
 
-    def generate_resolution(self, retrieved_resolutions):
+    def generate_resolution(self,query, retrieved_resolutions):
         combined_resolutions = " ".join([str(resolution) for resolution in retrieved_resolutions if pd.notna(resolution)])
         prompt = f"Based on these resolutions: {combined_resolutions}. Check if the query {query} has any related queries in combined_resolutions ,if found relatable queries Generate a generic resolution combining all matched ones : . If no matches or relatable issues are found in combing resolutions return a generic solution combining all :"
         completion = self.genai_model.generate_content(prompt)