anthrogo

type AnthropicModel string

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

type CompleteOptions struct {
	Metadata      any      `json:"metadata,omitempty"`
	StopSequences []string `json:"stop_sequences,omitempty"`
	Stream        bool     `json:"stream,omitempty"`
	Temperature   float64  `json:"temperature,omitempty"`
	TopK          int      `json:"top_k,omitempty"`
	TopP          float64  `json:"top_p,omitempty"`
}

type CompletionEvent struct {
	Event string
	Data  *CompletionEventData
	ID    string
	Retry int
}

type CompletionEventData struct {
	Completion string `json:"completion"`
	StopReason string `json:"stop_reason"`
	Model      string `json:"model"`
	Stop       string `json:"stop"`
	LogID      string `json:"log_id"`
}

type CompletionMessage struct {
	Role    Role
	Content string
}

type CompletionPayload struct {
	MaxTokensToSample int            `json:"max_tokens_to_sample"`
	Model             AnthropicModel `json:"model"`
	Prompt            string         `json:"prompt"`
	CompleteOptions
}

type CompletionResponse struct {
	Completion string `json:"completion"`
	StopReason string `json:"stop_reason"`
	Model      string `json:"model"`
}

type CompletionSSEDecoder struct {
	Reader *bufio.Reader
	// contains filtered or unexported fields
}

type ContentBlock struct {
	Type string `json:"type"`
	Text string `json:"text,omitempty"`
}

type ContentBlockDelta struct {
	Type  string `json:"type"`
	Index int    `json:"index"`
	Delta struct {
		Type string `json:"type"`
		Text string `json:"text"`
	} `json:"delta"`
}

type ContentBlockStart struct {
	Type         string `json:"type"`
	Index        int    `json:"index"`
	ContentBlock struct {
		Type string `json:"type"`
		Text string `json:"text"`
	} `json:"content_block"`
}

type ContentBlockStop struct {
	Type  string `json:"type"`
	Index int    `json:"index"`
}

type ContentType string

type Conversation struct {
	Messages []CompletionMessage
}

type DecodeOptions struct {
	ContentOnly bool
}

type ErrorData struct {
	Type  string `json:"type"`
	Error struct {
		Type    string `json:"type"`
		Message string `json:"message"`
	} `json:"error"`
}

type ErrorDetail struct {
	Type    string `json:"type"`
	Message string `json:"message"`
}

type ErrorResponse struct {
	Error ErrorDetail `json:"error"`
}

type EventData struct {
	Content string
	Data    any
}

type HttpClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type ImageSource struct {
	Type      string `json:"type"`
	MediaType string `json:"media_type"`
	Data      string `json:"data"`
}

type Message struct {
	Role    RoleType         `json:"role"`
	Content []MessageContent `json:"content"`
}

type MessageContent struct {
	Type  ContentType  `json:"type,omitempty"`
	Text  *string      `json:"text,omitempty"`
	Image *ImageSource `json:"source,omitempty"`
}

type MessageDelta struct {
	Type  string      `json:"type"`
	Delta interface{} `json:"delta"`
	Usage struct {
		OutputTokens int `json:"output_tokens"`
	} `json:"usage"`
}

type MessageEvent struct {
	Message *MessageEventPayload
	Err     *error
}

type MessageEventPayload struct {
	Event string
	Data  EventData
}

type MessagePayload struct {
	// The model to use for the request.
	Model AnthropicModel `json:"model"`
	// The messages to send to the model.
	Messages []Message `json:"messages"`
	// The maximum number of tokens to generate.
	MaxTokens int `json:"max_tokens"`
	// Sequences that will cause the model to stop generating.
	StopSequences []string `json:"stop_sequences,omitempty"`
	// Amount of randomness injected into the response.
	Temperature *float64 `json:"temperature,omitempty"`
	// Nucleus sampling.
	TopP *float64 `json:"top_p,omitempty"`
	// Only sample from the top K options for each subsequent token.
	TopK *int `json:"top_k,omitempty"`
	// An object describing metadata about the request.
	Metadata *Metadata `json:"metadata,omitempty"`
	// System prompt to provide to the model.
	System *string `json:"system,omitempty"`
	// Stream the response using server-sent events.
	Stream *bool `json:"stream,omitempty"`
}

type MessageResponse struct {
	ID           string         `json:"id"`
	Type         string         `json:"type"`
	Role         RoleType       `json:"role"`
	Content      []ContentBlock `json:"content"`
	Model        string         `json:"model"`
	StopReason   string         `json:"stop_reason"`
	StopSequence string         `json:"stop_sequence,omitempty"`
	Usage        Usage          `json:"usage"`
}

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

type MessageStart struct {
	Type    string `json:"type"`
	Message struct {
		ID           string   `json:"id"`
		Type         string   `json:"type"`
		Role         string   `json:"role"`
		Content      []string `json:"content"`
		Model        string   `json:"model"`
		StopReason   string   `json:"stop_reason"`
		StopSequence string   `json:"stop_sequence"`
		Usage        struct {
			InputTokens  int `json:"input_tokens"`
			OutputTokens int `json:"output_tokens"`
		} `json:"usage"`
	} `json:"message"`
}

type MessageStopData struct {
	Type string `json:"type"`
}

type Metadata struct {
	UserID string `json:"user_id,omitempty"`
}

type PingData struct {
	Type string `json:"type"`
}

type Role string

type RoleType string

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

type Usage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}