Merge pull request #91 from Nebhay/fix-FIREWORKS

Fix Fireworks

Author: jamiepine

src/config.rs

@@ -194,6 +194,7 @@ const MOONSHOT_PROVIDER_BASE_URL: &str = "https://api.moonshot.ai";
 const ZHIPU_PROVIDER_BASE_URL: &str = "https://api.z.ai/api/paas/v4";
 const ZAI_CODING_PLAN_BASE_URL: &str = "https://api.z.ai/api/coding/paas/v4";
 const NVIDIA_PROVIDER_BASE_URL: &str = "https://integrate.api.nvidia.com";
+const FIREWORKS_PROVIDER_BASE_URL: &str = "https://api.fireworks.ai/inference";
 pub(crate) const GEMINI_PROVIDER_BASE_URL: &str =
     "https://generativelanguage.googleapis.com/v1beta/openai";
 
@@ -1924,6 +1925,17 @@ impl Config {
                 });
         }
 
+        if let Some(fireworks_key) = llm.fireworks_key.clone() {
+            llm.providers
+                .entry("fireworks".to_string())
+                .or_insert_with(|| ProviderConfig {
+                    api_type: ApiType::OpenAiCompletions,
+                    base_url: FIREWORKS_PROVIDER_BASE_URL.to_string(),
+                    api_key: fireworks_key,
+                    name: None,
+                });
+        }
+
         if let Some(gemini_key) = llm.gemini_key.clone() {
             llm.providers
                 .entry("gemini".to_string())
@@ -2260,6 +2272,17 @@ impl Config {
                 });
         }
 
+        if let Some(fireworks_key) = llm.fireworks_key.clone() {
+            llm.providers
+                .entry("fireworks".to_string())
+                .or_insert_with(|| ProviderConfig {
+                    api_type: ApiType::OpenAiCompletions,
+                    base_url: FIREWORKS_PROVIDER_BASE_URL.to_string(),
+                    api_key: fireworks_key,
+                    name: None,
+                });
+        }
+
         if let Some(gemini_key) = llm.gemini_key.clone() {
             llm.providers
                 .entry("gemini".to_string())