llm

type BotConfig struct {
	ID                 string             `json:"id"`
	Name               string             `json:"name"`
	DisplayName        string             `json:"displayName"`
	CustomInstructions string             `json:"customInstructions"`
	Service            ServiceConfig      `json:"service"`
	EnableVision       bool               `json:"enableVision"`
	DisableTools       bool               `json:"disableTools"`
	ChannelAccessLevel ChannelAccessLevel `json:"channelAccessLevel"`
	ChannelIDs         []string           `json:"channelIDs"`
	UserAccessLevel    UserAccessLevel    `json:"userAccessLevel"`
	UserIDs            []string           `json:"userIDs"`
	TeamIDs            []string           `json:"teamIDs"`
	MaxFileSize        int64              `json:"maxFileSize"`
}

type ChannelAccessLevel int

type CompletionRequest struct {
	Posts   []Post
	Context *Context
}

type Context struct {
	// Server
	Time        string
	ServerName  string
	CompanyName string

	// Location
	Team    *model.Team
	Channel *model.Channel
	Thread  []Post // Normalized posts that already have been formatted. nil if not in a thread or a root post

	// User that is making the request
	RequestingUser *model.User

	// Bot Specific
	BotName            string
	BotUsername        string
	BotModel           string
	CustomInstructions string

	Tools      *ToolStore
	Parameters map[string]interface{}
}

type ContextOption func(*Context)

type EventType int

type File struct {
	MimeType string
	Size     int64
	Reader   io.Reader
}

type LanguageModel interface {
	ChatCompletion(conversation CompletionRequest, opts ...LanguageModelOption) (*TextStreamResult, error)
	ChatCompletionNoStream(conversation CompletionRequest, opts ...LanguageModelOption) (string, error)

	CountTokens(text string) int
	InputTokenLimit() int
}

type LanguageModelConfig struct {
	Model              string
	MaxGeneratedTokens int
	EnableVision       bool
	JSONOutputFormat   any
}

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

type LanguageModelOption func(*LanguageModelConfig)

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

type LanguageModelWrapper func(LanguageModel) LanguageModel

type Post struct {
	Role    PostRole
	Message string
	Files   []File
	ToolUse []ToolCall
}

type PostRole int

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

type ServiceConfig struct {
	Name         string `json:"name"`
	Type         string `json:"type"`
	APIKey       string `json:"apiKey"`
	OrgID        string `json:"orgId"`
	DefaultModel string `json:"defaultModel"`
	APIURL       string `json:"apiURL"`

	// Renaming the JSON field to inputTokenLimit would require a migration, leaving as is for now.
	InputTokenLimit         int  `json:"tokenLimit"`
	StreamingTimeoutSeconds int  `json:"streamingTimeoutSeconds"`
	SendUserID              bool `json:"sendUserID"`

	// Otherwise known as maxTokens
	OutputTokenLimit int `json:"outputTokenLimit"`
}

type TextStreamEvent struct {
	Type  EventType
	Value any
}

type TextStreamResult struct {
	Stream <-chan TextStreamEvent
}

type Tool struct {
	Name        string
	Description string
	Schema      *jsonschema.Schema
	Resolver    ToolResolver
}

type ToolArgumentGetter func(args any) error

type ToolCall struct {
	ID          string          `json:"id"`
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Arguments   json.RawMessage `json:"arguments"`
	Result      string          `json:"result"`
	Status      ToolCallStatus  `json:"status"`
}

type ToolCallStatus int

type ToolResolver func(context *Context, argsGetter ToolArgumentGetter) (string, error)

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

type TraceLog interface {
	Info(message string, keyValuePairs ...any)
}

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

type UserAccessLevel int