Update app.py

app.py

@@ -78,7 +78,10 @@ class ResponseExpertAgent(Agent):
         try:
             logger.info(f"Sending query for generation: {rephrased_query}")
             response = await hf_client.text_generation(rephrased_query, max_new_tokens=500, temperature=0.7)
-            return response['generated_text']
+            if 'generated_text' in response:
+                return response['generated_text']
+            else:
+                return "Failed to retrieve generated text from response."
         except Exception as e:
             logger.error(f"Failed to generate text due to: {str(e)}")
             return "Error in generating response. Please try again."