models

type ChatMessage struct {
	Role        string       `json:"role"`
	Content     string       `json:"content,omitempty"`
	ToolCalls   []ToolCall   `json:"tool_calls,omitempty"`
	ToolResults []ToolResult `json:"tool_results,omitempty"`
}

type GenerateMeta struct {
	DurationMs       int64   `json:"duration_ms"`
	PromptTokens     int     `json:"prompt_tokens,omitempty"`
	CompletionTokens int     `json:"completion_tokens,omitempty"`
	CostUSD          float64 `json:"cost_usd,omitempty"`
}

type GenerateRequest struct {
	Model      string        `json:"model"`
	Messages   []ChatMessage `json:"messages"`
	Tools      []ToolDef     `json:"tools,omitempty"`
	ToolChoice string        `json:"tool_choice,omitempty"`
	// MaxTokens caps the completion's output tokens. Zero leaves it to the adapter's own default
	// (Anthropic requires max_tokens on every request; OpenAI omits it, taking the API default). The
	// engine sets it from the agent's constraints.maxTokens (issue #514).
	MaxTokens int `json:"max_tokens,omitempty"`
	// Temperature is the sampling temperature to send to the provider. Nil leaves it unset so the
	// provider default applies; a non-nil value (including 0 for deterministic output) is sent
	// verbatim. Adapters translate it to the provider's request field (issue #388).
	Temperature *float64 `json:"temperature,omitempty"`
	// ResponseFormat, when non-nil, asks the provider to constrain the completion to a JSON Schema
	// ("structured outputs"). Nil leaves the output unconstrained. Adapters that support it translate
	// it to the provider request field (Anthropic output_config.format, OpenAI response_format); the
	// engine sets it for an agent whose constraints require structured output (issue #510).
	ResponseFormat *ResponseFormat `json:"response_format,omitempty"`
}

type GenerateResponse struct {
	// Content is assistant message text when the model replies directly.
	Content    string       `json:"content,omitempty"`
	ToolCalls  []ToolCall   `json:"tool_calls,omitempty"`
	StopReason string       `json:"stop_reason,omitempty"`
	Meta       GenerateMeta `json:"meta"`
}

type MockClient struct {
	Content string
	Meta    *GenerateMeta
	Script  []MockTurn
	// contains filtered or unexported fields
}

type MockTurn struct {
	Content    string
	ToolCalls  []ToolCall
	StopReason string
	// Meta is per-call accounting (token counts, cost). When nil, [MockClient.Meta] or a small default is used.
	// Token counts with CostUSD 0 are priced via the B1 table for req.Model (issue #164).
	Meta *GenerateMeta
	// Err, when set, is returned from Generate instead of a response.
	Err error
}

type ModelClient interface {
	Generate(ctx context.Context, req GenerateRequest) (GenerateResponse, error)
}

type OpenAIClient struct {
	APIKey     string
	BaseURL    string
	HTTPClient *http.Client
	// CostProvider selects the pricing table used to estimate CostUSD.
	// Empty means the OpenAI table (costProviderOpenAI).
	CostProvider string
}

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

type ResponseFormat struct {
	// Name is a short schema identifier some providers require (OpenAI json_schema.name); adapters
	// that do not need it (Anthropic) ignore it. It must match ^[a-zA-Z0-9_-]{1,64}$.
	Name string `json:"name,omitempty"`
	// Schema is the JSON Schema the completion must conform to.
	Schema json.RawMessage `json:"schema,omitempty"`
}

type ToolCall struct {
	ID        string          `json:"id"`
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments"`
}

type ToolDef struct {
	Name        string          `json:"name"`
	Description string          `json:"description,omitempty"`
	Parameters  json.RawMessage `json:"parameters,omitempty"`
}

type ToolResult struct {
	ToolCallID string `json:"tool_call_id"`
	Content    string `json:"content"`
	// IsError marks this result as a failed tool call. Providers that model it (Anthropic's
	// tool_result.is_error) receive the flag so the model treats the content as an error observation
	// to correct from, not a successful output; providers without the concept (OpenAI) ignore it and
	// carry the content alone. A failed or malformed tool call must always be answered by a
	// well-formed tool_result — never left as a dangling tool_use that poisons the next request (#524).
	IsError bool `json:"is_error,omitempty"`
}