mirror of
https://github.com/geoffsee/open-gsio.git
synced 2025-09-08 22:56:46 +00:00
Update model path handling logic for FireworksAI and refine supported model filtering.
This commit is contained in:
@@ -15,10 +15,21 @@ export class FireworksAiChatProvider extends BaseChatProvider {
|
|||||||
let modelPrefix = 'accounts/fireworks/models/';
|
let modelPrefix = 'accounts/fireworks/models/';
|
||||||
if (param.model.toLowerCase().includes('yi-')) {
|
if (param.model.toLowerCase().includes('yi-')) {
|
||||||
modelPrefix = 'accounts/yi-01-ai/models/';
|
modelPrefix = 'accounts/yi-01-ai/models/';
|
||||||
|
} else if (param.model.toLowerCase().includes('/perplexity/')) {
|
||||||
|
modelPrefix = 'accounts/perplexity/models/';
|
||||||
|
} else if (param.model.toLowerCase().includes('/sentientfoundation/')) {
|
||||||
|
modelPrefix = 'accounts/sentientfoundation/models/';
|
||||||
|
} else if (param.model.toLowerCase().includes('/sentientfoundation-serverless/')) {
|
||||||
|
modelPrefix = 'accounts/sentientfoundation-serverless/models/';
|
||||||
|
} else if (param.model.toLowerCase().includes('/instacart/')) {
|
||||||
|
modelPrefix = 'accounts/instacart/models/';
|
||||||
}
|
}
|
||||||
|
const finalModelIdentifier = param.model.includes(modelPrefix)
|
||||||
|
? param.model
|
||||||
|
: `${modelPrefix}${param.model}`;
|
||||||
|
console.log('using fireworks model', finalModelIdentifier);
|
||||||
return {
|
return {
|
||||||
model: `${modelPrefix}${param.model}`,
|
model: finalModelIdentifier,
|
||||||
messages: safeMessages,
|
messages: safeMessages,
|
||||||
stream: true,
|
stream: true,
|
||||||
};
|
};
|
||||||
|
@@ -168,12 +168,19 @@ const ChatService = types
|
|||||||
|
|
||||||
providerModels.set(
|
providerModels.set(
|
||||||
provider.name,
|
provider.name,
|
||||||
models.filter(
|
models.filter((mdl: any) => {
|
||||||
(mdl: any) =>
|
if ('supports_chat' in mdl && mdl.supports_chat) {
|
||||||
|
return true;
|
||||||
|
} else if ('supports_chat' in mdl && !mdl.supports_chat) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
!mdl.id.includes('whisper') &&
|
!mdl.id.includes('whisper') &&
|
||||||
!mdl.id.includes('tts') &&
|
!mdl.id.includes('tts') &&
|
||||||
!mdl.id.includes('guard'),
|
!mdl.id.includes('guard')
|
||||||
),
|
);
|
||||||
|
}),
|
||||||
);
|
);
|
||||||
|
|
||||||
// 2‑b. Retrieve metadata
|
// 2‑b. Retrieve metadata
|
||||||
@@ -318,7 +325,8 @@ const ChatService = types
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
if (message.includes('404')) {
|
if (message.includes('404')) {
|
||||||
throw new ClientError(`Something went wrong, try again.`, 413, {});
|
console.log(message);
|
||||||
|
throw new ClientError(`Something went wrong, try again.`, 404, {});
|
||||||
}
|
}
|
||||||
throw error;
|
throw error;
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user