import { schema } from '#triplit/schema';
import type { Entity } from '@triplit/client';

export const Providers = ['openrouter', 'ollama', 'cerebras', 'google', 'longcat', 'cohere'] as const;
export const SupportedModalities = ['text', 'image', 'audio', 'video', 'pdf'] as const;

export const providerBaseUrls = {
    openrouter: 'https://openrouter.ai/api/v1',
    ollama: '',
    cerebras: 'https://api.cerebras.ai/v1',
    google: 'https://generativelanguage.googleapis.com/v1beta',
    longcat: 'https://api.longcat.chat/openai/v1',
    cohere: 'https://api.cohere.ai/v1',
};

export type Model = Entity<typeof schema, 'models'> & { provider: Entity<typeof schema, 'providers'> };