add prime intellect key (#119)

Author: alexzhang13

rlm/clients/openai.py

@@ -14,6 +14,7 @@
 DEFAULT_OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 DEFAULT_OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 DEFAULT_VERCEL_API_KEY = os.getenv("AI_GATEWAY_API_KEY")
+DEFAULT_PRIME_API_KEY = os.getenv("PRIME_API_KEY")
 DEFAULT_PRIME_INTELLECT_BASE_URL = "https://api.pinference.ai/api/v1/"
 
 
@@ -38,6 +39,8 @@ def __init__(
                 api_key = DEFAULT_OPENROUTER_API_KEY
             elif base_url == "https://ai-gateway.vercel.sh/v1":
                 api_key = DEFAULT_VERCEL_API_KEY
+            elif base_url == DEFAULT_PRIME_INTELLECT_BASE_URL:
+                api_key = DEFAULT_PRIME_API_KEY
 
         # For vLLM, set base_url to local vLLM server address.
         self.client = openai.OpenAI(api_key=api_key, base_url=base_url, timeout=self.timeout)