Spaces:
Running
Running
Pulastya B
commited on
Commit
·
50f9263
1
Parent(s):
9ae6bf4
Fix: Strip whitespace from API keys
Browse files- Prevents 'Illegal header value' errors from trailing spaces
- Applies to all providers: Mistral, Groq, Gemini
- Fixes LocalProtocolError when API keys have whitespace
- src/orchestrator.py +3 -3
src/orchestrator.py
CHANGED
|
@@ -172,7 +172,7 @@ class DataScienceCopilot:
|
|
| 172 |
raise ValueError("Mistral API key must be provided or set in MISTRAL_API_KEY env var")
|
| 173 |
|
| 174 |
from mistralai.client import MistralClient
|
| 175 |
-
self.mistral_client = MistralClient(api_key=api_key)
|
| 176 |
self.model = os.getenv("MISTRAL_MODEL", "mistral-large-latest")
|
| 177 |
self.reasoning_effort = reasoning_effort
|
| 178 |
self.gemini_model = None
|
|
@@ -185,7 +185,7 @@ class DataScienceCopilot:
|
|
| 185 |
if not api_key:
|
| 186 |
raise ValueError("Groq API key must be provided or set in GROQ_API_KEY env var")
|
| 187 |
|
| 188 |
-
self.groq_client = Groq(api_key=api_key)
|
| 189 |
self.model = os.getenv("GROQ_MODEL", "llama-3.3-70b-versatile")
|
| 190 |
self.reasoning_effort = reasoning_effort
|
| 191 |
self.gemini_model = None
|
|
@@ -198,7 +198,7 @@ class DataScienceCopilot:
|
|
| 198 |
if not api_key:
|
| 199 |
raise ValueError("Google API key must be provided or set in GOOGLE_API_KEY or GEMINI_API_KEY env var")
|
| 200 |
|
| 201 |
-
genai.configure(api_key=api_key)
|
| 202 |
self.model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
|
| 203 |
|
| 204 |
# Configure safety settings to be more permissive for data science content
|
|
|
|
| 172 |
raise ValueError("Mistral API key must be provided or set in MISTRAL_API_KEY env var")
|
| 173 |
|
| 174 |
from mistralai.client import MistralClient
|
| 175 |
+
self.mistral_client = MistralClient(api_key=api_key.strip())
|
| 176 |
self.model = os.getenv("MISTRAL_MODEL", "mistral-large-latest")
|
| 177 |
self.reasoning_effort = reasoning_effort
|
| 178 |
self.gemini_model = None
|
|
|
|
| 185 |
if not api_key:
|
| 186 |
raise ValueError("Groq API key must be provided or set in GROQ_API_KEY env var")
|
| 187 |
|
| 188 |
+
self.groq_client = Groq(api_key=api_key.strip())
|
| 189 |
self.model = os.getenv("GROQ_MODEL", "llama-3.3-70b-versatile")
|
| 190 |
self.reasoning_effort = reasoning_effort
|
| 191 |
self.gemini_model = None
|
|
|
|
| 198 |
if not api_key:
|
| 199 |
raise ValueError("Google API key must be provided or set in GOOGLE_API_KEY or GEMINI_API_KEY env var")
|
| 200 |
|
| 201 |
+
genai.configure(api_key=api_key.strip())
|
| 202 |
self.model = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
|
| 203 |
|
| 204 |
# Configure safety settings to be more permissive for data science content
|