agent

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

type ContentBlock struct {
	Type string `json:"type"` // "text", "tool_use", "tool_result"
	Text string `json:"text,omitempty"`

	// For tool_use
	ID    string                 `json:"id,omitempty"`
	Name  string                 `json:"name,omitempty"`
	Input map[string]interface{} `json:"input,omitempty"`

	// For tool_result
	ToolUseID string `json:"tool_use_id,omitempty"`
	Content   string `json:"content,omitempty"`
	IsError   bool   `json:"is_error,omitempty"`
}

type ConversationHistory struct {
	Messages []Message `json:"messages"`
}

type Message struct {
	Role    string      `json:"role"`    // "user", "assistant"
	Content interface{} `json:"content"` // string or []ContentBlock
}

type QueryResultData struct {
	ColumnNames []string
	Rows        [][]interface{}
	TotalRows   int
	Truncated   bool
}

type QueryResultWithData struct {
	QueryResultData
	AllRows   [][]interface{}
	QueryText string
}

type Tool struct {
	Name        string      `json:"name"`
	Description string      `json:"description"`
	InputSchema ToolSchema  `json:"input_schema"`
	Handler     ToolHandler `json:"-"`
}

type ToolDefinition struct {
	Name        string
	Description string
	InputSchema anthropic.ToolInputSchemaParam
	Function    func(ctx context.Context, input json.RawMessage) (string, error)
}

type ToolHandler func(ctx context.Context, input map[string]interface{}) (*ToolResult, error)

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

type ToolResult struct {
	Content string `json:"content"`
	IsError bool   `json:"is_error,omitempty"`
}

type ToolSchema struct {
	Type       string                 `json:"type"`
	Properties map[string]interface{} `json:"properties"`
	Required   []string               `json:"required,omitempty"`
}