richardblythman commited on
Commit
c4e55cd
1 Parent(s): f62fea6
Files changed (1) hide show
  1. tabs/run_benchmark.py +2 -2
tabs/run_benchmark.py CHANGED
@@ -23,9 +23,9 @@ def run_benchmark_main(tool_name, model_name, num_questions, openai_api_key, ant
23
  if openrouter_api_key:
24
  kwargs["api_keys"]["openrouter"] = openrouter_api_key
25
 
26
- if "gpt" in model:
27
  kwargs["llm_provider"] = "openai"
28
- elif "claude" in model:
29
  kwargs["llm_provider"] = "anthropic"
30
  else:
31
  kwargs["llm_provider"] = "openrouter"
 
23
  if openrouter_api_key:
24
  kwargs["api_keys"]["openrouter"] = openrouter_api_key
25
 
26
+ if "gpt" in model_name:
27
  kwargs["llm_provider"] = "openai"
28
+ elif "claude" in model_name:
29
  kwargs["llm_provider"] = "anthropic"
30
  else:
31
  kwargs["llm_provider"] = "openrouter"