Update app.py

app.py

@@ -40,9 +40,6 @@ class LocalBartModel:
         self.model.eval()
 
     def __call__(self, prompt: str) -> str:
-        if not isinstance(prompt, str):
-            raise ValueError(f"LocalBartModel expects a string prompt, got {type(prompt)}")
-
         inputs = self.tokenizer(prompt, return_tensors="pt").to(self.device)
         with torch.no_grad():
             outputs = self.model.generate(
@@ -52,8 +49,11 @@ class LocalBartModel:
                 num_beams=5,
                 early_stopping=True,
             )
-        output_text = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return output_text.strip()
+        return self.tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+
+    def generate(self, *args, **kwargs):
+        # Forward generate calls for ToolCallingAgent compatibility
+        return self.model.generate(*args, **kwargs)
 
 class GaiaAgent:
     def __init__(self):
@@ -86,7 +86,7 @@ class GaiaAgent:
             elif isinstance(result, str):
                 return result.strip()
             elif isinstance(result, list):
-                # Try to find assistant response content in list
+                # Find assistant content if possible
                 for item in reversed(result):
                     if isinstance(item, dict) and item.get("role") == "assistant" and "content" in item:
                         return item["content"].strip()