add infra db dependency and sample tool server (#38)

add infra db dependency and sample tool server for the tool routing feature

Author: llx18

deployment/infra/azure-deployment.bicep

@@ -427,6 +427,20 @@ resource cacheContainer 'Microsoft.DocumentDB/databaseAccounts/sqlDatabases/cont
   }
 }
 
+resource toolContainer 'Microsoft.DocumentDB/databaseAccounts/sqlDatabases/containers@2023-04-15' = {
+  name: 'ToolContainer'
+  parent: cosmosDbSqlDb
+  properties: {
+    resource: {
+      id: 'ToolContainer'
+      partitionKey: {
+        paths: ['/id']
+        kind: 'Hash'
+      }
+    }
+  }
+}
+
 // Cosmos DB Data Contributor Role Assignment to UAI
 resource cosmosDbRoleAssignment 'Microsoft.DocumentDB/databaseAccounts/sqlRoleAssignments@2022-11-15' = {
   parent: cosmosDb

deployment/infra/azure-deployment.json

@@ -5,7 +5,7 @@
     "_generator": {
       "name": "bicep",
       "version": "0.38.33.27573",
-      "templateHash": "2243900866350713366"
+      "templateHash": "12874722858654947563"
     }
   },
   "parameters": {
@@ -478,6 +478,25 @@
         "[resourceId('Microsoft.DocumentDB/databaseAccounts/sqlDatabases', variables('cosmosDbAccountName'), 'McpGatewayDb')]"
       ]
     },
+    {
+      "type": "Microsoft.DocumentDB/databaseAccounts/sqlDatabases/containers",
+      "apiVersion": "2023-04-15",
+      "name": "[format('{0}/{1}/{2}', variables('cosmosDbAccountName'), 'McpGatewayDb', 'ToolContainer')]",
+      "properties": {
+        "resource": {
+          "id": "ToolContainer",
+          "partitionKey": {
+            "paths": [
+              "/id"
+            ],
+            "kind": "Hash"
+          }
+        }
+      },
+      "dependsOn": [
+        "[resourceId('Microsoft.DocumentDB/databaseAccounts/sqlDatabases', variables('cosmosDbAccountName'), 'McpGatewayDb')]"
+      ]
+    },
     {
       "type": "Microsoft.DocumentDB/databaseAccounts/sqlRoleAssignments",
       "apiVersion": "2022-11-15",

sample-tool-server/Dockerfile

@@ -0,0 +1,20 @@
+# Simple container for the dummy score API service
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Create a non-root user
+RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
+
+COPY requirements.txt ./
+RUN pip install --no-cache-dir --root-user-action=ignore -r requirements.txt
+
+# Copy the application code
+COPY app.py ./
+
+# Switch to non-root user
+USER appuser
+
+EXPOSE 8000
+
+CMD ["python", "app.py"]

sample-tool-server/app.py

@@ -0,0 +1,36 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+
+app = FastAPI()
+
+
+class WeatherRequest(BaseModel):
+    location: str | None = None
+    unit: str | None = None
+
+
+@app.post("/score")
+async def get_weather(payload: WeatherRequest):
+    """Mocked weather endpoint matching the MCP tool contract."""
+    location = payload.location
+    unit = (payload.unit or "fahrenheit").lower()
+    if not location:
+        raise HTTPException(status_code=400, detail="Missing 'location' field.")
+    if unit not in {"celsius", "fahrenheit"}:
+        raise HTTPException(status_code=400, detail="Invalid 'unit' value.")
+
+    # Static temperature keeps the mock predictable for tests.
+    base_temp_f = 72.0
+    temperature = base_temp_f if unit == "fahrenheit" else round((base_temp_f - 32) * 5 / 9, 1)
+    return {
+        "location": location,
+        "unit": unit,
+        "temperature": temperature,
+        "conditions": "clear",
+    }
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=8000)

sample-tool-server/requirements.txt

@@ -0,0 +1,2 @@
+fastapi==0.111.0
+uvicorn[standard]==0.30.0