activate time travel

agentops/cli.py

@@ -32,10 +32,6 @@
         if args.branch_name:
             fetch_time_travel_id(args.branch_name)
         if args.on:
-            set_time_travel_active_state("on")
+            set_time_travel_active_state(True)
         if args.off:
-            set_time_travel_active_state("off")
-
-
-if __name__ == "__main__":
-    main()
+            set_time_travel_active_state(False)

agentops/time_travel.py

@@ -1,10 +1,9 @@
 import json
 import yaml
+import os
 from .http_client import HttpClient
 from .exceptions import ApiServerException
-import os
 from .helpers import singleton
-from os import environ
 
 
 @singleton
@@ -32,13 +31,10 @@
 
 def fetch_time_travel_id(ttd_id):
     try:
-        endpoint = environ.get("AGENTOPS_API_ENDPOINT", "https://api.agentops.ai")
-        payload = json.dumps({"ttd_id": ttd_id}).encode("utf-8")
+        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.status_code}"
-            )
+            raise Exception(f"Failed to fetch TTD with status code {ttd_res.code}")
 
         prompt_to_returns_map = {
             "completion_overrides": {

examples/openai-gpt.ipynb

@@ -130,10 +130,11 @@
    },
    "outputs": [],
    "source": [
-    "message = ({\"role\": \"user\", \"content\": \"Write a 12 word poem about secret agents.\"},)\n",
+    "message = [{\"role\": \"user\", \"content\": \"Write a 12 word poem about secret agents.\"}]\n",
     "res = openai.chat.completions.create(\n",
-    "    model=\"gpt-3.5-turbo\", messages=message, temperature=0.5, stream=True\n",
-    ")"
+    "    model=\"gpt-3.5-turbo\", messages=message, temperature=0.5, stream=False\n",
+    ")\n",
+    "print(res.choices[0].message[\"content\"])"
    ]
   },
   {
@@ -282,7 +283,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.3"
+   "version": "3.12.4"
   }
  },
  "nbformat": 4,

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:main"