@aj-archipelago/cortex
Version:
Cortex is a GraphQL API for AI. It provides a simple, extensible interface for using AI services from OpenAI, Azure and others.
35 lines (28 loc) • 954 B
text/typescript
import {type ChatMessage, type CortexVariables, getCortexResponse} from "./utils";
export type MultiMessage = {
role: string;
content: string | string[];
}
const VISION_QUERY = `
query Vision($text: String, $contextId: String, $chatHistory: [MultiMessage], $aiName: String) {
sys_generator_video_vision(text: $text, contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName) {
result
tool
errors
warnings
}
}
`
export async function vision(contextId: string,
aiName: string,
chatHistory: (ChatMessage | MultiMessage)[],
text: string) {
const variables: Omit<CortexVariables, 'chatHistory'> & { chatHistory: (ChatMessage | MultiMessage)[] } = {
chatHistory,
contextId,
aiName,
text
}
const res = await getCortexResponse(variables as CortexVariables, VISION_QUERY);
return res.sys_generator_video_vision;
}