Usage with MCP

type CalculatorInput struct {
  Operation string  `json:"operation" jsonschema_description:"The arithmetic operation to perform (add, subtract, multiply, divide)"`
  X         float64 `json:"x" jsonschema_description:"First number"`
  Y         float64 `json:"y" jsonschema_description:"Second number"`
}

kai := ai.NewKarmaAI(
  ai.GPT4o,         // use a model that supports MCP/tool calling
  ai.OpenAI,        // pick a provider
  ai.WithMaxTokens(1000),
  ai.WithTemperature(1),
  ai.WithTopP(0.9),
  ai.WithTopK(50),

  // MCP config
  ai.SetMCPUrl("http://localhost:8086/mcp"),
  ai.SetMCPAuthToken(config.GetEnvRaw("TEST_TOKEN")),
  ai.SetMCPTools(ai.MCPTool{
    FriendlyName: "Calculator",
    ToolName:     "calculate",
    Description:  "Perform basic arithmetic operations (add, subtract, multiply, divide).",
    InputSchema:  CalculatorInput{},
  }),
)

history := models.AIChatHistory{
  Messages: []models.AIMessage{
    {
      Role:    models.User,
      Message: "Please calculate 123 + 456. Use the calculator tool provided.",
    },
  },
  ChatId:    "example-chat-2",
  CreatedAt: time.Now(),
  Title:     "Using MCP Tools",
}

resp, err := kai.ChatCompletion(history)
if err != nil { panic(err) }
fmt.Println(resp.AIResponse)

// Replace any existing MCP servers
func SetMCPServers(servers ...MCPServer) ai.Option

// Append a new MCP server
func AddMCPServer(url, authToken string, tools ...MCPTool) ai.Option

kai := ai.NewKarmaAI(
	ai.GPT4o,         // use a model that supports MCP/tool calling
  ai.OpenAI,        // pick a provider
  ai.AddMCPServer("http://localhost:8086/mcp", "token1",
    ai.MCPTool{FriendlyName: "Calc", ToolName: "calculate", InputSchema: CalculatorInput{}},
  ),
  ai.AddMCPServer("https://tools.mycompany.com/mcp", "token2",
    ai.MCPTool{FriendlyName: "Weather", ToolName: "weather_lookup", InputSchema: struct {
      City string `json:"city"`
    }{}},
  ),
)