udpate obj detection

Author: winzamark123

jigsawstack/__init__.py

@@ -15,7 +15,6 @@
 from .embedding import Embedding, AsyncEmbedding
 from .exceptions import JigsawStackError
 from .image_generation import ImageGeneration, AsyncImageGeneration
-from .object_detection import ObjectDetection, AsyncObjectDetection
 
 
 class JigsawStack:
@@ -119,11 +118,6 @@ def __init__(
             disable_request_logging=disable_request_logging,
         ).image_generation
 
-        self.object_detection = ObjectDetection(
-            api_key=api_key,
-            api_url=api_url,
-            disable_request_logging=disable_request_logging,
-        )
 
 
 class AsyncJigsawStack:
@@ -235,11 +229,6 @@ def __init__(
             disable_request_logging=disable_request_logging,
         ).image_generation
 
-        self.object_detection = AsyncObjectDetection(
-            api_key=api_key,
-            api_url=api_url,
-            disable_request_logging=disable_request_logging,
-        )
 
 
 # Create a global instance of the Web class

jigsawstack/vision.py

@@ -212,7 +212,7 @@ async def vocr(self, params: VOCRParams) -> OCRResponse:
 
     async def object_detection(self, params: ObjectDetectionParams) -> ObjectDetectionResponse:
         path = "/ai/object_detection"
-        resp = AsyncRequest(
+        resp = await AsyncRequest(
             config=self.config,
             path=path,
             params=cast(Dict[Any, Any], params),

tests/test_object_detection.py

@@ -15,7 +15,7 @@
 
 def test_object_detection_response():
     try:
-        result = jigsaw.object_detection.detect({"url": "https://rogilvkqloanxtvjfrkm.supabase.co/storage/v1/object/public/demo/Collabo%201080x842.jpg"})
+        result = jigsaw.vision.object_detection({"url": "https://rogilvkqloanxtvjfrkm.supabase.co/storage/v1/object/public/demo/Collabo%201080x842.jpg"})
         print(result)
         assert result["success"] == True
     except JigsawStackError as e:
@@ -26,7 +26,7 @@ def test_object_detection_response_async():
     async def _test():
         client = jigsawstack.AsyncJigsawStack()
         try:
-            result = await client.object_detection.detect({"url": "https://rogilvkqloanxtvjfrkm.supabase.co/storage/v1/object/public/demo/Collabo%201080x842.jpg"})
+            result = await client.vision.object_detection({"url": "https://rogilvkqloanxtvjfrkm.supabase.co/storage/v1/object/public/demo/Collabo%201080x842.jpg"})
             print(result)
             assert result["success"] == True
         except JigsawStackError as e: