mirror of
https://github.com/Litlyx/litlyx
synced 2025-12-10 07:48:37 +01:00
239 lines
9.3 KiB
TypeScript
239 lines
9.3 KiB
TypeScript
|
|
|
|
import OpenAI from "openai";
|
|
import { AiChatModel } from '@schema/ai/AiChatSchema';
|
|
import { ProjectCountModel } from '@schema/project/ProjectsCounts';
|
|
import { ProjectLimitModel } from '@schema/project/ProjectsLimits';
|
|
|
|
import { AiEventsInstance } from '../ai/functions/AI_Events';
|
|
import { AiVisitsInstance } from '../ai/functions/AI_Visits';
|
|
import { AiComposableChartInstance } from '../ai/functions/AI_ComposableChart';
|
|
|
|
const { AI_KEY, AI_ORG, AI_PROJECT } = useRuntimeConfig();
|
|
|
|
const OPENAI_MODEL: OpenAI.Chat.ChatModel = 'gpt-4o-mini';
|
|
|
|
const openai = new OpenAI({ apiKey: AI_KEY, organization: AI_ORG, project: AI_PROJECT });
|
|
|
|
const tools: OpenAI.Chat.Completions.ChatCompletionTool[] = [
|
|
...AiVisitsInstance.getTools(),
|
|
...AiEventsInstance.getTools(),
|
|
...AiComposableChartInstance.getTools()
|
|
]
|
|
|
|
|
|
const functions: any = {
|
|
...AiVisitsInstance.getHandlers(),
|
|
...AiEventsInstance.getHandlers(),
|
|
...AiComposableChartInstance.getHandlers()
|
|
}
|
|
|
|
|
|
async function getMessagesFromChatId(chat_id?: string) {
|
|
if (!chat_id) return;
|
|
const chatItem = await AiChatModel.findById(chat_id);
|
|
if (!chatItem) return;
|
|
return chatItem.messages;
|
|
}
|
|
|
|
async function addMessageToChat(message: any, chat_id?: string) {
|
|
if (!chat_id) return;
|
|
await AiChatModel.updateOne({ _id: chat_id }, { $push: { messages: message } });
|
|
}
|
|
|
|
async function createChatIfNotExist(pid: string, chat_id?: string) {
|
|
const chatItem = await AiChatModel.exists({ _id: chat_id });
|
|
if (chatItem) return chatItem._id.toString();
|
|
const newChatItem = await AiChatModel.create({ messages: [], project_id: pid, title: 'new chat' });
|
|
return newChatItem._id.toString();
|
|
}
|
|
|
|
async function setChatTitle(title: string, chat_id?: string) {
|
|
if (!chat_id) return;
|
|
await AiChatModel.updateOne({ _id: chat_id }, { title });
|
|
}
|
|
|
|
export async function updateChatStatus(chat_id: string, status: string, completed: boolean) {
|
|
await AiChatModel.updateOne({ _id: chat_id }, {
|
|
status,
|
|
completed
|
|
});
|
|
}
|
|
|
|
|
|
export function getChartsInMessage(message: OpenAI.Chat.Completions.ChatCompletionMessageParam) {
|
|
if (message.role != 'assistant') return [];
|
|
if (!message.tool_calls) return [];
|
|
if (message.tool_calls.length == 0) return [];
|
|
return message.tool_calls.filter(e => e.function.name === 'createComposableChart').map(e => e.function.arguments);
|
|
}
|
|
|
|
|
|
|
|
type FunctionCall = { name: string, argsRaw: string[], collecting: boolean, result: any, tool_call_id: string }
|
|
|
|
type DeltaCallback = (text: string) => any;
|
|
type FinishCallback = (functionsCount: number) => any;
|
|
type FunctionNameCallback = (name: string) => any;
|
|
type FunctionCallCallback = (name: string) => any;
|
|
type FunctionResultCallback = (name: string, result: any) => any;
|
|
|
|
type ElaborateResponseCallbacks = {
|
|
onDelta?: DeltaCallback,
|
|
onFunctionName?: FunctionNameCallback,
|
|
onFunctionCall?: FunctionCallCallback,
|
|
onFunctionResult?: FunctionResultCallback,
|
|
onFinish?: FinishCallback,
|
|
onChatId?: (chat_id: string) => any
|
|
}
|
|
|
|
async function elaborateResponse(messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[], pid: string, chat_id: string, callbacks?: ElaborateResponseCallbacks) {
|
|
|
|
const responseStream = await openai.beta.chat.completions.stream({ model: OPENAI_MODEL, messages, n: 1, tools });
|
|
|
|
const functionCalls: FunctionCall[] = [];
|
|
|
|
let lastFinishReason: "length" | "tool_calls" | "function_call" | "stop" | "content_filter" | null = null;
|
|
|
|
for await (const part of responseStream) {
|
|
|
|
const delta = part.choices[0].delta;
|
|
const finishReason = part.choices[0].finish_reason;
|
|
|
|
if (delta.content) await callbacks?.onDelta?.(delta.content);
|
|
|
|
if (delta.tool_calls) {
|
|
const toolCall = delta.tool_calls[0];
|
|
if (!toolCall.function) throw Error('Cannot get function from tool_calls');
|
|
|
|
const functionName = toolCall.function.name;
|
|
|
|
const functionCall: FunctionCall = functionName ?
|
|
{ name: functionName, argsRaw: [], collecting: true, result: null, tool_call_id: toolCall.id as string } :
|
|
functionCalls.at(-1) as FunctionCall;
|
|
|
|
if (functionName) functionCalls.push(functionCall);
|
|
|
|
if (functionName) await callbacks?.onFunctionName?.(functionName);
|
|
|
|
if (toolCall.function.arguments) functionCall.argsRaw.push(toolCall.function.arguments);
|
|
|
|
}
|
|
|
|
if (finishReason === "tool_calls" && functionCalls.at(-1)?.collecting) {
|
|
const functionCall: FunctionCall = functionCalls.at(-1) as FunctionCall;
|
|
await callbacks?.onFunctionCall?.(functionCall.name);
|
|
const args = JSON.parse(functionCall.argsRaw.join(''));
|
|
const functionResult = await functions[functionCall.name]({ project_id: pid, ...args });
|
|
functionCall.result = functionResult;
|
|
await callbacks?.onFunctionResult?.(functionCall.name, functionResult);
|
|
|
|
addMessageToChat({
|
|
role: 'assistant',
|
|
content: delta.content,
|
|
refusal: delta.refusal,
|
|
tool_calls: [
|
|
{
|
|
id: functionCall.tool_call_id,
|
|
type: 'function',
|
|
function: {
|
|
name: functionCall.name,
|
|
arguments: functionCall.argsRaw.join('')
|
|
}
|
|
}
|
|
],
|
|
parsed: null
|
|
}, chat_id);
|
|
|
|
addMessageToChat({
|
|
tool_call_id: functionCall.tool_call_id,
|
|
role: 'tool',
|
|
content: JSON.stringify(functionResult)
|
|
}, chat_id);
|
|
|
|
|
|
functionCall.collecting = false;
|
|
lastFinishReason = finishReason;
|
|
}
|
|
|
|
}
|
|
|
|
await callbacks?.onFinish?.(functionCalls.length);
|
|
|
|
const toolResponseMesages: OpenAI.Chat.Completions.ChatCompletionMessageParam[] = functionCalls.map(e => {
|
|
return { tool_call_id: e.tool_call_id, role: "tool", content: JSON.stringify(e.result) }
|
|
});
|
|
|
|
if (lastFinishReason == 'tool_calls') return await elaborateResponse([...responseStream.messages, ...toolResponseMesages], pid, chat_id, callbacks);
|
|
|
|
return responseStream;
|
|
}
|
|
|
|
|
|
export async function sendMessageOnChat(text: string, pid: string, initial_chat_id?: string, callbacks?: ElaborateResponseCallbacks) {
|
|
|
|
|
|
const messages: OpenAI.Chat.Completions.ChatCompletionMessageParam[] = []
|
|
const chat_id = await createChatIfNotExist(pid, initial_chat_id);
|
|
const chatMessages = await getMessagesFromChatId(chat_id);
|
|
|
|
await callbacks?.onChatId?.(chat_id);
|
|
|
|
if (chatMessages && chatMessages.length > 0) {
|
|
messages.push(...chatMessages);
|
|
} else {
|
|
const roleMessage: OpenAI.Chat.Completions.ChatCompletionMessageParam = {
|
|
role: 'system',
|
|
content: "Today ISO date: " + new Date().toISOString()
|
|
}
|
|
messages.push(roleMessage);
|
|
await addMessageToChat(roleMessage, chat_id);
|
|
|
|
await setChatTitle(text.substring(0, 110), chat_id);
|
|
}
|
|
|
|
const userMessage: OpenAI.Chat.Completions.ChatCompletionMessageParam = { role: 'user', content: text }
|
|
messages.push(userMessage);
|
|
await addMessageToChat(userMessage, chat_id);
|
|
|
|
try {
|
|
const streamResponse = await elaborateResponse(messages, pid, chat_id, callbacks);
|
|
await addMessageToChat({ role: 'assistant', refusal: null, content: await streamResponse.finalContent() }, chat_id);
|
|
return { content: '', charts: [] };
|
|
} catch (ex: any) {
|
|
console.error(ex);
|
|
return { content: ex.message, charts: [] };
|
|
}
|
|
|
|
// let response = await openai.chat.completions.create({ model: OPENAI_MODEL, messages, n: 1, tools });
|
|
|
|
// const chartsData: string[][] = [];
|
|
|
|
// while ((response.choices[0].message.tool_calls?.length || 0) > 0) {
|
|
// await addMessageToChat(response.choices[0].message, chat_id);
|
|
// messages.push(response.choices[0].message);
|
|
// if (response.choices[0].message.tool_calls) {
|
|
|
|
// console.log('Tools to call', response.choices[0].message.tool_calls.length);
|
|
// chartsData.push(getChartsInMessage(response.choices[0].message));
|
|
|
|
// for (const toolCall of response.choices[0].message.tool_calls) {
|
|
// const functionName = toolCall.function.name;
|
|
// console.log('Calling tool function', functionName);
|
|
// const functionToCall = functions[functionName];
|
|
// const functionArgs = JSON.parse(toolCall.function.arguments);
|
|
// const functionResponse = await functionToCall({ project_id: pid, ...functionArgs });
|
|
// messages.push({ tool_call_id: toolCall.id, role: "tool", content: JSON.stringify(functionResponse) });
|
|
// await addMessageToChat({ tool_call_id: toolCall.id, role: "tool", content: JSON.stringify(functionResponse) }, chat_id);
|
|
// }
|
|
// }
|
|
// response = await openai.chat.completions.create({ model: OPENAI_MODEL, messages, n: 1, tools });
|
|
// }
|
|
// await addMessageToChat(response.choices[0].message, chat_id);
|
|
// await ProjectLimitModel.updateOne({ project_id: pid }, { $inc: { ai_messages: 1 } })
|
|
// return { content: response.choices[0].message.content, charts: chartsData.filter(e => e.length > 0).flat() };
|
|
|
|
|
|
}
|
|
|