diff options
Diffstat (limited to 'internal/llm')
| -rw-r--r-- | internal/llm/ollama.go | 6 | ||||
| -rw-r--r-- | internal/llm/openai.go | 6 | ||||
| -rw-r--r-- | internal/llm/openrouter.go | 6 |
3 files changed, 18 insertions, 0 deletions
diff --git a/internal/llm/ollama.go b/internal/llm/ollama.go index ade62a9..be93ab0 100644 --- a/internal/llm/ollama.go +++ b/internal/llm/ollama.go @@ -23,6 +23,12 @@ type ollamaClient struct { defaultTemperature *float64 } +// Ensure ollamaClient implements Client and Streamer. +var ( + _ Client = ollamaClient{} + _ Streamer = ollamaClient{} +) + type ollamaChatRequest struct { Model string `json:"model"` Messages []oaMessage `json:"messages"` diff --git a/internal/llm/openai.go b/internal/llm/openai.go index eccd558..cf18d9b 100644 --- a/internal/llm/openai.go +++ b/internal/llm/openai.go @@ -24,6 +24,12 @@ type openAIClient struct { defaultTemperature *float64 } +// Ensure openAIClient implements Client and Streamer. +var ( + _ Client = openAIClient{} + _ Streamer = openAIClient{} +) + type oaChatRequest struct { Model string `json:"model"` Messages []oaMessage `json:"messages"` diff --git a/internal/llm/openrouter.go b/internal/llm/openrouter.go index 60a594a..451e9ad 100644 --- a/internal/llm/openrouter.go +++ b/internal/llm/openrouter.go @@ -21,6 +21,12 @@ type openRouterClient struct { defaultTemperature *float64 } +// Ensure openRouterClient implements Client and Streamer. +var ( + _ Client = openRouterClient{} + _ Streamer = openRouterClient{} +) + func init() { RegisterProvider("openrouter", openRouterProviderFactory) } |
