llama-cloud 0.1.37__py3-none-any.whl → 0.1.38__py3-none-any.whl

llama_cloud/resources/data_sources/client.py

@@ -33,7 +33,11 @@ class DataSourcesClient:
         self._client_wrapper = client_wrapper
 
     def list_data_sources(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[DataSource]:
         """
         List data sources for a given project.
@@ -43,6 +47,8 @@ class DataSourcesClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -55,7 +61,7 @@ class DataSourcesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -74,6 +80,7 @@ class DataSourcesClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSourceCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSource:
         """
         Create a new data source.
@@ -84,6 +91,8 @@ class DataSourcesClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSourceCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
         from llama_cloud.client import LlamaCloud
@@ -103,7 +112,7 @@ class DataSourcesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -122,6 +131,7 @@ class DataSourcesClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSourceCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSource:
         """
         Upserts a data source.
@@ -133,6 +143,8 @@ class DataSourcesClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSourceCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
         from llama_cloud.client import LlamaCloud
@@ -152,7 +164,7 @@ class DataSourcesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -293,7 +305,11 @@ class AsyncDataSourcesClient:
         self._client_wrapper = client_wrapper
 
     async def list_data_sources(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[DataSource]:
         """
         List data sources for a given project.
@@ -303,6 +319,8 @@ class AsyncDataSourcesClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -315,7 +333,7 @@ class AsyncDataSourcesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -334,6 +352,7 @@ class AsyncDataSourcesClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSourceCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSource:
         """
         Create a new data source.
@@ -344,6 +363,8 @@ class AsyncDataSourcesClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSourceCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
         from llama_cloud.client import AsyncLlamaCloud
@@ -363,7 +384,7 @@ class AsyncDataSourcesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -382,6 +403,7 @@ class AsyncDataSourcesClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSourceCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSource:
         """
         Upserts a data source.
@@ -393,6 +415,8 @@ class AsyncDataSourcesClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSourceCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
         from llama_cloud.client import AsyncLlamaCloud
@@ -412,7 +436,7 @@ class AsyncDataSourcesClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:

llama_cloud/resources/data_sources/types/data_source_update_component.py

@@ -6,6 +6,7 @@ from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSo
 from ....types.cloud_box_data_source import CloudBoxDataSource
 from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
 from ....types.cloud_jira_data_source import CloudJiraDataSource
+from ....types.cloud_jira_data_source_v_2 import CloudJiraDataSourceV2
 from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
 from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
 from ....types.cloud_s_3_data_source import CloudS3DataSource
@@ -22,5 +23,6 @@ DataSourceUpdateComponent = typing.Union[
     CloudNotionPageDataSource,
     CloudConfluenceDataSource,
     CloudJiraDataSource,
+    CloudJiraDataSourceV2,
     CloudBoxDataSource,
 ]

llama_cloud/resources/embedding_model_configs/client.py

@@ -31,13 +31,19 @@ class EmbeddingModelConfigsClient:
         self._client_wrapper = client_wrapper
 
     def list_embedding_model_configs(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[EmbeddingModelConfig]:
         """
         Parameters:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -50,7 +56,7 @@ class EmbeddingModelConfigsClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -70,6 +76,7 @@ class EmbeddingModelConfigsClient:
         organization_id: typing.Optional[str] = None,
         name: str,
         embedding_config: EmbeddingModelConfigCreateEmbeddingConfig,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Create a new embedding model configuration within a specified project.
@@ -82,13 +89,15 @@ class EmbeddingModelConfigsClient:
             - name: str. The name of the embedding model config.
 
             - embedding_config: EmbeddingModelConfigCreateEmbeddingConfig. The embedding configuration for the embedding model config.
+
+            - project_id: typing.Optional[str].
         """
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"name": name, "embedding_config": embedding_config}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -107,6 +116,7 @@ class EmbeddingModelConfigsClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: EmbeddingModelConfigUpdate,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Upserts an embedding model config.
@@ -118,13 +128,15 @@ class EmbeddingModelConfigsClient:
             - organization_id: typing.Optional[str].
 
             - request: EmbeddingModelConfigUpdate.
+
+            - project_id: typing.Optional[str].
         """
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -144,6 +156,7 @@ class EmbeddingModelConfigsClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: EmbeddingModelConfigUpdate,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Update an embedding model config by ID.
@@ -156,6 +169,8 @@ class EmbeddingModelConfigsClient:
             - organization_id: typing.Optional[str].
 
             - request: EmbeddingModelConfigUpdate.
+
+            - project_id: typing.Optional[str].
         """
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
@@ -164,7 +179,7 @@ class EmbeddingModelConfigsClient:
             ),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -183,6 +198,7 @@ class EmbeddingModelConfigsClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> None:
         """
         Delete an embedding model config by ID.
@@ -193,6 +209,8 @@ class EmbeddingModelConfigsClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -209,7 +227,7 @@ class EmbeddingModelConfigsClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/embedding-model-configs/{embedding_model_config_id}"
             ),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -228,13 +246,19 @@ class AsyncEmbeddingModelConfigsClient:
         self._client_wrapper = client_wrapper
 
     async def list_embedding_model_configs(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[EmbeddingModelConfig]:
         """
         Parameters:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -247,7 +271,7 @@ class AsyncEmbeddingModelConfigsClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -267,6 +291,7 @@ class AsyncEmbeddingModelConfigsClient:
         organization_id: typing.Optional[str] = None,
         name: str,
         embedding_config: EmbeddingModelConfigCreateEmbeddingConfig,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Create a new embedding model configuration within a specified project.
@@ -279,13 +304,15 @@ class AsyncEmbeddingModelConfigsClient:
             - name: str. The name of the embedding model config.
 
             - embedding_config: EmbeddingModelConfigCreateEmbeddingConfig. The embedding configuration for the embedding model config.
+
+            - project_id: typing.Optional[str].
         """
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"name": name, "embedding_config": embedding_config}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -304,6 +331,7 @@ class AsyncEmbeddingModelConfigsClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: EmbeddingModelConfigUpdate,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Upserts an embedding model config.
@@ -315,13 +343,15 @@ class AsyncEmbeddingModelConfigsClient:
             - organization_id: typing.Optional[str].
 
             - request: EmbeddingModelConfigUpdate.
+
+            - project_id: typing.Optional[str].
         """
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/embedding-model-configs"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -341,6 +371,7 @@ class AsyncEmbeddingModelConfigsClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: EmbeddingModelConfigUpdate,
+        project_id: typing.Optional[str] = None,
     ) -> EmbeddingModelConfig:
         """
         Update an embedding model config by ID.
@@ -353,6 +384,8 @@ class AsyncEmbeddingModelConfigsClient:
             - organization_id: typing.Optional[str].
 
             - request: EmbeddingModelConfigUpdate.
+
+            - project_id: typing.Optional[str].
         """
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
@@ -361,7 +394,7 @@ class AsyncEmbeddingModelConfigsClient:
             ),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -380,6 +413,7 @@ class AsyncEmbeddingModelConfigsClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> None:
         """
         Delete an embedding model config by ID.
@@ -390,6 +424,8 @@ class AsyncEmbeddingModelConfigsClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -406,7 +442,7 @@ class AsyncEmbeddingModelConfigsClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/embedding-model-configs/{embedding_model_config_id}"
             ),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300: