indexify 0.0.9__tar.gz → 0.0.10__tar.gz

{indexify-0.0.9 → indexify-0.0.10}/PKG-INFO

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

{indexify-0.0.9 → indexify-0.0.10}/indexify/__init__.py

@@ -1,10 +1,12 @@
 from .index import Index
 from .client import IndexifyClient
 from .extraction_policy import ExtractionPolicy
+from .client import IndexifyClient, Document
 from .settings import DEFAULT_SERVICE_URL
 
 __all__ = [
     "Index",
+    "Document",
     "IndexifyClient",
     "ExtractionPolicy",
     "DEFAULT_SERVICE_URL",

{indexify-0.0.9 → indexify-0.0.10}/indexify/client.py

@@ -50,7 +50,7 @@ class IndexifyClient:
         response = self.get(f"namespaces/{self.namespace}")
         response.raise_for_status()
         resp_json = response.json()
-        # initialize extractor_bindings
+        # initialize extraction_policies
         for eb in resp_json["namespace"]["extraction_policies"]:
             self.extraction_policies.append(ExtractionPolicy.from_dict(eb))
 
@@ -404,5 +404,6 @@ class IndexifyClient:
             response = self.post(
                 f"namespaces/{self.namespace}/upload_file",
                 files={"file": f},
+                timeout=None,
             )
             response.raise_for_status()

{indexify-0.0.9 → indexify-0.0.10}/indexify/extractor.py

@@ -16,22 +16,15 @@ class ExtractorSchema:
     outputs: dict[str, Union[EmbeddingSchema, dict]]
 
 
-@dataclass
-class Extractor:
-    name: str
-    description: str
-    input_params: dict
-    outputs: ExtractorSchema
-
-
 class Extractor:
     def __init__(
-        self, name: str, description: str, input_params: dict, outputs: ExtractorSchema
+        self, name: str, description: str, input_params: dict, outputs: ExtractorSchema, input_mime_types: list[str]
     ):
         self.name = name
         self.description = description
         self.input_params = input_params
         self.outputs = outputs
+        self.input_mime_types = input_mime_types
 
     @classmethod
     def from_dict(cls, data):
@@ -39,11 +32,12 @@ class Extractor:
             name=data["name"],
             description=data["description"],
             input_params=data["input_params"],
+            input_mime_types=data["input_mime_types"],
             outputs=data["outputs"],
         )
 
     def __repr__(self) -> str:
-        return f"Extractor(name={self.name}, description={self.description}, input_params={self.input_params}, outputs={self.outputs})"
+        return f"Extractor(name={self.name}, description={self.description}, input_params={self.input_params}, input_mime_types={self.input_mime_types}, outputs={self.outputs})"
 
     def __str__(self) -> str:
         return self.__repr__()

{indexify-0.0.9 → indexify-0.0.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.0.9"
+version = "0.0.10"
 description = "Python Client for Indexify"
 authors = ["Diptanu Gon Choudhury <diptanuc@gmail.com>", "Vijay Parthasarathy <vijay2win@gmail.com>"]
 license = "Apache 2.0"