use pydantic

web/agent/src/agent.py

@@ -0,0 +1,62 @@
+"""
+PydanticAI agent with MCP tools from Cavepedia server.
+"""
+
+import logging
+import httpx
+
+from pydantic_ai import Agent
+from pydantic_ai.models.google import GoogleModel
+
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+MCP_URL = "https://mcp.caving.dev/mcp"
+
+logger.info("Initializing Cavepedia agent...")
+
+def check_mcp_available(url: str, timeout: float = 5.0) -> bool:
+    """Check if MCP server is reachable."""
+    try:
+        # Just check if we can connect - don't need a full MCP handshake
+        response = httpx.get(url, timeout=timeout, follow_redirects=True)
+        # Any response (even 4xx/5xx) means server is reachable
+        # 502 means upstream is down, so treat as unavailable
+        if response.status_code == 502:
+            logger.warning(f"MCP server returned 502 Bad Gateway")
+            return False
+        return True
+    except Exception as e:
+        logger.warning(f"MCP server not reachable: {e}")
+        return False
+
+# Try to configure MCP if server is available
+toolsets = []
+if check_mcp_available(MCP_URL):
+    try:
+        from pydantic_ai.mcp import MCPServerStreamableHTTP
+        mcp_server = MCPServerStreamableHTTP(
+            url=MCP_URL,
+            timeout=30.0,
+        )
+        toolsets.append(mcp_server)
+        logger.info(f"MCP server configured: {MCP_URL}")
+    except Exception as e:
+        logger.warning(f"Could not configure MCP server: {e}")
+else:
+    logger.info("MCP server unavailable - running without MCP tools")
+
+# Create the agent with Google Gemini model
+agent = Agent(
+    model=GoogleModel("gemini-2.5-pro"),
+    toolsets=toolsets if toolsets else None,
+    instructions="""You are a helpful caving assistant. Help users with all aspects of caving including cave exploration, safety, surveying techniques, cave locations, geology, equipment, history, conservation, and any other caving-related topics.
+
+IMPORTANT RULES:
+1. Always cite your sources at the end of each response when possible.
+2. If you're not certain about information, say so clearly. Do NOT make up information or hallucinate facts.
+3. Provide accurate, helpful, and safety-conscious information.""",
+)
+
+logger.info(f"Agent initialized successfully (MCP: {'enabled' if toolsets else 'disabled'})")