Create basic thought action observation loop agent

app.py

@@ -8,54 +8,76 @@ import json
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
+
     def __init__(self, metadata_path="metadata.jsonl"):
-        # Load metadata.jsonl
         self.metadata = self._load_metadata(metadata_path)
         print("BasicAgent initialized with metadata")
 
-    def _load_metadata(self, file_path):
-        """Load metadata from a JSONL file, parsing each line as a JSON object."""
-        data = []
-        try:
-            with open(file_path, 'r', encoding='utf-8') as f:
-                for line_number, line in enumerate(f, 1):
-                    line = line.strip()
-                    if not line:
-                        continue
-                    try:
-                        obj = json.loads(line)
-                        if isinstance(obj, dict):
-                            data.append(obj)
-                        else:
-                            print(f"Skipping line {line_number}: not a dictionary")
-                    except json.JSONDecodeError as e:
-                        print(f"Error parsing line {line_number}: {e}")
-            print(f"Loaded metadata from '{file_path}' with {len(data)} entries")
-            return data
-        except FileNotFoundError:
-            print(f"Metadata file '{file_path}' not found. Proceeding without metadata.")
-            return []
-        except Exception as e:
-            print(f"Unexpected error loading metadata from '{file_path}': {e}")
-            return []
+    def think(self, question, context):
+
+        if "FOUND:" in context:
+            return "I found the answer, I should return it"
+
+        if "search" not in context:
+            return "I should search metadata for this question"
+
+        return "The search did not help, answer unknown"
+
+    def decide_action(self, thought):
+
+        if "search" in thought.lower():
+            return "search_metadata"
+
+        if "return" in thought.lower():
+            return "answer"
+
+        return "answer"
+
+    def run_action(self, action):
+
+        if action == "search_metadata":
+
+            for item in self.metadata:
+                if item.get("Question") == self.current_question:
+                    return f"FOUND:{item.get('Final answer')}"
+
+            return "NOT_FOUND"
+
+        if action == "answer":
+
+            for item in self.metadata:
+                if item.get("Question") == self.current_question:
+                    return f"FINAL_ANSWER:{item.get('Final answer')}"
+
+            return "FINAL_ANSWER:unknown"
+
+        return "No action executed"
 
     def __call__(self, question: str) -> str:
-        """Search metadata for the question and return the final answer or 'unknown'."""
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        
-        # Search metadata.jsonl for the question
-        for item in self.metadata:
-            if item.get("Question") == question:
-                final_answer = item.get("Final answer")
-                if final_answer:
-                    print(f"Found answer in metadata: {final_answer}")
-                    return final_answer
-                else:
-                    print("Question found in metadata, but no final answer provided.")
-        
-        # Fallback if question not found
-        print("Question not found in metadata. Returning 'unknown'.")
+
+        print(f"Agent received question: {question}")
+
+        self.current_question = question
+        context = ""
+
+        for step in range(5):
+
+            thought = self.think(question, context)
+            print("Thought:", thought)
+
+            action = self.decide_action(thought)
+            print("Action:", action)
+
+            observation = self.run_action(action)
+            print("Observation:", observation)
+
+            context += f"\nThought:{thought}\nObservation:{observation}"
+
+            if "FINAL_ANSWER:" in observation:
+                return observation.replace("FINAL_ANSWER:", "").strip()
+
         return "unknown"
+    
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """