llms

type ChatGeneration struct {
	// Message is the generated AI message.
	Message *core.AIMessage `json:"message"`

	// GenerationInfo contains generation-specific metadata.
	GenerationInfo map[string]any `json:"generation_info,omitempty"`
}

type ChatModel interface {
	core.Runnable[[]core.Message, *core.AIMessage]

	// Generate performs a chat completion and returns detailed results
	// including token usage.
	Generate(ctx context.Context, messages []core.Message, opts ...core.Option) (*ChatResult, error)

	// BindTools returns a new ChatModel that will use the given tool definitions
	// when generating responses.
	BindTools(tools ...ToolDefinition) ChatModel

	// WithStructuredOutput configures the model to return structured output
	// matching the given JSON schema.
	WithStructuredOutput(schema map[string]any) ChatModel
}

type ChatResult struct {
	// Generations contains the generated messages.
	Generations []*ChatGeneration `json:"generations"`

	// LLMOutput contains provider-specific output data.
	LLMOutput map[string]any `json:"llm_output,omitempty"`
}

type TokenUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type ToolDefinition struct {
	// Name of the tool.
	Name string `json:"name"`

	// Description of what the tool does.
	Description string `json:"description"`

	// Parameters is a JSON Schema describing the tool's parameters.
	Parameters map[string]any `json:"parameters"`
}