Function callCohere

packages/baseai/src/dev/llms/call-cohere.ts:9–42

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

import type { Message, Pipe } from 'types/pipe';

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

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

		const providerOptions = { provider: COHERE, llmApiKey };

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

function buildModelParams(
	pipe: Pipe,