Function callGoogle

packages/baseai/src/dev/llms/call-google.ts:11–51

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

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

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

		// Transform params according to provider's format
		const transformedRequestParams = transformToProviderRequest({
			provider: GOOGLE,
			params: modelParams,
			fn: 'chatComplete'
		});

		applyJsonModeIfEnabledForGoogle(transformedRequestParams, pipe);

		dlog('Google request params', transformedRequestParams);

		const providerOptions = { provider: GOOGLE, llmApiKey };

		return await handleProviderRequest({
			providerOptions,
			inputParams: modelParams,
			endpoint: 'chatComplete',
			transformedRequestParams
		});
	} catch (error: any) {
		improveGoogleErrorMessage(error);
		handleLlmError({ error, provider: GOOGLE });
	}
}

function buildModelParams(
	pipe: Pipe,