MCP Tools

import (
    "context"
    "github.com/hastekit/hastekit-sdk-go/pkg/agents/mcpclient"
)

mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
    mcpclient.WithTransport("sse"),
)
if err != nil {
    log.Fatal(err)
}

mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
    mcpclient.WithTransport("sse"),
	mcpclient.WithHeaders(map[string]string{
        "token": "your-token",
    }),
)

// Only use specific tools
mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
    mcpclient.WithTransport("sse"),
		mcpclient.WithToolFilter("list_users"),
	)

mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
    mcpclient.WithTransport("sse"),
    mcpclient.WithDeferredTools("rarely_used_tool"),
)

mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
    mcpclient.WithTransport("sse"),
		mcpclient.WithCacheTTL(5*time.Minute),
		mcpclient.WithSchemaCache(myRedisSchemaCache),
	)

mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
		mcpclient.WithTransport("streamable-http"), // or "sse"
	)

package main

import (
    "context"
    "fmt"
    "log"
    "os"

	"github.com/bytedance/sonic"
	"github.com/hastekit/hastekit-sdk-go/pkg/agents"
	"github.com/hastekit/hastekit-sdk-go/pkg/agents/history"
	"github.com/hastekit/hastekit-sdk-go/pkg/agents/mcpclient"
	"github.com/hastekit/hastekit-sdk-go/pkg/gateway/llm/responses"
	hastekit "github.com/hastekit/hastekit-sdk-go"
)

func main() {
    // Initialize LLM client
    client := hastekit.NewLLMClient([]hastekit.ProviderConfig{
        {
            ProviderName: hastekit.ProviderOpenAI,
            ApiKeys: []*hastekit.APIKeyConfig{
                {
                    Name:   "Key 1",
                    APIKey: os.Getenv("OPENAI_API_KEY"),
                },
            },
        },
    })

    // Bind a model
    model := client.Model("OpenAI/gpt-4o-mini")

    // Create conversation history
    conv := hastekit.NewFileHistory("./conversations")

    // Connect to MCP server
    mcpClient, err := mcpclient.NewClient(context.Background(), "http://localhost:9001/sse",
		mcpclient.WithHeaders(map[string]string{
			"token": "your-token",
		}),
		mcpclient.WithToolFilter("list_users"),
	)

    // Create agent with MCP tools
    agent := hastekit.NewAgent(&hastekit.AgentConfig{
        Name:        "Hello world agent",
        Instruction: hastekit.NewPrompt("You are helpful assistant."),
        LLM:         model,
        History:     conv,
        McpServers:  []agents.MCPToolset{mcpClient},
    })

    // Execute agent
    handle, err := agent.Execute(context.Background(), &agents.AgentInput{
        Message: history.Message{
            Messages: []responses.InputMessageUnion{
                responses.UserMessage("Hello!"),
            },
        },
    })
    if err != nil {
        log.Fatal(err)
    }
    out, err := handle.Result()
    if err != nil {
        log.Fatal(err)
    }

    b, _ := sonic.Marshal(out)
    fmt.Println(string(b))
}