aiutil

type Assistant struct {
	ID            string `json:"id"`
	CreatedAt     int    `json:"created_at"`
	AssistantBase `json:",inline"`
}

type AssistantBase struct {
	Model          Model         `json:"model"` // gpt-4o
	Name           string        `json:"name"`
	Description    string        `json:"description"`
	Tools          []Tool        `json:"tools"`
	ToolResources  ToolResources `json:"tool_resources"`
	Instructions   string        `json:"instructions"`
	ResponseFormat interface{}   `json:"response_format,omitempty"`
}

type ChatChoice struct {
	FinishReason string  `json:"finish_reason"`
	Index        int     `json:"index"`
	Logprobs     string  `json:"logprobs"`
	Message      Message `json:"message"`
}

type ChatDefaultResponse struct {
	Choices           []ChatChoice `json:"choices,omitempty"`
	Created           int          `json:"created,omitempty"`
	ID                string       `json:"id,omitempty"`
	Model             string       `json:"model,omitempty"`
	Object            string       `json:"object,omitempty"`
	SystemFingerprint string       `json:"system_fingerprint"`
	Usage             *Usage       `json:"usage,omitempty"`
	Error             *Error       `json:"error,omitempty"`
}

type ChatRequest struct {
	Model            Model                  `json:"model"`
	Messages         []Message              `json:"messages"`
	ResponseFormat   *MessageResponseFormat `json:"response_format,omitempty"`
	FrequencyPenalty float64                `json:"frequency_penalty,omitempty"` // -2 to 2: Low value = more deterministic, high value = more random
	Temperature      float64                `json:"temperature,omitempty"`       // 0-2: Low value = more deterministic, high value = more random
}

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

type DeleteResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Deleted bool   `json:"deleted"`
}

type Error struct {
	Code    string `json:"code,omitempty"`
	Message string `json:"message,omitempty"`
	Param   string `json:"param,omitempty"`
	Type    string `json:"type,omitempty"`
}

type File struct {
	Object       string  `json:"object"`
	ID           string  `json:"id"`
	Purpose      string  `json:"purpose"`
	Filename     string  `json:"filename"`
	Bytes        int     `json:"bytes"`
	CreatedAt    int     `json:"created_at"`
	Status       string  `json:"status"`
	StatusDetail *string `json:"status_details"`
}

type JSONSchema []byte

type Message struct {
	Role    Role   `json:"role"`
	Content string `json:"content"`
	Refusal string `json:"refusal,omitempty"` // Optional: Only in responses, if the assistant refuses to respond.
}

type MessageResponseFormat struct {
	Type       string                          `json:"type,omitempty"`
	JSONSchema MessageResponseFormatJSONSchema `json:"json_schema,omitempty"`
}

type MessageResponseFormatJSONSchema struct {
	Description string     `json:"description,omitempty"`
	Name        string     `json:"name,omitempty"`
	Strict      bool       `json:"strict,omitempty"`
	Schema      JSONSchema `json:"schema,omitempty"`
}

type Model string

type NewAssistantRequest struct {
	AssistantBase `json:",inline"`
}

type NewThreadAndRunRequest struct {
	AssistantID    string                 `json:"assistant_id"`
	Thread         ThreadInput            `json:"thread,omitempty"`         // Optional (if we want to provide initial messages)
	ToolResources  ToolResources          `json:"tool_resources,omitempty"` // Maybe allows us to provide more files to the assistant?
	ResponseFormat *MessageResponseFormat `json:"response_format,omitempty"`
}

type Role string

type Thread struct {
	ID         string `json:"id,omitempty"`
	ThreadBase `json:",inline"`
}

type ThreadBase struct{}

type ThreadInput struct {
	Messages   []ThreadMessageInput `json:"messages,omitempty"` // only for initial messages, in the Thread request
	ThreadBase `json:",inline"`
}

