Function buildModelParams

packages/baseai/src/dev/llms/call-fireworks.ts:49–79

(
	pipe: Pipe,
	stream: boolean,
	messages: Message[]
)

}

function buildModelParams(
	pipe: Pipe,
	stream: boolean,
	messages: Message[]
): ModelParams {
	// Create model strings for Fireworks AI
	const pipeModel = pipe.model.split(':')[1];
	const model =
		pipeModel === 'yi-large'
			? 'accounts/yi-01-ai/models/yi-large'
			: `accounts/fireworks/models/${pipeModel}`;
	const {
		top_p,
		max_tokens,
		temperature,
		presence_penalty,
		frequency_penalty,
		stop
	} = pipe;
	return {
		messages,
		stream,
		model,
		top_p,
		max_tokens,
		temperature,
		presence_penalty,
		frequency_penalty,
		stop
	};
}