Skip to content

Commit 5b84a31

Browse files
πŸ› FIX: Fireworks models (#32)
* πŸ› FIX: Fireworks models * πŸ› FIX: Fireworks models
1 parent 9381623 commit 5b84a31

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

β€Žpackages/baseai/src/dev/llms/call-fireworks.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ export async function callFireworks({
3030
dlog('Fireworks request params', transformedRequestParams);
3131

3232
// Fireworks llama-3.1 405b behaves weirdly with stop value. Bug on their side. Omitting it.
33-
if (pipe.config.model.name === 'llama-v3p1-405b-instruct')
33+
if (pipe.model.name === 'llama-v3p1-405b-instruct')
3434
delete transformedRequestParams['stop'];
3535

3636
const providerOptions = { provider: FIREWORKS_AI, llmApiKey };
@@ -53,9 +53,9 @@ function buildModelParams(
5353
): ModelParams {
5454
// Create model strings for Fireworks AI
5555
const modelString =
56-
pipe.config.model.name === 'yi-large'
56+
pipe.model.name === 'yi-large'
5757
? 'accounts/yi-01-ai/models/yi-large'
58-
: `accounts/fireworks/models/${pipe.config.model.name}`;
58+
: `accounts/fireworks/models/${pipe.model.name}`;
5959
return {
6060
messages,
6161
stream,

0 commit comments

Comments
Β (0)