All files / kernel-language-model-service/src/ollama nodejs.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13x
13x
2x
 
11x
11x
 
3x
 
 
 
 
 
 
 
import { Ollama } from 'ollama';
 
import { OllamaBaseService } from './base.ts';
import { defaultClientConfig } from './constants.ts';
import type { OllamaClient, OllamaNodejsConfig } from './types.ts';
 
/**
 * Node.js-specific implementation of the Ollama service.
 * Extends OllamaBaseService to provide a concrete implementation for Node.js environments.
 * Requires an explicit fetch endowment.
 */
export class OllamaNodejsService extends OllamaBaseService<OllamaClient> {
  /**
   * Creates a new Ollama Node.js service.
   *
   * @param config - The configuration for the service
   * @param config.endowments - Required endowments for the service
   * @param config.endowments.fetch - The fetch implementation to use for HTTP requests
   * @param config.clientConfig - Optional configuration for the Ollama client
   * @throws {Error} When fetch is not provided in endowments
   */
  constructor(config: OllamaNodejsConfig) {
    const { endowments, clientConfig = {} } = config;
    if (!endowments?.fetch) {
      throw new Error('Must endow a fetch implementation.');
    }
    const resolvedConfig = { ...defaultClientConfig, ...clientConfig };
    super(
      async () =>
        new Ollama({
          ...resolvedConfig,
          fetch: endowments.fetch,
        }) as OllamaClient,
    );
  }
}