diff --git a/.gitignore b/.gitignore index e8ae1ba2..4db649aa 100644 --- a/.gitignore +++ b/.gitignore @@ -164,4 +164,4 @@ cython_debug/ .DS_Store agentops_time_travel.json -agentops_time_travel.yaml \ No newline at end of file +.agentops_time_travel.yaml \ No newline at end of file diff --git a/agentops/cli.py b/agentops/cli.py index d3f52ee0..29b9e90b 100644 --- a/agentops/cli.py +++ b/agentops/cli.py @@ -7,7 +7,7 @@ def main(): subparsers = parser.add_subparsers(dest="command") timetravel_parser = subparsers.add_parser( - "timetravel", help="Time Travel Debugging commands" + "timetravel", help="Time Travel Debugging commands", aliases=["tt"] ) timetravel_parser.add_argument( "branch_name", @@ -28,7 +28,7 @@ def main(): args = parser.parse_args() - if args.command == "timetravel": + if args.command in ["timetravel", "tt"]: if args.branch_name: fetch_time_travel_id(args.branch_name) if args.on: diff --git a/agentops/time_travel.py b/agentops/time_travel.py index 98112fcc..e68bb1fe 100644 --- a/agentops/time_travel.py +++ b/agentops/time_travel.py @@ -3,8 +3,6 @@ from .http_client import HttpClient import os from .helpers import singleton -from dotenv import load_dotenv -from .log_config import logger from os import environ @@ -82,7 +80,7 @@ def fetch_prompt_override_from_time_travel_cache(kwargs): def check_time_travel_active(): script_dir = os.path.dirname(os.path.abspath(__file__)) parent_dir = os.path.dirname(script_dir) - config_file_path = os.path.join(parent_dir, "agentops_time_travel.yaml") + config_file_path = os.path.join(parent_dir, ".agentops_time_travel.yaml") with open(config_file_path, "r") as config_file: config = yaml.safe_load(config_file) @@ -94,7 +92,7 @@ def check_time_travel_active(): def set_time_travel_active_state(active_setting): - config_path = "agentops_time_travel.yaml" + config_path = ".agentops_time_travel.yaml" try: with open(config_path, "r") as config_file: config = yaml.safe_load(config_file) or {} diff --git a/tests/test_time_travel.py b/tests/core_manual_tests/time_travel.py similarity index 100% rename from tests/test_time_travel.py rename to tests/core_manual_tests/time_travel.py