summaryrefslogtreecommitdiff
path: root/internal/llm
diff options
context:
space:
mode:
Diffstat (limited to 'internal/llm')
-rw-r--r--internal/llm/ollama.go6
-rw-r--r--internal/llm/openai.go6
-rw-r--r--internal/llm/openrouter.go6
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)
}