fix

Gapminder · Feb 26, 2025 · 29c5ed6 · 29c5ed6
1 parent fbb5bb2
commit 29c5ed6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/automation-api/lib/pilot/send_batch_prompt.py b/automation-api/lib/pilot/send_batch_prompt.py
@@ -56,16 +56,16 @@ def main():
         method = args.method.lower()
         # Create batch job instance
         if method == "openai":
-            provider = args.provider.lower()
-            if provider:
+            if args.provider:
+                provider = args.provider.lower()
                 batch_job = OpenAIBatchJob(args.jsonl_file, provider=provider)
             else:
                 batch_job = OpenAIBatchJob(args.jsonl_file)
         elif method in ["anthropic", "vertex"]:
             batch_job = PROVIDER_CLASSES[method](args.jsonl_file)
         else:
-            provider = args.provider.lower()
-            if provider:
+            if args.provider:
+                provider = args.provider.lower()
                 batch_job = LiteLLMBatchJob(
                     args.jsonl_file, provider=provider, num_processes=args.processes
                 )