llm

type AgentConfigUpdate struct {
	ID           string
	Instructions *string
	ToolsAdded   []string
	ToolsRemoved []string
	CreatedAt    time.Time
}

type AgentHandoff struct {
	ID         string
	OldAgentID *string
	NewAgentID string
	CreatedAt  time.Time
}

type AudioContent struct {
	Frames     []any
	Transcript string
}

type ChatChunk struct {
	ID    string
	Delta *ChoiceDelta
	Usage *CompletionUsage
}

type ChatContent struct {
	Text  string
	Image *ImageContent
	Audio *AudioContent
}

type ChatContext struct {
	Items       []ChatItem          `json:"items"`
	OnItemAdded func(item ChatItem) `json:"-"`
}

type ChatItem interface {
	GetID() string
	GetType() string
	GetCreatedAt() time.Time
}

type ChatMessage struct {
	ID                   string
	Role                 ChatRole
	Content              []ChatContent
	Interrupted          bool
	TranscriptConfidence *float64
	Extra                map[string]any
	CreatedAt            time.Time
}

type ChatOption func(*ChatOptions)

type ChatOptions struct {
	Tools             []interface{}
	ToolChoice        ToolChoice
	ParallelToolCalls bool
}

type ChatRole string

type ChoiceDelta struct {
	Role      ChatRole
	Content   string
	ToolCalls []FunctionToolCall
	Extra     map[string]any
}

type CompletionUsage struct {
	CompletionTokens   int
	PromptTokens       int
	PromptCachedTokens int
	TotalTokens        int
}

type DiffOps struct {
	ToRemove []string
	ToCreate [][2]*string // [previous_item_id, id]
	ToUpdate [][2]*string // [previous_item_id, id]
}

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

type FunctionCall struct {
	ID        string
	CallID    string
	Name      string
	Arguments string
	Extra     map[string]any
	GroupID   *string
	CreatedAt time.Time
}

type FunctionCallOutput struct {
	ID        string
	CallID    string
	Name      string
	Output    string
	IsError   bool
	CreatedAt time.Time
}

type FunctionToolCall struct {
	Type      string
	Name      string
	Arguments string
	CallID    string
	Extra     map[string]any
}

type ImageContent struct {
	ID              string
	Image           any
	InferenceWidth  *int
	InferenceHeight *int
	InferenceDetail string
	MimeType        string
}

type LLM interface {
	Chat(ctx context.Context, chatCtx *ChatContext, opts ...ChatOption) (LLMStream, error)
}

type LLMFactory func(model string) (LLM, error)

type LLMStream interface {
	Next() (*ChatChunk, error)
	Close() error
}

type MCPServer interface {
	Initialize(ctx context.Context) error
	ListTools(ctx context.Context) ([]Tool, error)
	Close() error
}

type MCPServerHTTP struct {
	URL           string
	TransportType string
	AllowedTools  []string
	Headers       map[string]string
	// contains filtered or unexported fields
}

type MCPServerStdio struct {
	Command string
	Args    []string
	Env     map[string]string
	Cwd     string
	// contains filtered or unexported fields
}

type MetricsReport struct {
	Usage     telemetry.UsageSummary
	CreatedAt time.Time
}

type ProviderTool interface {
	IsProviderTool() bool
	Name() string
	ProviderSchema(format string) map[string]any
}

type RawFunctionTool struct {
	ToolName        string
	ToolDescription string
	ToolParameters  map[string]any
	ExecuteFunc     func(ctx context.Context, args map[string]any) (any, error)
}

type RealtimeCapabilities struct {
	MessageTruncation       bool
	TurnDetection           bool
	UserTranscription       bool
	AutoToolReplyGeneration bool
	AudioOutput             bool
}

type RealtimeEvent struct {
	Type     RealtimeEventType
	Data     []byte // For audio frames
	Text     string // For text deltas
	Function *FunctionToolCall
	Error    error
}

type RealtimeEventType string

type RealtimeModel interface {
	Session() (RealtimeSession, error)
	Close() error
}

type RealtimeSession interface {
	UpdateInstructions(instructions string) error
	UpdateChatContext(chatCtx *ChatContext) error
	UpdateTools(tools []interface{}) error
	Interrupt() error
	Close() error
	EventCh() <-chan RealtimeEvent
	PushAudio(frame *model.AudioFrame) error
}

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

type Tool interface {
	ID() string
	Name() string
	Description() string
	Parameters() map[string]any
	Execute(ctx context.Context, args any) (any, error)
}

type ToolChoice any

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

type ToolWithArgs interface {
	Tool
	Args() any
}

type ToolWithReply interface {
	Tool
	IsReplyRequired() bool
}

type Toolset interface {
	ID() string
	Tools() []interface{}
}