provider.go

package connector

import (
	"context"
	"fmt"
)

// LLMProvider implements a specific AI provider's chat completion API.
type LLMProvider interface {
	ChatCompletion(ctx context.Context, req *ChatRequest) (*ChatResponse, error)
}

// EmbeddingProvider implements a specific provider's text-embeddings API.
type EmbeddingProvider interface {
	Embeddings(ctx context.Context, req *EmbeddingRequest) (*EmbeddingResponse, error)
}

// EmbeddingRequest is the provider-agnostic embeddings request format.
type EmbeddingRequest struct {
	Model      string
	Inputs     []string
	Dimensions int    // 0 = provider default
	InputType  string // provider-specific hint (Cohere: search_document/search_query/...); "" = provider default
	Credential map[string]string
}

// EmbeddingResponse is the provider-agnostic embeddings response format.
// Embeddings are returned in the same order as EmbeddingRequest.Inputs.
type EmbeddingResponse struct {
	Embeddings [][]float64
	Model      string
	Usage      ChatUsage // PromptTokens / TotalTokens are meaningful for embeddings
}

// ChatRequest is the provider-agnostic request format.
type ChatRequest struct {
	Model        string
	Messages     []ChatMessage
	Tools        []ChatTool
	OutputSchema map[string]any
	MaxTokens    int
	Credential   map[string]string
}

// ChatMessage represents a single message in the conversation.
// Valid Role values: "system", "user", "assistant", "tool".
// For "tool" messages (tool results), set ToolCallID to match the original call.
// Note: "tool" role maps to ConversationRoleUser in Bedrock (tool results are user messages).
type ChatMessage struct {
	Role       string
	Content    string
	ToolCalls  []ToolCall // reuses existing ToolCall type from ai.go
	ToolCallID string
}

// ChatTool defines a tool the model can invoke.
type ChatTool struct {
	Name        string
	Description string
	InputSchema map[string]any
}

// ChatResponse is the provider-agnostic response format.
type ChatResponse struct {
	Text         string
	ToolCalls    []ToolCall
	FinishReason string // "stop" or "tool_calls"
	Usage        ChatUsage
	Model        string
}

// ChatUsage tracks token consumption.
type ChatUsage struct {
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
}

// RetryableError wraps an error to indicate the caller should retry.
type RetryableError struct {
	Err error
}

func (e *RetryableError) Error() string {
	return fmt.Sprintf("retryable: %v", e.Err)
}

func (e *RetryableError) Unwrap() error {
	return e.Err
}