feat(cha-769): support shared locations

Author: Rafael Marinho

stream_chat/async_chat/client.py

@@ -860,18 +860,26 @@ async def query_drafts(
         data: Dict[str, Union[str, Dict[str, Any], List[SortParam]]] = {
             "user_id": user_id
         }
-
         if filter is not None:
             data["filter"] = cast(dict, filter)
-
         if sort is not None:
             data["sort"] = cast(dict, sort)
-
         if options is not None:
             data.update(cast(dict, options))
-
         return await self.post("drafts/query", data=data)
 
+    async def get_user_locations(self, user_id: str, **options: Any) -> StreamResponse:
+        params = {"user_id": user_id, **options}
+        return await self.get("users/live_locations", params=params)
+
+    async def update_user_location(
+        self, message_id: str, options: Optional[SharedLocationsOptions] = None,
+    ) -> StreamResponse:
+        data = {"message_id": message_id}
+        if options is not None:
+            data.update(cast(dict, options))
+        return await self.put("users/live_locations", data=data)
+
     async def close(self) -> None:
         await self.session.close()
 
@@ -886,15 +894,4 @@ async def __aexit__(
     ) -> None:
         await self.close()
 
-    async def get_user_locations(self, user_id: str, **options: Any) -> StreamResponse:
-        params = {"user_id": user_id, **options}
-        return await self.get("users/live_locations", params=params)
-
-    async def update_user_location(
-        self, message_id: str, options: Optional[SharedLocationsOptions] = None,
-    ) -> StreamResponse:
-        data = {"message_id": message_id, **options}
-        if options is not None:
-            data.update(cast(dict, options))
-        return await self.put("users/live_locations", data=data)
-    
+    

stream_chat/client.py

@@ -18,6 +18,7 @@
     SegmentType,
     SegmentUpdatableFields,
 )
+from stream_chat.types.shared_locations import SharedLocationsOptions
 
 if sys.version_info >= (3, 8):
     from typing import Literal
@@ -824,3 +825,15 @@ def query_drafts(
         if options is not None:
             data.update(cast(dict, options))
         return self.post("drafts/query", data=data)
+
+    def get_user_locations(self, user_id: str, **options: Any) -> StreamResponse:
+        params = {"user_id": user_id, **options}
+        return self.get("users/live_locations", params=params)
+
+    def update_user_location(
+        self, message_id: str, options: Optional[SharedLocationsOptions] = None,
+    ) -> StreamResponse:
+        data = {"message_id": message_id}
+        if options is not None:
+            data.update(cast(dict, options))
+        return self.put("users/live_locations", data=data)