indexify 0.0.26__py3-none-any.whl → 0.0.28__py3-none-any.whl

indexify/client.py

@@ -622,16 +622,22 @@ class IndexifyClient:
             json=req,
             headers={"Content-Type": "application/json"},
         )
-        return response.json()
+        response.raise_for_status()
+        return response.json()["content_id"]
 
-    def wait_for_extraction(self, content_id: str):
+    def wait_for_extraction(self, content_ids: Union[str, List[str]]):
         """
         Wait for extraction to complete for a given content id
 
         Args:
             - content_id (str): id of content
         """
-        response = self.get(f"namespaces/{self.namespace}/content/{content_id}/wait")
+        if type(content_ids) == str:
+            content_ids = [content_ids]
+        print("Waiting for extraction to complete for content id: ", ",".join(content_ids))
+        for content_id in content_ids:
+            response = self.get(f"namespaces/{self.namespace}/content/{content_id}/wait")
+            print("Extraction completed for content id: ", content_id)
         response.raise_for_status()
 
     def generate_unique_hex_id(self):

indexify/extraction_policy.py

@@ -33,11 +33,13 @@ class ExtractionPolicy:
 class ExtractionGraph:
     id: str
     name: str
+    description: str
     extraction_policies: List[ExtractionPolicy]
 
     @classmethod
     def from_dict(cls, json: dict):
         json["id"] = json.get("id", None)
+        json["description"] = json.get("description", None)
         if "namespace" in json.keys():
             json.pop("namespace")
         return ExtractionGraph(**json)
@@ -47,6 +49,11 @@ class ExtractionGraph:
         import yaml
 
         return ExtractionGraph.from_dict(yaml.load(spec, Loader=yaml.FullLoader))
+    
+    @staticmethod
+    def from_yaml_file(path: str):
+        with open(path, "r") as f:
+            return ExtractionGraph.from_yaml(f.read())
 
     def to_dict(self) -> dict:
         filtered_dict = {k: v for k, v in asdict(self).items() if v is not None}

{indexify-0.0.26.dist-info → indexify-0.0.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.0.26
+Version: 0.0.28
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.0.26.dist-info → indexify-0.0.28.dist-info}/RECORD

@@ -1,14 +1,14 @@
 indexify/__init__.py,sha256=Y40-Ur_tL7kGGs-reh9BTfEYGe-KyGxgdg-CmoFsXRQ,473
-indexify/client.py,sha256=oDu_MAEonxOAWKRl9KHG9ixxR_nMy_WO0f3Mi3LgjnI,21807
+indexify/client.py,sha256=qECQvvVPvzMDra8OQ-94u7J5ABFWSEmEx_uIEdJ3RiU,22159
 indexify/data_containers.py,sha256=r1wxJPtsmXbyKvb17fqxm-dPjKz51oZ62f8A8Zxls1c,361
 indexify/error.py,sha256=3umTeYb0ugtUyehV1ibfvaeACxAONPyWPc-1HRN4d1M,856
 indexify/exceptions.py,sha256=vjd5SPPNFIEW35GorSIodsqvm9RKHQm9kdp8t9gv-WM,111
-indexify/extraction_policy.py,sha256=dIyQK3N-QOpQ0BPjiZ_635o8A5ITNxaz1syQ_FPaE0k,1851
+indexify/extraction_policy.py,sha256=POluredrBw6DzTN0OyfPLaLFP5-2DoWGRK0V6w68R28,2080
 indexify/extractor.py,sha256=sWFLlXHgEfWlmiKAXN6ytUt_uG7th-XGNHqz-TG39gs,1216
 indexify/index.py,sha256=RvxYhJXEth-GKvqzlMiz5PuN1eIbZk84pt20piA1Gsw,504
 indexify/settings.py,sha256=LSaWZ0ADIVmUv6o6dHWRC3-Ry5uLbCw2sBSg1e_U7UM,99
 indexify/utils.py,sha256=rDN2lrsAs9noJEIjfx6ukmC2SAIyrlUt7QU-kaBjujM,125
-indexify-0.0.26.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.0.26.dist-info/METADATA,sha256=aCSbQl8iETigq0wIwnDmqFJ8jsaINmmhE_5fINCPXY4,1798
-indexify-0.0.26.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-indexify-0.0.26.dist-info/RECORD,,
+indexify-0.0.28.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.0.28.dist-info/METADATA,sha256=3r9wATECrfXFfmI68yU2cUOtrOBR4nT_ajPnag8IdZg,1798
+indexify-0.0.28.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+indexify-0.0.28.dist-info/RECORD,,