aisdk

type AnthropicProviderMetadata struct {
	Signature string `json:"signature,omitempty"` // Optional signature for reasoning
}

type Attachment struct {
	Name        string `json:"name,omitempty"`
	ContentType string `json:"contentType,omitempty"`
	URL         string `json:"url"`
}

type Chat struct {
	ID       string    `json:"id"`
	Messages []Message `json:"messages"`
}

type DataPart struct {
	TypeSuffix string `json:"-"` // Not serialized, used to construct type
	Data       any    `json:"data"`
}

type DataStream iter.Seq2[DataStreamPart, error]

type DataStreamPart interface {
	Type() string
}

type ErrorPart struct {
	ErrorText string `json:"errorText"`
}

type FilePart struct {
	URL       string `json:"url"`
	MediaType string `json:"mediaType"`
}

type FinishPart struct {
	FinishReason FinishReason `json:"finishReason,omitzero"`
}

type FinishReason string

type FinishStepPart struct{}

type GoogleProviderMetadata struct {
	ThoughtSignature []byte `json:"thoughtSignature,omitempty"` // Thought signature for function calls (Gemini 3+)
}

type GoogleStreamIterator interface {
	Next() (*genai.GenerateContentResponse, error)
}

type Message struct {
	ID          string           `json:"id"`
	CreatedAt   *json.RawMessage `json:"createdAt,omitempty"`
	Content     string           `json:"content"`
	Role        string           `json:"role"`
	Parts       []Part           `json:"parts,omitempty"`
	Annotations []any            `json:"annotations,omitempty"`
	Attachments []Attachment     `json:"experimental_attachments,omitempty"`
}

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

type MessageStartPart struct {
	MessageID string `json:"messageId,omitempty"`
}

type Part struct {
	Type PartType `json:"type"`

	// Type: "text"
	Text string `json:"text,omitempty"`

	// Type: "reasoning"
	Reasoning string            `json:"reasoning,omitempty"`
	Details   []ReasoningDetail `json:"details,omitempty"`

	// Type: "tool-invocation"
	ToolCallID string              `json:"toolCallId"`
	ToolName   string              `json:"toolName"`
	Input      any                 `json:"input"`
	Output     any                 `json:"output,omitempty"`
	State      ToolInvocationState `json:"state,omitempty"`
	ErrorText  string              `json:"errorText,omitempty"`

	// Type: "source"
	Source *SourceInfo `json:"source,omitempty"`

	// Type: "file"
	MimeType string `json:"mimeType,omitempty"`
	Data     []byte `json:"data,omitempty"`

	ProviderMetadata *ProviderMetadata `json:"providerMetadata,omitzero"`
}

type PartType string

type ProviderMetadata struct {
	Anthropic *AnthropicProviderMetadata `json:"anthropic,omitzero"`
	Google    *GoogleProviderMetadata    `json:"google,omitzero"`
}

type ReasoningDeltaPart struct {
	ID               string           `json:"id"`
	Delta            string           `json:"delta"`
	ProviderMetadata ProviderMetadata `json:"providerMetadata,omitzero"`
}

type ReasoningDetail struct {
	Type      string `json:"type"`
	Text      string `json:"text,omitempty"`
	Signature string `json:"signature,omitempty"`
	Data      string `json:"data,omitempty"`
}

type ReasoningEndPart struct {
	ID string `json:"id"`
}

type ReasoningStartPart struct {
	ID string `json:"id"`
}

type Schema struct {
	Required   []string       `json:"required"`
	Properties map[string]any `json:"properties"`
}

type SourceDocumentPart struct {
	SourceID  string `json:"sourceId"`
	MediaType string `json:"mediaType"`
	Title     string `json:"title"`
}

type SourceInfo struct {
	URI         string         `json:"uri,omitempty"`
	ContentType string         `json:"contentType,omitempty"`
	Data        string         `json:"data,omitempty"`
	Metadata    map[string]any `json:"metadata,omitempty"`
}

type SourceUrlPart struct {
	SourceID string `json:"sourceId"`
	URL      string `json:"url"`
}

type StartStepStreamPart struct {
}

type StreamProcessor func(DataStreamPart)

type StreamTransformer func(DataStreamPart) DataStreamPart

type TextDeltaPart struct {
	ID    string `json:"id"`
	Delta string `json:"delta"`
}

type TextEndPart struct {
	ID string `json:"id"`
}

type TextStartPart struct {
	ID string `json:"id"`
}

type Tool struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Schema      Schema `json:"schema"`
}

type ToolCall struct {
	ID   string         `json:"id"`
	Name string         `json:"name"`
	Args map[string]any `json:"args"`
}

type ToolCallResult interface {
	Part | []Part | any
}

type ToolInputAvailablePart struct {
	ToolCallID       string           `json:"toolCallId"`
	ToolName         string           `json:"toolName"`
	Input            map[string]any   `json:"input"`
	ProviderMetadata ProviderMetadata `json:"providerMetadata,omitzero"`
}

type ToolInputDeltaPart struct {
	ToolCallID     string `json:"toolCallId"`
	InputTextDelta string `json:"inputTextDelta"`
}

type ToolInputStartPart struct {
	ToolCallID string `json:"toolCallId"`
	ToolName   string `json:"toolName"`
}

type ToolInvocationState string

type ToolOutputAvailablePart struct {
	ToolCallID string `json:"toolCallId"`
	Output     any    `json:"output"`
}

type Usage struct {
	PromptTokens     *int64 `json:"promptTokens"`
	CompletionTokens *int64 `json:"completionTokens"`
}