Update kadi_apy_bot.py

kadi_apy_bot.py

@@ -93,7 +93,7 @@ class KadiAPYBot:
                     {query}
             """
         )
-        return self.llm.invoke(rewrite_prompt)
+        return self.llm.invoke(rewrite_prompt).content
 
     
     def predict_library_usage(self, query):
@@ -113,7 +113,7 @@ class KadiAPYBot:
                 Respond with only the exact corresponding option and do not include any additional comments, explanations, or text."
             """
         )
-        return self.llm.predict(prompt)
+        return self.llm.predict(prompt).content
 
     def retrieve_contexts(self, query, k, filter = None):
         """