llama-cloud 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -1040,6 +1040,52 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def update_pipeline_data_source(
+        self, pipeline_id: str, data_source_id: str, *, sync_interval: typing.Optional[float] = OMIT
+    ) -> PipelineDataSource:
+        """
+        Update the configuration of a data source in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+
+            - sync_interval: typing.Optional[float]. The interval at which the data source should be synced.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.update_pipeline_data_source(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if sync_interval is not OMIT:
+            _request["sync_interval"] = sync_interval
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}",
+            ),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PipelineDataSource, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def delete_pipeline_data_source(self, pipeline_id: str, data_source_id: str) -> None:
         """
         Delete a data source from a pipeline.
@@ -1116,6 +1162,44 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_pipeline_data_source_status(self, pipeline_id: str, data_source_id: str) -> ManagedIngestionStatusResponse:
+        """
+        Get the status of a data source for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.get_pipeline_data_source_status(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatusResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def run_search(
         self,
         pipeline_id: str,
@@ -2652,6 +2736,52 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def update_pipeline_data_source(
+        self, pipeline_id: str, data_source_id: str, *, sync_interval: typing.Optional[float] = OMIT
+    ) -> PipelineDataSource:
+        """
+        Update the configuration of a data source in a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+
+            - sync_interval: typing.Optional[float]. The interval at which the data source should be synced.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.update_pipeline_data_source(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if sync_interval is not OMIT:
+            _request["sync_interval"] = sync_interval
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}",
+            ),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PipelineDataSource, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def delete_pipeline_data_source(self, pipeline_id: str, data_source_id: str) -> None:
         """
         Delete a data source from a pipeline.
@@ -2728,6 +2858,46 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def get_pipeline_data_source_status(
+        self, pipeline_id: str, data_source_id: str
+    ) -> ManagedIngestionStatusResponse:
+        """
+        Get the status of a data source for a pipeline.
+
+        Parameters:
+            - pipeline_id: str.
+
+            - data_source_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.get_pipeline_data_source_status(
+            pipeline_id="string",
+            data_source_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/",
+                f"api/v1/pipelines/{pipeline_id}/data-sources/{data_source_id}/status",
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ManagedIngestionStatusResponse, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def run_search(
         self,
         pipeline_id: str,

llama_cloud/types/pipeline_data_source.py

@@ -34,6 +34,13 @@ class PipelineDataSource(pydantic.BaseModel):
     project_id: str
     data_source_id: str = pydantic.Field(description="The ID of the data source.")
     pipeline_id: str = pydantic.Field(description="The ID of the pipeline.")
+    last_synced_at: dt.datetime = pydantic.Field(description="The last time the data source was automatically synced.")
+    sync_interval: typing.Optional[float] = pydantic.Field(
+        description="The interval at which the data source should be synced."
+    )
+    sync_schedule_set_by: typing.Optional[str] = pydantic.Field(
+        description="The id of the user who set the sync schedule."
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_data_source_create.py

@@ -20,6 +20,9 @@ class PipelineDataSourceCreate(pydantic.BaseModel):
     """
 
     data_source_id: str = pydantic.Field(description="The ID of the data source.")
+    sync_interval: typing.Optional[float] = pydantic.Field(
+        description="The interval at which the data source should be synced."
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

{llama_cloud-0.0.14.dist-info → llama_cloud-0.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.0.14
+Version: 0.0.15
 Summary: 
 Author: Logan Markewich
 Author-email: logan@runllama.ai

{llama_cloud-0.0.14.dist-info → llama_cloud-0.0.15.dist-info}/RECORD

@@ -41,7 +41,7 @@ llama_cloud/resources/organizations/client.py,sha256=akn_3sytJW_VhuLVBbP0TKiKKbB
 llama_cloud/resources/parsing/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 llama_cloud/resources/parsing/client.py,sha256=epdS59BEsxx9nQryTV_Eemd3RAhESyMN2K4uP2gSpPQ,40700
 llama_cloud/resources/pipelines/__init__.py,sha256=Ww7n75XUkq-aqy7WuxhO9rRbBmi_VsYMpkZpDyw8oYs,1147
-llama_cloud/resources/pipelines/client.py,sha256=iDfmxzrxHpmJo9sKaFrHLStM0bvPKKeNGsgbM3dc8h8,129938
+llama_cloud/resources/pipelines/client.py,sha256=Ek_eSin0f32Fucf7BhfAqKT8xgJWagLt_np9oYZdcqI,136672
 llama_cloud/resources/pipelines/types/__init__.py,sha256=PFl5hsq0GHQRwZLR-L7igQ5NBO8GH_rJPOcRR_16ODk,1275
 llama_cloud/resources/pipelines/types/pipeline_file_update_custom_metadata_value.py,sha256=trI48WLxPcAqV9207Q6-3cj1nl4EGlZpw7En56ZsPgg,217
 llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py,sha256=0XHOqSS-yiSREOi8-kE9yjgKNzxqzAcxDhHDHfDHEkk,2494
@@ -177,10 +177,10 @@ llama_cloud/types/pipeline.py,sha256=u1EoHhP9UQab3y0rolNiLEd51JqzxfAZeMyRDPencE8
 llama_cloud/types/pipeline_create.py,sha256=Dv9wh5Uu4MoITP6bunhaem1ToRcs3Fw2gr3fw7HO7Fk,3217
 llama_cloud/types/pipeline_create_embedding_config.py,sha256=Lu03mWVS7XrqvhSsjV4H2OE69Qf32kw7k67ZoV-P5Kg,2440
 llama_cloud/types/pipeline_create_transform_config.py,sha256=CiMil0NrwvxR34CAzrSWw9Uo0117tz409sptH1k_r48,854
-llama_cloud/types/pipeline_data_source.py,sha256=A3AlRzTD7zr1y-u5O5LFESqIupbbG-fqUndQgeYj77w,2062
+llama_cloud/types/pipeline_data_source.py,sha256=uiTu6BkXgizhkuqr6GHiS8ZBhtnLcwcitMFkwS6woaE,2465
 llama_cloud/types/pipeline_data_source_component.py,sha256=Pk_K0Gv7xSWe5BKCdxz82EFd6AQDvZGN-6t3zg9h8NY,265
 llama_cloud/types/pipeline_data_source_component_one.py,sha256=W9ntkcrg6bNOJgSe1GCUX8AjnY0RDwBYo9QQiFWGZio,951
-llama_cloud/types/pipeline_data_source_create.py,sha256=dAxf2mHQTegDbev1MJnEpFEpOpgRhj2sCnnKtTit8tQ,1136
+llama_cloud/types/pipeline_data_source_create.py,sha256=0QPQNT6dvLaO5bZGX4QJWo5-2T44dQRjs2R5HwDaFa4,1280
 llama_cloud/types/pipeline_data_source_custom_metadata_value.py,sha256=8n3r60sxMx4_udW0yzJZxzyWeK6L3cc2-jLGZFW4EDs,217
 llama_cloud/types/pipeline_deployment.py,sha256=3sWAIdeov3CYFZMCAWwCR46ShHA6XAzSqmc18qryHzM,1669
 llama_cloud/types/pipeline_embedding_config.py,sha256=ucK2AZdIgOYCQaR87Wt2H3Jq4OX25iG5JFovZeo4UQo,2362
@@ -221,7 +221,7 @@ llama_cloud/types/user_organization_create.py,sha256=YESlfcI64710OFdQzgGD4a7aItg
 llama_cloud/types/user_organization_delete.py,sha256=Z8RSRXc0AGAuGxv6eQPC2S1XIdRfNCXBggfEefgPseM,1209
 llama_cloud/types/validation_error.py,sha256=yZDLtjUHDY5w82Ra6CW0H9sLAr18R0RY1UNgJKR72DQ,1084
 llama_cloud/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
-llama_cloud-0.0.14.dist-info/LICENSE,sha256=_iNqtPcw1Ue7dZKwOwgPtbegMUkWVy15hC7bffAdNmY,1067
-llama_cloud-0.0.14.dist-info/METADATA,sha256=5UmRVYEZcSPcrsoqDsvEWTr8RkVLydnjyGAwpPyweuo,751
-llama_cloud-0.0.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_cloud-0.0.14.dist-info/RECORD,,
+llama_cloud-0.0.15.dist-info/LICENSE,sha256=_iNqtPcw1Ue7dZKwOwgPtbegMUkWVy15hC7bffAdNmY,1067
+llama_cloud-0.0.15.dist-info/METADATA,sha256=t8rU9koSxTTowXcLKluWOWVFVIkhKBJXvCU-9Kggfr8,751
+llama_cloud-0.0.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llama_cloud-0.0.15.dist-info/RECORD,,