diff --git a/packages/core/src/llm/ollama.ts b/packages/core/src/llm/ollama.ts index 4ddcdf91105a20c4e7982594815fa6e0c7e81bc2..78f6889aca19bab7f163e050bc6be02e4772fa8d 100644 --- a/packages/core/src/llm/ollama.ts +++ b/packages/core/src/llm/ollama.ts @@ -37,14 +37,18 @@ export class Ollama extends BaseEmbedding implements LLM { additionalChatOptions?: Record<string, unknown>; callbackManager?: CallbackManager; + protected modelMetadata: Partial<LLMMetadata>; + constructor( init: Partial<Ollama> & { // model is required model: string; + modelMetadata?: Partial<LLMMetadata>; }, ) { super(); this.model = init.model; + this.modelMetadata = init.modelMetadata ?? {}; Object.assign(this, init); } @@ -56,6 +60,7 @@ export class Ollama extends BaseEmbedding implements LLM { maxTokens: undefined, contextWindow: this.contextWindow, tokenizer: undefined, + ...this.modelMetadata, }; }