llm

type Agent struct {
	Name              string
	Task              string
	Personality       string
	CompletionRequest CompletionRequest
	Runtime           *shared.RuntimeContext
}

type ChatCompletionRequest struct {
	Messages         []ChatMessage
	Temperature      float64
	MaxTokens        float64
	TopP             float64
	FrequencyPenalty float64
	PresencePenalty  float64
	ResponseFormat   *openai.ChatCompletionNewParamsResponseFormatUnion
}

type ChatMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type CompletionRequest struct {
	ChatCompletionRequest
	ApiKey string
}

type MCPAgent struct {
	Agent
	MCPClient *MCPClient
	Tools     []string // Available MCP tools for this agent
}

type MCPAgentConfig struct {
	Name          string
	Task          string
	Personality   string
	Tools         []string
	MaxTokens     int
	UseTextFormat bool
}

type MCPClient struct {
	ServerURL  string
	HTTPClient *http.Client
	SessionID  string
	Tools      map[string]MCPTool
}

type MCPError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}

type MCPRequest struct {
	Method string      `json:"method"`
	Params interface{} `json:"params"`
}

type MCPResponse struct {
	Result interface{} `json:"result"`
	Error  *MCPError   `json:"error,omitempty"`
}

type MCPTool struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	InputSchema map[string]interface{} `json:"inputSchema"`
}

type ModelVariant int

type Service struct {
	// contains filtered or unexported fields
}

type ServiceBuilder struct {
	// contains filtered or unexported fields
}