@@ -21,6 +21,7 @@ type Config struct {
|
||||
RemoveFillerWords bool
|
||||
CustomPrompt string
|
||||
Keywords []string
|
||||
BaseURL string
|
||||
}
|
||||
|
||||
// NewAdapter creates an LLM adapter based on the provider
|
||||
@@ -36,6 +37,14 @@ func NewAdapter(cfg Config) (Adapter, error) {
|
||||
return nil, fmt.Errorf("Groq API key required")
|
||||
}
|
||||
return NewGroqAdapter(cfg), nil
|
||||
case "llama-swap":
|
||||
if cfg.APIKey == "" {
|
||||
return nil, fmt.Errorf("LlamaSwap API key required")
|
||||
}
|
||||
if cfg.BaseURL == "" {
|
||||
return nil, fmt.Errorf("LlamaSwap base_url required")
|
||||
}
|
||||
return NewOpenAIAdapter(cfg), nil
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported LLM provider: %s", cfg.Provider)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user