type ThreadMessage struct {
	ThreadMessageBase `json:",inline"`
	Content           []ThreadMessageContent `json:"content"`
	ID                string                 `json:"id"`
	CreatedAt         int                    `json:"created_at"`
	AssistantID       string                 `json:"assistant_id"`
	ThreadID          string                 `json:"thread_id"`
	RunID             string                 `json:"run_id"`
}

type ThreadMessageBase struct {
	Role Role `json:"role"`
}

type ThreadMessageContent struct {
	Type string                   `json:"type"`
	Text ThreadMessageContentText `json:"text"`
}

type ThreadMessageContentText struct {
	Value       string        `json:"value"`
	Annotations []interface{} `json:"annotations"`
}

type ThreadMessageInput struct {
	ThreadMessageBase `json:",inline"`
	Content           string `json:"content"`
}

type ThreadMessagesResponse struct {
	Object string          `json:"object"`
	Data   []ThreadMessage `json:"data"`
}

type ThreadRun struct {
	ID                  string   `json:"id"`
	Object              string   `json:"object"` // Always "thread.run"
	CreatedAt           int      `json:"created_at"`
	AssistantID         string   `json:"assistant_id"`
	ThreadID            string   `json:"thread_id"`
	Status              string   `json:"status"`
	StartedAt           int      `json:"started_at"`
	ExpiresAt           int      `json:"expires_at"`
	CancelledAt         int      `json:"cancelled_at"`
	FailedAt            int      `json:"failed_at"`
	CompletedAt         int      `json:"completed_at"`
	LastError           *Error   `json:"last_error,omitempty"`
	Model               Model    `json:"model"`
	Instructions        string   `json:"instructions"`
	Tools               []Tool   `json:"tools"`
	Metadata            struct{} `json:"metadata"`
	IncompleteDetails   string   `json:"incomplete_details"`
	Usage               Usage    `json:"usage"`
	Temperature         float64  `json:"temperature"`
	TopP                float64  `json:"top_p"`
	MaxPromptTokens     int      `json:"max_prompt_tokens"`
	MaxCompletionTokens int      `json:"max_completion_tokens"`
	TruncationStrategy  struct {
		Type         string `json:"type"`
		LastMessages string `json:"last_messages"`
	} `json:"truncation_strategy"`
	ResponseFormat    interface{} `json:"response_format"`
	ToolChoice        string      `json:"tool_choice"`
	ParallelToolCalls bool        `json:"parallel_tool_calls"`
}

type ThreadRunInput struct {
	AssistantID        string                 `json:"assistant_id"`
	Model              Model                  `json:"model"` // If provided, will override the assistant's model
	AdditionalMessages []ThreadMessageInput   `json:"additional_messages,omitempty"`
	ResponseFormat     *MessageResponseFormat `json:"response_format,omitempty"`
}

type Tool struct {
	Type string `json:"type"` // code_interpreter, file_search
}

type ToolResources struct {
	CodeInterpreter ToolResourcesCodeInterpreter `json:"code_interpreter"`
	FileSearch      ToolResourcesFileSearch      `json:"file_search,omitempty"`
}

type ToolResourcesCodeInterpreter struct {
	FileIDs []string `json:"file_ids,omitempty"`
}

type ToolResourcesFileSearch struct {
	VectorStores   []ToolResourcesFileSearchVectorStore `json:"vector_stores,omitempty"`
	VectorStoreIDs []string                             `json:"vector_store_ids,omitempty"`
}

type ToolResourcesFileSearchVectorStore struct {
	FileIDs []string `json:"file_ids,omitempty"`
}

type Usage struct {
	CompletionTokens       int `json:"completion_tokens"`
	CompletionTokensDetail struct {
		ReasoningTokens int `json:"reasoning_tokens"`
	} `json:"completion_tokens_details"`
	PromptTokens       int `json:"prompt_tokens"`
	PromptTokensDetail struct {
		CachedTokens int `json:"cached_tokens"`
	} `json:"prompt_tokens_details"`
	TotalTokens int `json:"total_tokens"`
}