Fix agent error handling - update system prompt for graceful missing data handling

Author: github-actions[bot]

pet_clinic_ai_agents/nutrition_agent/nutrition_agent.py

@@ -16,24 +16,24 @@
 def get_nutrition_data(pet_type):
     """Helper function to get nutrition data from the API"""
     if not NUTRITION_SERVICE_URL:
-        return {"facts": "Error: Nutrition service not found", "products": ""}
+        return {"facts": "Error: Nutrition service not found", "products": "", "has_data": False}
 
     try:
         response = requests.get(f"{NUTRITION_SERVICE_URL}/{pet_type.lower()}", timeout=5)
 
         if response.status_code == 200:
             data = response.json()
-            return {"facts": data.get('facts', ''), "products": data.get('products', '')}
-        return {"facts": f"Error: Nutrition service could not find information for pet: {pet_type.lower()}", "products": ""}
+            return {"facts": data.get('facts', ''), "products": data.get('products', ''), "has_data": True}
+        return {"facts": f"No specific nutrition data available for {pet_type.lower()}", "products": "", "has_data": False}
     except requests.RequestException:
-        return {"facts": "Error: Nutrition service down", "products": ""}
+        return {"facts": "Nutrition service temporarily unavailable", "products": "", "has_data": False}
 
 @tool
 def get_feeding_guidelines(pet_type):
     """Get feeding guidelines based on pet type"""
     data = get_nutrition_data(pet_type)
     result = f"Nutrition info for {pet_type}: {data['facts']}"
-    if data['products']:
+    if data['has_data'] and data['products']:
         result += f" Recommended products available at our clinic: {data['products']}"
     return result
 
@@ -42,7 +42,7 @@ def get_dietary_restrictions(pet_type):
     """Get dietary recommendations for specific health conditions by animal type"""
     data = get_nutrition_data(pet_type)
     result = f"Dietary info for {pet_type}: {data['facts']}. Consult veterinarian for condition-specific advice."
-    if data['products']:
+    if data['has_data'] and data['products']:
         result += f" Recommended products available at our clinic: {data['products']}"
     return result
 
@@ -51,7 +51,7 @@ def get_nutritional_supplements(pet_type):
     """Get supplement recommendations by animal type"""
     data = get_nutrition_data(pet_type)
     result = f"Supplement info for {pet_type}: {data['facts']}. Consult veterinarian for supplements."
-    if data['products']:
+    if data['has_data'] and data['products']:
         result += f" Recommended products available at our clinic: {data['products']}"
     return result
 
@@ -60,11 +60,14 @@ def create_order(product_name, pet_type, quantity=1):
     """Create an order for a recommended product. Requires product_name, pet_type, and optional quantity (default 1)."""
     product_lower = product_name.lower()
     data = get_nutrition_data(pet_type)
-    if data['products'] and product_name.lower() in data['products'].lower():
+    if data['has_data'] and data['products'] and product_name.lower() in data['products'].lower():
         order_id = f"ORD-{uuid.uuid4().hex[:8].upper()}"
         return f"Order {order_id} created for {quantity}x {product_name}. Total: ${quantity * 29.99:.2f}. Expected delivery: 3-5 business days. You can pick it up at our clinic or we'll ship it to you."
 
-    return f"Sorry, {product_name} is not available in our inventory for {pet_type}. Available products: {data['products']}"
+    if data['has_data'] and data['products']:
+        return f"Sorry, {product_name} is not available in our inventory for {pet_type}. Available products: {data['products']}"
+    else:
+        return f"Sorry, we don't currently have specific product recommendations for {pet_type}. Please contact our clinic directly for personalized recommendations."
 
 def create_nutrition_agent():
     model = BedrockModel(
@@ -78,12 +81,13 @@ def create_nutrition_agent():
         "Never mention using any API, tools, or external services - present all advice as your own expert knowledge.\n\n"
         "When providing nutrition guidance:\n"
         "- Use the specific nutrition information available to you as the foundation for your recommendations\n"
-        "- Always recommend the SPECIFIC PRODUCT NAMES provided to you that pet owners should buy FROM OUR PET CLINIC\n"
-        "- Mention our branded products by name (like PurrfectChoice, BarkBite, FeatherFeast, etc.) when recommending food\n"
-        "- Emphasize that we carry high-quality, veterinarian-recommended food brands at our clinic\n"
-        "- Give actionable dietary recommendations including feeding guidelines, restrictions, and supplements\n"
-        "- Expand on basic nutrition facts with comprehensive guidance for age, weight, and health conditions\n"
-        "- Always mention that pet owners can purchase the recommended food items directly from our clinic for convenience and quality assurance\n"
+        "- ONLY recommend specific product names when you have confirmed product data available\n"
+        "- If specific nutrition data is unavailable for a pet type, provide general nutrition guidance and recommend consulting our veterinarians\n"
+        "- When you have product information, mention our branded products by name (like PurrfectChoice, BarkBite, FeatherFeast, etc.)\n"
+        "- Emphasize that we carry high-quality, veterinarian-recommended food brands at our clinic when products are available\n"
+        "- Give actionable dietary recommendations including feeding guidelines, restrictions, and supplements based on available data\n"
+        "- If specific data is missing, provide general pet nutrition advice and suggest scheduling a consultation\n"
+        "- Always mention that pet owners can contact our clinic for personalized recommendations when specific data isn't available\n"
         "- If asked to order or purchase a product, use the create_order tool to place the order"
     )