supporting tuple

AgentOps-AI · Aug 16, 2024 · 7a41ada · 7a41ada
1 parent 351725c
commit 7a41ada
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/agentops/time_travel.py b/agentops/time_travel.py
@@ -62,9 +62,9 @@ def fetch_completion_override_from_time_travel_cache(kwargs):
 
 # NOTE: This is specific to the messages: [{'role': '...', 'content': '...'}, ...] format
 def find_cache_hit(prompt_messages, completion_overrides):
-    if not isinstance(prompt_messages, list):
+    if not isinstance(prompt_messages, (list, tuple)):
         print(
-            "Time Travel Error - unexpected type for prompt_messages. Expected 'list'. Got ",
+            "Time Travel Error - unexpected type for prompt_messages. Expected 'list' or 'tuple'. Got ",
             type(prompt_messages),
         )
         return None

diff --git a/tests/core_manual_tests/time_travel.py b/tests/core_manual_tests/time_travel.py
@@ -11,12 +11,12 @@
 
 try:
     chat_completion_1 = client.chat.completions.create(
-        messages=(
+        messages=[
             {
                 "content": "Come up with a random superpower that isn't time travel. Just return the superpower in the format: 'Superpower: [superpower]'",
                 "role": "user",
-            },
-        ),
+            }
+        ],
         model="gpt-3.5-turbo-0125",
     )
     content1 = chat_completion_1.choices[0].message.content