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

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

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 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47  1x 1x 1x             1x 1x       1x   1x 1x 1x     1x     1x   1x   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 { AIProvider } from '@shared/types/constants';
import {
  createModelResponseParams,
  openAICreateModelResponseTransformer,
  openAIDeleteModelResponseTransformer,
  openAIGetModelResponseTransformer,
  openAIListInputItemsResponseTransformer,
} from '../open-ai-base';
import { xaiAPIConfig } from './api';
import {
  xaiChatCompleteConfig,
  xaiChatCompleteResponseTransform,
} from './chat-complete';
import { xaiModelCapabilities } from './model-capabilities';
 
export const xaiConfig: AIProviderConfig = {
  api: xaiAPIConfig,
  modelCapabilities: xaiModelCapabilities,
 
  // Chat Completions API
  [FunctionName.CHAT_COMPLETE]: xaiChatCompleteConfig,
 
  // Responses API (using OpenAI base implementation)
  [FunctionName.CREATE_MODEL_RESPONSE]: createModelResponseParams([]),
 
  responseTransforms: {
    // Chat Completions API
    [FunctionName.CHAT_COMPLETE]: xaiChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]: xaiChatCompleteResponseTransform,
 
    // Responses API
    [FunctionName.CREATE_MODEL_RESPONSE]: openAICreateModelResponseTransformer(
      AIProvider.XAI,
    ),
    [FunctionName.GET_MODEL_RESPONSE]: openAIGetModelResponseTransformer(
      AIProvider.XAI,
    ),
    [FunctionName.DELETE_MODEL_RESPONSE]: openAIDeleteModelResponseTransformer(
      AIProvider.XAI,
    ),
    [FunctionName.LIST_RESPONSE_INPUT_ITEMS]:
      openAIListInputItemsResponseTransformer(AIProvider.XAI),
  },
};