d15

app.py

@@ -6,6 +6,8 @@ import pandas as pd
 from huggingface_hub import InferenceClient
 from dotenv import load_dotenv
 import json
+from huggingface_hub import InferenceClient
+import traceback
 
 load_dotenv()
 
@@ -13,6 +15,8 @@ load_dotenv()
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
+
+
 class BasicAgent:
     def __init__(self):
         self.token = os.getenv("token")
@@ -22,11 +26,16 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         prompt = f"Answer this question concisely and clearly. Only return the final answer.\nQuestion: {question}"
         try:
-            response = self.client.text_generation(prompt, max_new_tokens=100)
-            return response.strip()
+            response = self.client.conversational(
+                messages=[{"role": "user", "content": prompt}]
+            )
+            print("✅ Raw response:", response)
+            return response[0]["generated_text"].strip()
         except Exception as e:
-            print(f"Error calling inference API: {e}")
-            return f"error: {e} | token used: {self.token}"
+            print("❌ Error calling inference API")
+            traceback.print_exc()
+            return f"error: {str(e)}"
+
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):