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

Files changed (1) hide show
  1. 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