All files / api/src/ai-providers/ollama index.ts

100% Statements 15/15
100% Branches 0/0
100% Functions 0/0
100% Lines 15/15

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 16 17 18 19 20 21 22  1x 1x 1x         1x   1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x  
import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import { ollamaAPIConfig } from './api';
import {
  ollamaChatCompleteConfig,
  ollamaChatCompleteResponseTransform,
  ollamaChatCompleteStreamChunkTransform,
} from './chat-complete';
import { ollamaEmbedConfig, ollamaEmbedResponseTransform } from './embed';
 
export const ollamaConfig: AIProviderConfig = {
  api: ollamaAPIConfig,
  [FunctionName.CHAT_COMPLETE]: ollamaChatCompleteConfig,
  [FunctionName.STREAM_CHAT_COMPLETE]: ollamaChatCompleteConfig,
  [FunctionName.EMBED]: ollamaEmbedConfig,
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: ollamaChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]: ollamaChatCompleteStreamChunkTransform,
    [FunctionName.EMBED]: ollamaEmbedResponseTransform,
  },
};