Update api_clients.py

api_clients.py

@@ -21,15 +21,26 @@ from web_extraction import extract_website_content, enhance_query_with_search
 
 # HF Inference Client
 HF_TOKEN = os.getenv('HF_TOKEN')
+GROQ_API_KEY = os.getenv('GROQ_API_KEY')
+FIREWORKS_API_KEY = os.getenv('FIREWORKS_API_KEY')
 
 def get_inference_client(model_id):
-    """Return an InferenceClient with provider based on model_id."""
-    provider = "groq" if model_id == "moonshotai/Kimi-K2-Instruct" else "auto"
-    return InferenceClient(
-        provider=provider,
-        api_key=HF_TOKEN,
-        bill_to="huggingface"
-    )
+    """Return an InferenceClient configured for Hugging Face, Groq, or Fireworks AI."""
+    if model_id == "moonshotai/Kimi-K2-Instruct":
+        return InferenceClient(
+            base_url="https://api.groq.com/openai/v1",
+            api_key=GROQ_API_KEY
+        )
+    elif model_id.startswith("fireworks/"):
+        return InferenceClient(
+            base_url="https://api.fireworks.ai/inference/v1",
+            api_key=FIREWORKS_API_KEY
+        )
+    else:
+        return InferenceClient(
+            model=model_id,
+            api_key=HF_TOKEN
+        )
 
 # Tavily Search Client
 TAVILY_API_KEY = os.getenv('TAVILY_API_KEY')