openai_assistant

Documentation

Overview

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: MIT

Index

• Constants
• type FileRegistry
  • func (r *FileRegistry) AddFile(filePath, fileID string, fileSize int64)
  • func (r *FileRegistry) GetFileID(filePath string) (string, bool)
• type OpenAIAssistantClient
  • func NewOpenAIAssistantClient(apiKey, model string, logger *zap.Logger) (*OpenAIAssistantClient, error)
  • func (c *OpenAIAssistantClient) Cleanup() error
  • func (c *OpenAIAssistantClient) GetModelName() string
  • func (c *OpenAIAssistantClient) SendPrompt(ctx context.Context, prompt string, history []models.Message, maxTokens int) (string, error)

Constants

const (
	MaxAssistantFiles       = 20
	MaxFileSizeBytes        = 512 * 1024 * 1024 // 512MB
	AssistantAPIBaseURL     = "https://api.openai.com/v1"
	DefaultPollingInterval  = 1 * time.Second
	MaxPollingInterval      = 5 * time.Second
	DefaultPollingTimeout   = 5 * time.Minute
	DefaultAssistantModel   = "gpt-4o"
	DefaultAssistantName    = "ChatCLI Assistant"
	DefaultAssistantTimeout = 10 * time.Minute
)

Types

type FileRegistry

type FileRegistry struct {
	Files     map[string]string // Mapeia caminhos locais para IDs de arquivo na OpenAI
	TotalSize int64             // Tamanho total dos arquivos carregados
	// contains filtered or unexported fields
}

FileRegistry gerencia o cache de arquivos já enviados para a OpenAI

func (*FileRegistry) AddFile

func (r *FileRegistry) AddFile(filePath, fileID string, fileSize int64)

AddFile adiciona um arquivo ao registro se ele já não existir

func (*FileRegistry) GetFileID

func (r *FileRegistry) GetFileID(filePath string) (string, bool)

GetFileID retorna o ID do arquivo se estiver no registro

type OpenAIAssistantClient

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

OpenAIAssistantClient implementa a interface LLMClient usando a API de Assistentes da OpenAI

func NewOpenAIAssistantClient

func NewOpenAIAssistantClient(apiKey, model string, logger *zap.Logger) (*OpenAIAssistantClient, error)

NewOpenAIAssistantClient cria uma nova instância de OpenAIAssistantClient

func (*OpenAIAssistantClient) Cleanup

func (c *OpenAIAssistantClient) Cleanup() error

Método para limpar threads ao fim da app

func (*OpenAIAssistantClient) GetModelName

func (c *OpenAIAssistantClient) GetModelName() string

GetModelName retorna o nome do modelo utilizado

func (*OpenAIAssistantClient) SendPrompt

func (c *OpenAIAssistantClient) SendPrompt(ctx context.Context, prompt string, history []models.Message, maxTokens int) (string, error)

SendPrompt envia uma mensagem para o thread atual e retorna a resposta