MCP Servers (Beta)

import os
from datagrid_ai import Datagrid

client = Datagrid(api_key=os.environ.get("DATAGRID_API_KEY"))

response = client.converse(
    prompt="Tell me about my projects",
    chat_mode="full_agent",
    config={
        "mcp_servers": [
            {
                "type": "inline_mcp",
                "server_label": "project-api",
                "server_url": "https://mcp.mycompany.com",
                "server_description": "Project management API",
                "authorization": "Bearer your-api-token"
            }
        ]
    }
)

print(response.content[0].text)

import Datagrid from 'datagrid-ai';

const client = new Datagrid({
  apiKey: process.env['DATAGRID_API_KEY']
});

const response = await client.converse({
  prompt: 'Tell me about my projects',
  chat_mode: 'full_agent',
  config: {
    mcp_servers: [
      {
        type: 'inline_mcp',
        server_label: 'project-api',
        server_url: 'https://mcp.mycompany.com',
        server_description: 'Project management API',
        authorization: 'Bearer your-api-token'
      }
    ]
  }
});

console.log(response.content[0].text);

curl -X POST https://api.datagrid.com/v1/converse \
  -H "Authorization: Bearer $DATAGRID_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "prompt": "Tell me about my projects",
    "chat_mode": "full_agent",
    "config": {
      "mcp_servers": [{
        "type": "inline_mcp",
        "server_label": "project-api",
        "server_url": "https://mcp.mycompany.com",
        "server_description": "Project management API",
        "authorization": "Bearer your-api-token"
      }]
    }
  }'

response = client.converse(
    prompt="Get the weather forecast and add it to my calendar",
    config={
        "mcp_servers": [
            {
                "type": "inline_mcp",
                "server_label": "weather",
                "server_url": "https://mcp.example.com/weather",
                "authorization": "Bearer weather-token"
            },
            {
                "type": "inline_mcp",
                "server_label": "calendar",
                "server_url": "https://mcp.example.com/calendar",
                "authorization": "Bearer calendar-token"
            }
        ]
    }
)

const response = await client.converse({
  prompt: 'Get the weather forecast and add it to my calendar',
  config: {
    mcp_servers: [
      {
        type: 'inline_mcp',
        server_label: 'weather',
        server_url: 'https://mcp.example.com/weather',
        authorization: 'Bearer weather-token'
      },
      {
        type: 'inline_mcp',
        server_label: 'calendar',
        server_url: 'https://mcp.example.com/calendar',
        authorization: 'Bearer calendar-token'
      }
    ]
  }
});