Automated version control system: the agentops cli command is back!

agentops/cli.py

@@ -1,35 +1,40 @@
 import argparse
+import importlib.metadata
+
 from .time_travel import fetch_time_travel_id, set_time_travel_active_state
 
 
+def version_command():
+    try:
+        version = importlib.metadata.version("agentops")
+        print(f"AgentOps version: {version}")
+    except importlib.metadata.PackageNotFoundError:
+        print("AgentOps package not found. Are you in development mode?")
+
+
 def main():
     parser = argparse.ArgumentParser(description="AgentOps CLI")
-    subparsers = parser.add_subparsers(dest="command")
-
-    timetravel_parser = subparsers.add_parser("timetravel", help="Time Travel Debugging commands", aliases=["tt"])
-    timetravel_parser.add_argument(
-        "branch_name",
-        type=str,
-        nargs="?",
-        help="Given a branch name, fetches the cache file for Time Travel Debugging. Turns on feature by default",
-    )
-    timetravel_parser.add_argument(
-        "--on",
-        action="store_true",
-        help="Turns on Time Travel Debugging",
-    )
-    timetravel_parser.add_argument(
-        "--off",
-        action="store_true",
-        help="Turns off Time Travel Debugging",
-    )
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
+    # Version command
+    version_parser = subparsers.add_parser("version", help="Show AgentOps version")
+
+    # Time travel command
+    time_travel_parser = subparsers.add_parser("time-travel", help="Time travel related commands")
+    time_travel_parser.add_argument("--id", help="Get time travel ID", action="store_true")
+    time_travel_parser.add_argument("--off", help="Turn off time travel", action="store_true")
 
     args = parser.parse_args()
 
-    if args.command in ["timetravel", "tt"]:
-        if args.branch_name:
-            fetch_time_travel_id(args.branch_name)
-        if args.on:
-            set_time_travel_active_state(True)
+    if args.command == "version":
+        version_command()
+    elif args.command == "time-travel":
+        if args.id:
+            ttd_id = None  # This would typically come from somewhere
+            print(fetch_time_travel_id(ttd_id))
         if args.off:
             set_time_travel_active_state(False)
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -1,10 +1,10 @@
 [build-system]
-requires = ["hatchling"]
+requires = ["hatchling", "hatch-vcs"]
 build-backend = "hatchling.build"
 
 [project]
 name = "agentops"
-version = "0.3.24"
+dynamic = ["version"]
 authors = [
   { name="Alex Reibman", email="areibman@gmail.com" },
   { name="Shawn Qiu", email="siyangqiu@gmail.com" },
@@ -39,6 +39,10 @@ dependencies = [
     "opentelemetry-exporter-otlp-proto-http>=1.27.0; python_version>='3.10'",
 ]
 
+[project.scripts]
+agentops = "agentops.cli:main"
+
+
 [dependency-groups]
 test = [
     "openai>=1.0.0",
@@ -187,3 +191,5 @@ exclude = [
 [tool.hatch.metadata]
 allow-direct-references = true
 
+[tool.hatch.version]
+source = "vcs"