arshy commited on
Commit
1bea577
·
1 Parent(s): 44c0013
Files changed (2) hide show
  1. olas-predict-benchmark +1 -1
  2. tabs/run_benchmark.py +3 -10
olas-predict-benchmark CHANGED
@@ -1 +1 @@
1
- Subproject commit b132e2c4b218a9035b9a62fb5a4254e9f21dd9e6
 
1
+ Subproject commit b4c08922f923bcd3fb7dfeef1bdb647eead72d85
tabs/run_benchmark.py CHANGED
@@ -1,14 +1,5 @@
1
  import os
2
- import importlib
3
- import sys
4
- # from pathlib import Path
5
  from benchmark.run_benchmark import run_benchmark
6
- # from dotenv import load_dotenv
7
-
8
-
9
- # file_path = Path(__file__).resolve()
10
- # parent_path = file_path.parent.parent
11
- # load_dotenv(parent_path / ".env")
12
 
13
 
14
  def run_benchmark_main(tool_name, model_name, openai_api_key, anthropic_api_key):
@@ -16,12 +7,14 @@ def run_benchmark_main(tool_name, model_name, openai_api_key, anthropic_api_key)
16
  # Empyt the results directory
17
  os.system("rm -rf results/*")
18
 
 
 
19
  # Set the benchmark parameters
20
  kwargs = {}
21
  kwargs["num_questions"] = 2
22
  kwargs["tools"] = [tool_name]
23
  if model_name:
24
- kwargs["model"] = [model_name]
25
  kwargs["api_keys"] = {}
26
  if openai_api_key:
27
  kwargs["api_keys"]["openai"] = openai_api_key
 
1
  import os
 
 
 
2
  from benchmark.run_benchmark import run_benchmark
 
 
 
 
 
 
3
 
4
 
5
  def run_benchmark_main(tool_name, model_name, openai_api_key, anthropic_api_key):
 
7
  # Empyt the results directory
8
  os.system("rm -rf results/*")
9
 
10
+ print(f"Running benchmark with the following parameters: {tool_name}, {model_name}, {openai_api_key}, {anthropic_api_key}")
11
+
12
  # Set the benchmark parameters
13
  kwargs = {}
14
  kwargs["num_questions"] = 2
15
  kwargs["tools"] = [tool_name]
16
  if model_name:
17
+ kwargs["model"] = model_name
18
  kwargs["api_keys"] = {}
19
  if openai_api_key:
20
  kwargs["api_keys"]["openai"] = openai_api_key