🐛 FIX: threads implementation

Author: saqibameen

examples/threads/threads.append.py

@@ -19,7 +19,7 @@ def main():
     lb = Langbase(api_key=langbase_api_key)
 
     # Thread ID to append messages to
-    thread_id = "thread_123456789"  # Replace with your actual thread ID
+    thread_id = "thread_123"  # Replace with your actual thread ID
 
     # Messages to append
     messages = [

examples/threads/threads.delete.py

@@ -19,9 +19,7 @@ def main():
     lb = Langbase(api_key=langbase_api_key)
 
     # Thread ID to delete
-    thread_id = (
-        "431bac51-929c-4257-8251-baefcd251d3a"  # Replace with your actual thread ID
-    )
+    thread_id = "thread_123"  # Replace with your actual thread ID
 
     # Delete the thread
     try:

examples/threads/threads.get.py

@@ -20,7 +20,7 @@ def main():
     lb = Langbase(api_key=langbase_api_key)
 
     # Thread ID to retrieve
-    thread_id = "thread-123"  # Replace with your thread ID
+    thread_id = "thread_123"  # Replace with your thread ID
 
     # Get the specific thread
     try:

examples/threads/threads.messages.list.py

@@ -21,9 +21,7 @@ def main():
 
     # List all threads
     try:
-        threads = lb.threads.messages.list(
-            thread_id="3a958893-6175-4d96-9053-876ff1b37227"
-        )
+        threads = lb.threads.messages.list(thread_id="thread_123")
 
         print(json.dumps(threads, indent=2))
 

examples/threads/threads.update.py

@@ -20,7 +20,7 @@ def main():
     lb = Langbase(api_key=langbase_api_key)
 
     # Thread ID to update
-    thread_id = "thread_123456789"  # Replace with your actual thread ID
+    thread_id = "thread_123"  # Replace with your actual thread ID
 
     # New metadata to set for the thread
     updated_metadata = {
@@ -31,7 +31,8 @@ def main():
     # Update the thread metadata
     try:
         updated_thread = lb.threads.update(
-            {"thread_id": thread_id, "metadata": updated_metadata}
+            thread_id=thread_id,
+            metadata=updated_metadata,
         )
 
         print(json.dumps(updated_thread, indent=2))

langbase/primitives/threads.py

@@ -121,8 +121,6 @@ def append(
         Returns:
             List of added messages
         """
-        options = {"messages": messages}
-
         return self.request.post(
-            THREAD_MESSAGES_ENDPOINT.format(thread_id=thread_id), options
+            THREAD_MESSAGES_ENDPOINT.format(thread_id=thread_id), messages
         )

tests/test_threads.py

@@ -224,7 +224,7 @@ def test_threads_append(self, langbase_client, mock_responses):
         assert len(responses.calls) == 1
         request = responses.calls[0].request
         validate_response_headers(request.headers, AUTH_AND_JSON_CONTENT_HEADER)
-        assert json.loads(request.body) == {"messages": request_body["messages"]}
+        assert json.loads(request.body) == request_body["messages"]
         assert (
             request.url
             == f"{BASE_URL}{THREAD_MESSAGES_ENDPOINT.format(thread_id=thread_id)}"