activate time travel

agentops/__init__.py

@@ -8,6 +8,7 @@
 from .helpers import check_agentops_update
 from .log_config import logger
 from .session import Session
+from .cli import cli
 import threading
 
 try:

agentops/cli.py

@@ -2,7 +2,7 @@
 from .time_travel import fetch_time_travel_id, set_time_travel_active_state
 
 
-def main():
+def cli():
     parser = argparse.ArgumentParser(description="AgentOps CLI")
     subparsers = parser.add_subparsers(dest="command")
 
@@ -35,7 +35,3 @@ def main():
             set_time_travel_active_state("on")
         if args.off:
             set_time_travel_active_state("off")
-
-
-if __name__ == "__main__":
-    main()

pyproject.toml

@@ -41,3 +41,6 @@ Issues = "https://github.com/AgentOps-AI/agentops/issues"
 
 [tool.autopep8]
 max_line_length = 120
+
+[project.scripts]
+agentops = "agentops:cli"