Merge pull request #1782 from helixml/feature/fireworks_ai

add fireworks provider

Author: nessie993

frontend/assets/img/providers/fireworks.png

@@ -36,6 +36,7 @@ import vllmLogo from '../../../assets/img/vllm-logo.png'
 import helixLogo from '../../../assets/img/logo.png'
 import googleLogo from '../../../assets/img/providers/google.svg'
 import anthropicLogo from '../../../assets/img/providers/anthropic.png'
+import fireworksLogo from '../../../assets/img/providers/fireworks.png'
 import DarkDialog from '../dialog/DarkDialog';
 import useLightTheme from '../../hooks/useLightTheme';
 
@@ -72,6 +73,10 @@ const ProviderIcon: React.FC<{ provider: TypesProviderEndpoint }> = ({ provider
     return <Avatar src={anthropicLogo} sx={{ width: 32, height: 32 }} variant="square" />;
   }
 
+  if (provider.base_url?.startsWith('https://api.fireworks.ai/')) {
+    return <Avatar src={fireworksLogo} sx={{ width: 32, height: 32 }} variant="square" />;
+  }
+
   // Check provider models, if it has more than 1 and "owned_by" = "vllm", then show vllm logo
   if (provider.available_models && provider.available_models.length > 0 && provider.available_models[0].owned_by === "vllm") {
     return <Avatar src={vllmLogo} sx={{ width: 32, height: 32, bgcolor: '#fff' }} variant="square" />;

frontend/src/components/create/AdvancedModelPicker.tsx

@@ -28,6 +28,7 @@ import vllmLogo from '../../../assets/img/vllm-logo.png';
 import helixLogo from '../../../assets/img/logo.png';
 import googleLogo from '../../../assets/img/providers/google.svg';
 import anthropicLogo from '../../../assets/img/providers/anthropic.png';
+import fireworksLogo from '../../../assets/img/providers/fireworks.png';
 import { Bot } from 'lucide-react';
 
 interface ModelWithProvider extends TypesOpenAIModel {
@@ -49,6 +50,9 @@ const getProviderIcon = (provider: TypesProviderEndpoint): React.ReactNode => {
     if (provider.name === 'anthropic' || provider.base_url?.startsWith('https://api.anthropic.com/')) {
         return <Avatar src={anthropicLogo} sx={{ width: 24, height: 24 }} variant="square" />;
     }
+    if (provider.base_url?.startsWith('https://api.fireworks.ai/')) {
+        return <Avatar src={fireworksLogo} sx={{ width: 24, height: 24 }} variant="square" />;
+    }
     if (provider.available_models?.length && provider.available_models[0].owned_by === "vllm") {
         return <Avatar src={vllmLogo} sx={{ width: 24, height: 24, bgcolor: '#fff' }} variant="square" />;
     }

frontend/src/components/dashboard/PricingTable.tsx

@@ -8,6 +8,7 @@ import AWSLogo from './logos/aws';
 // Direct image imports
 import togetheraiLogo from '../../../assets/img/together-logo.png'
 import googleLogo from '../../../assets/img/providers/google.svg';
+import fireworksLogo from '../../../assets/img/providers/fireworks.png';
 import { OllamaIcon } from "../icons/ProviderIcons";
 
 export interface Provider {
@@ -86,10 +87,19 @@ export const PROVIDERS: Provider[] = [
     alias: ['togetherai', 'togetherai-api'],
     name: 'TogetherAI',
     description: 'Integrate with TogetherAI for ultra-fast LLM inference.',
-    logo: togetheraiLogo, 
+    logo: togetheraiLogo,
     base_url: "https://api.together.xyz/v1",
     setup_instructions: "Get your API key from https://api.together.xyz/"
   },
+  {
+    id: 'user/fireworks',
+    alias: ['fireworks', 'fireworks-api'],
+    name: 'Fireworks AI',
+    description: 'Fireworks AI offers cheap and fast LLM infernece.',
+    logo: fireworksLogo,
+    base_url: "https://api.fireworks.ai/inference/v1",
+    setup_instructions: "Register at https://fireworks.ai and get your API key from https://app.fireworks.ai/settings/users/api-keys"
+  },
   {
     id: 'user/ollama',
     alias: ['ollama', 'ollama-api'],