Create agent.py

agent.py

@@ -0,0 +1,71 @@
+# 17/08/2025 -AO
+import math
+import re
+import requests
+from typing import Optional
+
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
+class BasicAgent:
+    """
+    Very simple baseline agent.
+    - Tries to solve basic arithmetic expressions found in the question.
+    - If it cannot, returns a default safe string.
+    Keep responses concise because the grader does exact-match comparison.
+    """
+
+    def __init__(self, api_url: str = DEFAULT_API_URL):
+        self.api_url = api_url
+        # Precompile a simple arithmetic pattern like "12 + 3 * 4" or "(5+2)/7"
+        self.expr_pattern = re.compile(r"^[\s\d\+\-\*/\(\)\.]+$")
+
+    def try_eval_math(self, text: str) -> Optional[str]:
+        # Extract a math-like snippet if the whole string looks like math.
+        s = text.strip()
+        if not s:
+            return None
+        if self.expr_pattern.match(s):
+            try:
+                # Safe eval using eval on limited globals/locals.
+                val = eval(s, {"__builtins": {}}, {"math": math})
+                # Format as plain value string.
+                return str(val)
+            except Exception:
+                return None
+        return None
+
+    def __call__(self, question: str) -> str:
+        # Try direct arithmetic
+        ans = self.try_eval_math(question)
+        if ans is not None:
+            return ans
+
+        # Extremely small heuristics: look for numbers and operations within the sentence.
+        digits = re.findall(r"[\d\.]+", question)
+        if len(digits) >= 2:
+            try:
+                # If words like 'sum' or 'add' appear, add the numbers.
+                ql = question.lower()
+                nums = list(map(float, digits))
+                if any(k in ql for k in ["sum", "add", "+"]):
+                    return str(sum(nums))
+                if any(k in ql for k in ["difference", "subtract", "-"]):
+                    return str(nums[0] - sum(nums[1:]))
+                if any(k in ql for k in ["product", "multiply", "*"]):
+                    p = 1.0
+                    for n in nums:
+                        p *= n
+                    return str(p)
+                if any(k in ql for k in ["divide", "ratio", "/"]):
+                    try:
+                        res = nums[0]
+                        for n in nums[1:]:
+                            res /= n
+                        return str(res)
+                    except ZeroDivisionError:
+                        return "undefined"
+            except Exception:
+                pass
+
+        # Fallback default answer; keep minimal for exact-match grading.
+        return "unknown"