UNPKG

@directus/api

Version:

Directus is a real-time API and App dashboard for managing SQL database content

17 lines (16 loc) 831 B
import type { ProviderType } from '@directus/ai'; import { type LanguageModelUsage, type StreamTextResult, type Tool, type UIMessage } from 'ai'; import { type AISettings } from '../../providers/index.js'; import type { ChatContext } from '../models/chat-request.js'; export interface CreateUiStreamOptions { provider: ProviderType; model: string; tools: { [x: string]: Tool; }; aiSettings: AISettings; systemPrompt?: string; context?: ChatContext; onUsage?: (usage: Pick<LanguageModelUsage, 'inputTokens' | 'outputTokens' | 'totalTokens'>) => void | Promise<void>; } export declare const createUiStream: (messages: UIMessage[], { provider, model, tools, aiSettings, systemPrompt, context, onUsage }: CreateUiStreamOptions) => Promise<StreamTextResult<Record<string, Tool<any, any>>, any>>;