Create mock_smolagents.py

mock_smolagents.py

@@ -0,0 +1,81 @@
+# mock_smolagents.py
+# Minimal mock implementation for environments where smolagents is unavailable
+
+from typing import Callable, List
+
+
+def tool(func: Callable) -> Callable:
+    """Mock @tool decorator"""
+    func.name = func.__name__
+    return func
+
+
+class LiteLLMModel:
+    """Mock LLM model"""
+    def __init__(self, model_id: str, api_key: str = None):
+        self.model_id = model_id
+        self.api_key = api_key
+
+
+class ToolCallingAgent:
+    def __init__(
+        self,
+        model,
+        tools,
+        max_steps=5,
+        verbose=False,
+        add_base_tools=False,
+    ):
+        self.tools = tools
+        self.verbose = verbose
+
+    def run(self, task: str) -> str:
+        task_lower = task.lower()
+
+        # -------- Calculator --------
+        if "calculate" in task_lower or any(op in task_lower for op in ["+", "-", "*", "/"]):
+            for tool in self.tools:
+                if tool.name == "calculator":
+                    import re
+                    expr = re.sub(r"[^\d\+\-\*\/\(\)\.\s]", "", task)
+                    return tool(expr)
+
+        # -------- Dataset Info --------
+        if "dataset" in task_lower or "mnist" in task_lower or "glue" in task_lower:
+            for tool in self.tools:
+                if tool.name == "get_hf_dataset_info":
+                    if "mnist" in task_lower:
+                        return tool("mnist")
+                    elif "glue" in task_lower:
+                        return tool("glue")
+                    return tool("mnist")
+
+        # -------- Translation --------
+        if "translate" in task_lower:
+            for tool in self.tools:
+                if tool.name == "translate_text":
+                    try:
+                        parts = task_lower.replace("translate", "").split("to")
+                        text = parts[0].strip()
+                        lang = parts[1].strip().title()
+                        return tool(text, lang)
+                    except Exception:
+                        return tool("hello", "Spanish")
+
+        # -------- Web Search --------
+        if "search" in task_lower or "find" in task_lower:
+            for tool in self.tools:
+                if tool.name == "search_web":
+                    query = task_lower.replace("search", "").replace("find", "").strip()
+                    return tool(query)
+
+        # -------- Fallback --------
+        return (
+            "🤖 **Mock Agent Response**\n\n"
+            "I couldn’t understand the request.\n\n"
+            "Try:\n"
+            "- `calculate 2 + 2`\n"
+            "- `translate hello to French`\n"
+            "- `tell me about mnist dataset`\n"
+            "- `search machine learning`"
+        )