48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package ai
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
// Client is AI provider transport (real HTTP or fake).
|
||
type Client interface {
|
||
Complete(ctx context.Context, apiKey, model, prompt string) (string, error)
|
||
// CompleteStream 串流 delta；onDelta 可為 nil。回傳完整正文。
|
||
CompleteStream(ctx context.Context, apiKey, model, prompt string, onDelta func(chunk string) error) (string, error)
|
||
ListModels(ctx context.Context, apiKey string) ([]string, error)
|
||
}
|
||
|
||
// FakeClient for tests / when no real key path needed.
|
||
type FakeClient struct {
|
||
Fail bool
|
||
Models []string
|
||
}
|
||
|
||
func (f *FakeClient) Complete(ctx context.Context, apiKey, model, prompt string) (string, error) {
|
||
if f.Fail || apiKey == "" {
|
||
return "", fmt.Errorf("ai call failed: no key or forced fail")
|
||
}
|
||
_ = SystemPromptForLanguage(ResponseLanguageFrom(ctx)) // ensure always defined for real clients
|
||
if len(prompt) > 200 {
|
||
prompt = prompt[:200]
|
||
}
|
||
return fmt.Sprintf("[%s] %s", model, strings.TrimSpace(prompt)), nil
|
||
}
|
||
|
||
func (f *FakeClient) ListModels(_ context.Context, apiKey string) ([]string, error) {
|
||
if f.Fail || apiKey == "" {
|
||
return nil, fmt.Errorf("list models failed: no key or forced fail")
|
||
}
|
||
if len(f.Models) > 0 {
|
||
return append([]string(nil), f.Models...), nil
|
||
}
|
||
return []string{"grok-3", "grok-2", "grok-beta"}, nil
|
||
}
|
||
|
||
// ModelsForProvider — static catalog (fallback when live list fails).
|
||
func ModelsForProvider(provider string) []string {
|
||
return FallbackModels(provider)
|
||
}
|