summaryrefslogtreecommitdiff
path: root/internal/llm/ollama.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/llm/ollama.go')
-rw-r--r--internal/llm/ollama.go6
1 files changed, 6 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"`