# All files / src/providers/ollama Embedding.ts

0% Statements 0/3
0% Branches 0/2
0% Functions 0/2
0% Lines 0/3

Press n or j to go to the next uncovered block, b, p or k for the previous block.

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15

     import { OpenAIEmbedding } from "../openai/Embedding.js";
    import { OllamaCapabilities } from "./Capabilities.js";

    export class OllamaEmbedding extends OpenAIEmbedding {
      protected override getProviderName(): string {
        return "ollama";
      }

      protected override validateModel(model: string): void {
        if (!OllamaCapabilities.supportsEmbeddings(model)) {
          throw new Error(`Model ${model} does not support embeddings.`);
        }
      }
    }