Skip to content

Commit

Permalink
Changes for PR
Browse files Browse the repository at this point in the history
  • Loading branch information
HowieG committed Jul 31, 2024
1 parent 9c298b4 commit 156628f
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 7 deletions.
2 changes: 1 addition & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -164,4 +164,4 @@ cython_debug/
.DS_Store

agentops_time_travel.json
agentops_time_travel.yaml
.agentops_time_travel.yaml
4 changes: 2 additions & 2 deletions agentops/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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:
Expand Down
6 changes: 2 additions & 4 deletions agentops/time_travel.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down Expand Up @@ -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)
Expand All @@ -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 {}
Expand Down
File renamed without changes.

0 comments on commit 156628f

Please sign in to comment.