feat: implement Provider interface for registry support

sgomez · Jun 29, 2024 · cd73c6b · cd73c6b
1 parent 6925c01
commit cd73c6b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/ollama/src/ollama-provider.ts b/packages/ollama/src/ollama-provider.ts
@@ -23,6 +23,16 @@ export interface OllamaProvider {
     modelId: OllamaEmbeddingModelId,
     settings?: OllamaEmbeddingSettings,
   ): OllamaEmbeddingModel
+
+  languageModel(
+    modelId: OllamaChatModelId,
+    settings?: OllamaChatSettings,
+  ): OllamaChatLanguageModel
+
+  textEmbedding(
+    modelId: OllamaEmbeddingModelId,
+    settings?: OllamaEmbeddingSettings,
+  ): OllamaEmbeddingModel
 }
 
 export interface OllamaProviderSettings {
@@ -92,6 +102,8 @@ export function createOllama(
 
   provider.chat = createChatModel
   provider.embedding = createEmbeddingModel
+  provider.languageModel = createChatModel
+  provider.textEmbedding = createEmbeddingModel
 
   return provider as OllamaProvider
 }