Function callXAI

packages/baseai/src/dev/llms/call-xai.ts:10–40

({
	pipe,
	stream,
	llmApiKey,
	messages,
	paramsTools
}: {
	pipe: Pipe;
	stream: boolean;
	llmApiKey: string;
	messages: Message[];
	paramsTools: PipeTool[] | undefined;
})

import type { PipeTool } from 'types/tools';

export async function callXAI({
	pipe,
	stream,
	llmApiKey,
	messages,
	paramsTools
}: {
	pipe: Pipe;
	stream: boolean;
	llmApiKey: string;
	messages: Message[];
	paramsTools: PipeTool[] | undefined;
}) {
	try {
		const modelParams = buildModelParams(pipe, stream, messages);

		// LLM.
		const groq = new OpenAI({
			apiKey: llmApiKey,
			baseURL: 'https://api.x.ai/v1'
		});

		// Add tools (functions) to modelParams
		addToolsToParams(modelParams, pipe, paramsTools);
		dlog('modelParams', modelParams);

		return await groq.chat.completions.create(modelParams as any);
	} catch (error: any) {
		handleLlmError({ error, provider: X_AI });
	}
}

function buildModelParams(
	pipe: Pipe,