Azure MCP Server

Instructions

Implementation Reference

• mcp_server_azure/azure_server.py:302-314 (handler)

  Handler logic inside handle_cosmosdb_operations that executes the cosmosdb_item_query tool by querying items in the specified container using the provided SQL query and parameters.

  elif (
      name == "cosmosdb_item_query"
  ):  # Renamed table to container, simplified query
      container_client = database.get_container_client(
          arguments["container_name"]
      )
      items = list(
          container_client.query_items(
              query=arguments["query"],
              parameters=arguments.get("parameters", []),  # Optional parameters
          )
      )
      response = {"items": items}

• mcp_server_azure/azure_tools.py:295-327 (schema)

  Input schema definition for the cosmosdb_item_query tool, specifying required container_name and query, with optional database_name and parameters.

  Tool(
      name="cosmosdb_item_query",  # Renamed table to container, simplified query
      description="Query items in a Cosmos DB container using SQL",  # Updated description
      inputSchema={
          "type": "object",
          "properties": {
              "container_name": {  # Renamed from table_name
                  "type": "string",
                  "description": "Name of the Cosmos DB container",  # Updated description
              },
              "database_name": {
                  "type": "string",
                  "description": "Name of the Cosmos DB database (optional, defaults to 'defaultdb')",
              },
              "query": {
                  "type": "string",
                  "description": "Cosmos DB SQL query string",
              },
              "parameters": {
                  "type": "array",
                  "description": "Parameters for the SQL query (optional)",
                  "items": {
                      "type": "object",
                      "properties": {
                          "name": {"type": "string"},
                          "value": {},  # Value can be any type
                      },
                  },
              },
          },
          "required": ["container_name", "query"],
      },
  ),