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.`);
    }
  }
}