client

Documentation

Overview

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

Index

• type LLMClient
• type LLMError
  • func (e *LLMError) Error() string
• type MockLLMClient
  • func (m *MockLLMClient) GetModelName() string
  • func (m *MockLLMClient) SendPrompt(ctx context.Context, prompt string, history []models.Message, maxTokens int) (string, error)

Types

type LLMClient

type LLMClient interface {
	// GetModelName retorna o nome do modelo de linguagem utilizado pelo cliente.
	GetModelName() string

	// SendPrompt envia um prompt para o modelo de linguagem e retorna a resposta.
	// O contexto (ctx) pode ser usado para controlar o tempo de execução e cancelamento.
	// O histórico (history) contém as mensagens anteriores da conversa.
	// Retorna uma string com a resposta do modelo e um erro, caso ocorra.
	SendPrompt(ctx context.Context, prompt string, history []models.Message, maxTokens int) (string, error)
}

LLMClient define os métodos que um cliente LLM deve implementar

type LLMError

type LLMError struct {
	Code    int
	Message string
}

LLMError representa um erro personalizado para o cliente LLM

func (*LLMError) Error

func (e *LLMError) Error() string

Error implementa a interface de erro para LLMError

type MockLLMClient

type MockLLMClient struct {
	Response string
	Err      error
}

MockLLMClient é um mock que implementa a interface LLMClient

func (*MockLLMClient) GetModelName

func (m *MockLLMClient) GetModelName() string

func (*MockLLMClient) SendPrompt

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