Fix all other client calls (#100)

Fix the remaining client calls for document ai.

Verified for readme.

Author: stangirala

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tensorlake"
-version = "0.1.58"
+version = "0.1.59"
 description = "Tensorlake SDK for Document Ingestion API and Serverless Workflows"
 authors = ["Tensorlake Inc. <[email protected]>"]
 homepage = "https://github.com/tensorlakeai/tensorlake"

src/tensorlake/documentai/client.py

@@ -31,7 +31,6 @@ def __init__(self, api_key: str = ""):
             self.api_key = os.getenv("TENSORLAKE_API_KEY").strip()
 
         self._client = httpx.Client(base_url=DOC_AI_BASE_URL, timeout=None)
-        self._async_client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
         self.__file_uploader__ = FileUploader(api_key=api_key)
 
     def __headers__(self):
@@ -56,7 +55,8 @@ async def get_job_async(self, job_id: str) -> Job:
         """
         Get the result of a job by its ID asynchronously.
         """
-        response = await self._async_client.get(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.get(
             url=f"jobs/{job_id}",
             headers=self.__headers__(),
         )
@@ -73,7 +73,8 @@ async def delete_job_async(self, job_id: str):
         """
         Delete a job by its ID asynchronously.
         """
-        response = await self._async_client.delete(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.delete(
             url=f"jobs/{job_id}",
             headers=self.__headers__(),
         )
@@ -89,7 +90,8 @@ async def jobs_async(self, cursor: Optional[str] = None) -> PaginatedResult[Job]
         """
         Get a list of jobs asynchronously.
         """
-        response = await self._async_client.get(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.get(
             url="/jobs",
             headers=self.__headers__(),
             params={"cursor": cursor} if cursor else None,
@@ -179,7 +181,8 @@ async def files_async(
         """
         Get a list of files asynchronously.
         """
-        response = await self._async_client.get(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.get(
             url="/files",
             headers=self.__headers__(),
             params={"cursor": cursor} if cursor else None,
@@ -234,7 +237,8 @@ async def parse_async(
         """
         Parse a document asynchronously.
         """
-        response = await self._async_client.post(
+        client = httpx.Client(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.post(
             url="/parse",
             headers=self.__headers__(),
             json=self.__create_parse_req__(file, options, deliver_webhook),
@@ -283,7 +287,8 @@ async def upload_async(self, path: Union[str, Path]) -> str:
             httpx.HTTPError: If the request fails
             FileNotFoundError: If the file doesn't exist
         """
-        return await self.__file_uploader__.upload_file_async(path)
+        uploader = FileUploader(api_key=api_key)
+        return await uploader.upload_file_async(path)
 
     def delete_file(self, file_id: str):
         """
@@ -295,7 +300,8 @@ async def delete_file_async(self, file_id: str):
         """
         Delete a file by its ID asynchronously.
         """
-        response = await self._async_client.delete(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.delete(
             url=f"files/{file_id}",
             headers=self.__headers__(),
         )
@@ -332,7 +338,8 @@ async def create_dataset_async(
             if existing_dataset:
                 return existing_dataset
 
-        response = await self._async_client.post(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.post(
             url="datasets",
             headers=self.__headers__(),
             json={
@@ -354,19 +361,14 @@ def get_dataset(self, name: str) -> Optional[Dataset]:
             Dataset: The dataset.
         """
 
-        async def asyncfunc():
-            dataset = await self.get_dataset_async(name)
-            return dataset
-
-        loop = asyncio.get_event_loop()
-        dataset = loop.run_until_complete(asyncfunc())
-        return dataset
+        return asyncio.run(self.get_dataset_async(name))
 
     async def get_dataset_async(self, name: str) -> Optional[Dataset]:
         """
         Get a dataset by its ID asynchronously.
         """
-        response = await self._async_client.get(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.get(
             url=f"datasets/{name}",
             headers=self.__headers__(),
         )
@@ -406,7 +408,8 @@ async def delete_dataset_async(self, name: str):
         """
         Delete a dataset by its ID asynchronously.
         """
-        response = await self._async_client.delete(
+        client = httpx.AsyncClient(base_url=DOC_AI_BASE_URL, timeout=None)
+        response = await client.delete(
             url=f"datasets/{name}",
             headers=self.__headers__(),
         )