From 620388d489250b1b0335785380a27e487feafb47 Mon Sep 17 00:00:00 2001 From: Shawn Qiu Date: Tue, 13 Aug 2024 00:16:18 -0700 Subject: [PATCH] imports --- agentops/time_travel.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/agentops/time_travel.py b/agentops/time_travel.py index 4fc6fa74..5b3b9a6f 100644 --- a/agentops/time_travel.py +++ b/agentops/time_travel.py @@ -4,7 +4,6 @@ from .http_client import HttpClient from .exceptions import ApiServerException from .helpers import singleton -from os import environ @singleton @@ -32,7 +31,7 @@ def __init__(self): def fetch_time_travel_id(ttd_id): try: - endpoint = environ.get("AGENTOPS_API_ENDPOINT", "https://api.agentops.ai") + endpoint = os.environ.get("AGENTOPS_API_ENDPOINT", "https://api.agentops.ai") ttd_res = HttpClient.get(f"{endpoint}/v2/ttd/{ttd_id}") if ttd_res.code != 200: raise Exception(f"Failed to fetch TTD with status code {ttd_res.code}")