SDK compatibility

plastic-labs · May 7, 2024 · c2d40db · c2d40db
1 parent 503875f
commit c2d40db
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/sdk/honcho/client.py b/sdk/honcho/client.py
@@ -449,7 +449,7 @@ async def get_user(self, name: str):
         Returns:
             AsyncUser: The User object
         """
-        url = f"{self.base_url}/users/{name}"
+        url = f"{self.base_url}/users/name/{name}"
         response = await self.client.get(url)
         response.raise_for_status()
         data = response.json()
@@ -793,7 +793,7 @@ async def get_collection(self, name: str):
             AsyncCollection: The Session object of the requested Session
 
         """
-        url = f"{self.base_url}/collections/{name}"
+        url = f"{self.base_url}/collections/name/{name}"
         response = await self.honcho.client.get(url)
         response.raise_for_status()
         data = response.json()

diff --git a/sdk/honcho/sync_client.py b/sdk/honcho/sync_client.py
@@ -449,7 +449,7 @@ def get_user(self, name: str):
         Returns:
             User: The User object
         """
-        url = f"{self.base_url}/users/{name}"
+        url = f"{self.base_url}/users/name/{name}"
         response = self.client.get(url)
         response.raise_for_status()
         data = response.json()
@@ -793,7 +793,7 @@ def get_collection(self, name: str):
             Collection: The Session object of the requested Session
 
         """
-        url = f"{self.base_url}/collections/{name}"
+        url = f"{self.base_url}/collections/name/{name}"
         response = self.honcho.client.get(url)
         response.raise_for_status()
         data = response.json()

diff --git a/sdk/tests/test_sync.py b/sdk/tests/test_sync.py
@@ -251,7 +251,9 @@ def test_paginated_messages():
         created_session.create_message(is_user=False, content="Hi")
 
     page_size = 7
-    get_message_response = created_session.get_messages(page=1, page_size=page_size)
+    get_message_response = created_session.get_messages(
+        page=1, page_size=page_size
+    )
 
     assert get_message_response is not None
     assert isinstance(get_message_response, GetMessagePage)
@@ -446,7 +448,9 @@ def test_collection_query():
     collection = user.create_collection(col_name)
 
     # Add documents
-    doc1 = collection.create_document(content="The user loves puppies", metadata={})
+    doc1 = collection.create_document(
+        content="The user loves puppies", metadata={}
+    )
     doc2 = collection.create_document(content="The user owns a dog", metadata={})
     doc3 = collection.create_document(content="The user is a doctor", metadata={})