core

type AIMessage struct {
	BaseMessage
	ToolCalls     []ToolCall     `json:"tool_calls,omitempty"`
	UsageMetadata *UsageMetadata `json:"usage_metadata,omitempty"`
}

type AgentActionData struct {
	Tool      string `json:"tool"`
	ToolInput string `json:"tool_input"`
	Log       string `json:"log"`
}

type AgentFinishData struct {
	Output map[string]any `json:"output"`
	Log    string         `json:"log"`
}

type BaseCallbackHandler struct{}

type BaseMessage struct {
	Content          string         `json:"content"`
	Name             string         `json:"name,omitempty"`
	ID               string         `json:"id,omitempty"`
	AdditionalKwargs map[string]any `json:"additional_kwargs,omitempty"`
	ResponseMetadata map[string]any `json:"response_metadata,omitempty"`
}

type CallbackHandler interface {
	// LLM callbacks
	OnLLMStart(ctx context.Context, prompts []string, runID string, parentRunID string, extras map[string]any)
	OnChatModelStart(ctx context.Context, messages []Message, runID string, parentRunID string, extras map[string]any)
	OnLLMNewToken(ctx context.Context, token string, runID string)
	OnLLMEnd(ctx context.Context, output *LLMResult, runID string)
	OnLLMError(ctx context.Context, err error, runID string)

	// Chain callbacks
	OnChainStart(ctx context.Context, inputs map[string]any, runID string, parentRunID string, extras map[string]any)
	OnChainEnd(ctx context.Context, outputs map[string]any, runID string)
	OnChainError(ctx context.Context, err error, runID string)

	// Tool callbacks
	OnToolStart(ctx context.Context, toolName string, input string, runID string, parentRunID string)
	OnToolEnd(ctx context.Context, output string, runID string)
	OnToolError(ctx context.Context, err error, runID string)

	// Agent callbacks
	OnAgentAction(ctx context.Context, action AgentActionData, runID string)
	OnAgentFinish(ctx context.Context, finish AgentFinishData, runID string)

	// Retriever callbacks
	OnRetrieverStart(ctx context.Context, query string, runID string, parentRunID string)
	OnRetrieverEnd(ctx context.Context, documents []*Document, runID string)
	OnRetrieverError(ctx context.Context, err error, runID string)

	// Text callbacks
	OnText(ctx context.Context, text string, runID string)
}

type Document struct {
	// PageContent is the text content of the document.
	PageContent string `json:"page_content"`

	// Metadata contains arbitrary key-value pairs associated with the document.
	Metadata map[string]any `json:"metadata,omitempty"`

	// ID is an optional unique identifier for the document.
	ID string `json:"id,omitempty"`
}

type FunctionMessage struct {
	BaseMessage
}

type GenericMessage struct {
	BaseMessage
	Role string `json:"role"`
}

type HumanMessage struct {
	BaseMessage
}

type LLMResult struct {
	Generations []string       `json:"generations"`
	LLMOutput   map[string]any `json:"llm_output,omitempty"`
}

type Message interface {
	// GetType returns the message type (human, ai, system, tool, function).
	GetType() MessageType
	// GetContent returns the text content of the message.
	GetContent() string
	// GetName returns the optional name associated with the message.
	GetName() string
	// GetAdditionalKwargs returns additional provider-specific data.
	GetAdditionalKwargs() map[string]any
}

type MessageType string

type Option func(*RunnableConfig)

type Runnable[I, O any] interface {
	// Invoke transforms a single input into an output.
	Invoke(ctx context.Context, input I, opts ...Option) (O, error)

	// Stream transforms an input and streams output chunks as they are produced.
	// The caller must consume the returned StreamIterator until it is exhausted
	// or call Close() to release resources.
	Stream(ctx context.Context, input I, opts ...Option) (*StreamIterator[O], error)

	// Batch transforms multiple inputs in parallel.
	// Implementations honor WithMaxConcurrency when they delegate to the shared
	// batch helper used throughout this library.
	Batch(ctx context.Context, inputs []I, opts ...Option) ([]O, error)

	// GetName returns the name of this runnable for tracing and debugging.
	GetName() string
}

type RunnableConfig struct {
	// Tags for this call and any sub-calls (used for filtering and tracing).
	Tags []string

	// Metadata for this call and any sub-calls.
	Metadata map[string]any

	// Callbacks are the callback handlers for this call.
	Callbacks []CallbackHandler

	// RunName overrides the default name for tracing.
	RunName string

	// MaxConcurrency limits parallel calls in Batch operations.
	// 0 means no limit.
	MaxConcurrency int

	// RecursionLimit is the maximum recursion depth. Default is 25.
	RecursionLimit int

	// Configurable holds runtime values for configurable fields.
	Configurable map[string]any

	// RunID is a unique identifier for this run. Auto-generated if empty.
	RunID string

	// Stop sequences to pass to the model.
	Stop []string
}

type StreamChunk[T any] struct {
	Value T
	Err   error
}

type StreamIterator[T any] struct {
	// contains filtered or unexported fields
}

type SystemMessage struct {
	BaseMessage
}

type ToolCall struct {
	ID   string          `json:"id"`
	Name string          `json:"name"`
	Args json.RawMessage `json:"args"`
	Type string          `json:"type,omitempty"`
}

type ToolMessage struct {
	BaseMessage
	ToolCallID string `json:"tool_call_id"`
}

type UsageMetadata struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
	TotalTokens  int `json:"total_tokens"`
}