Update ragchain.py

ragchain.py

@@ -41,11 +41,7 @@ class RAGChain:
                     {query}
             """
         )
-
-        rewritten_query_response = self.llm.invoke(rewrite_prompt)
-        rewritten_query = rewritten_query_response.content.strip()
-        return rewritten_query
-
+        return self.llm.invoke(rewrite_prompt)
 
     
     def predict_library_usage(self, query):