Spaces:
Runtime error
Runtime error
updates
Browse files- olas-predict-benchmark +1 -1
- tabs/run_benchmark.py +3 -10
olas-predict-benchmark
CHANGED
@@ -1 +1 @@
|
|
1 |
-
Subproject commit
|
|
|
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"] =
|
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
|