aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-24 09:24:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

Helicone Lock Manager fix (#263)

Browse Source 
Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
This commit is contained in:
merwanehamadi
2023-08-06 11:30:03 -07:00
committed by GitHub
parent f2c0f87cf8
commit e32713be68
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
agbenchmark/generate_test.py
View File

@@ -1,5 +1,6 @@
import glob
import importlib
import os
import sys
import types
from collections import deque
@@ -58,7 +59,8 @@ def create_single_test(
def test_method(self, config: Dict[str, Any], request) -> None: # type: ignore
from helicone.lock import HeliconeLockManager
HeliconeLockManager.write_custom_property("challenge", self.data.name)
if os.environ.get("HELICONE_API_KEY"):
HeliconeLockManager.write_custom_property("challenge", self.data.name)
cutoff = self.data.cutoff or 60
self.setup_challenge(config, cutoff)

6
agbenchmark/start_benchmark.py
View File

@@ -17,8 +17,10 @@ from agbenchmark.utils.utils import (
CURRENT_DIRECTORY = Path(__file__).resolve().parent
BENCHMARK_START_TIME = datetime.now().strftime("%Y-%m-%d-%H:%M")
HeliconeLockManager.write_custom_property("benchmark_start_time", BENCHMARK_START_TIME)
if os.environ.get("HELICONE_API_KEY"):
HeliconeLockManager.write_custom_property(
"benchmark_start_time", BENCHMARK_START_TIME
)
(
HOME_DIRECTORY,