supermemory 3.7.0__py3-none-any.whl → 3.19.0__py3-none-any.whl

supermemory/resources/connections.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
+from typing import Dict, Union, Iterable, Optional
 from typing_extensions import Literal
 
 import httpx
@@ -11,7 +11,9 @@ from ..types import (
     connection_list_params,
     connection_create_params,
     connection_import_params,
-    connection_get_by_tags_params,
+    connection_configure_params,
+    connection_resources_params,
+    connection_get_by_tag_params,
     connection_list_documents_params,
     connection_delete_by_provider_params,
 )
@@ -28,8 +30,10 @@ from .._response import (
 from .._base_client import make_request_options
 from ..types.connection_list_response import ConnectionListResponse
 from ..types.connection_create_response import ConnectionCreateResponse
+from ..types.connection_configure_response import ConnectionConfigureResponse
 from ..types.connection_get_by_id_response import ConnectionGetByIDResponse
-from ..types.connection_get_by_tags_response import ConnectionGetByTagsResponse
+from ..types.connection_resources_response import ConnectionResourcesResponse
+from ..types.connection_get_by_tag_response import ConnectionGetByTagResponse
 from ..types.connection_delete_by_id_response import ConnectionDeleteByIDResponse
 from ..types.connection_list_documents_response import ConnectionListDocumentsResponse
 from ..types.connection_delete_by_provider_response import ConnectionDeleteByProviderResponse
@@ -59,7 +63,7 @@ class ConnectionsResource(SyncAPIResource):
 
     def create(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         document_limit: int | Omit = omit,
@@ -137,6 +141,41 @@ class ConnectionsResource(SyncAPIResource):
             cast_to=ConnectionListResponse,
         )
 
+    def configure(
+        self,
+        connection_id: str,
+        *,
+        resources: Iterable[Dict[str, object]],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConnectionConfigureResponse:
+        """
+        Configure resources for a connection (supported providers: GitHub for now)
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not connection_id:
+            raise ValueError(f"Expected a non-empty value for `connection_id` but received {connection_id!r}")
+        return self._post(
+            f"/v3/connections/{connection_id}/configure",
+            body=maybe_transform({"resources": resources}, connection_configure_params.ConnectionConfigureParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ConnectionConfigureResponse,
+        )
+
     def delete_by_id(
         self,
         connection_id: str,
@@ -172,7 +211,7 @@ class ConnectionsResource(SyncAPIResource):
 
     def delete_by_provider(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -243,9 +282,9 @@ class ConnectionsResource(SyncAPIResource):
             cast_to=ConnectionGetByIDResponse,
         )
 
-    def get_by_tags(
+    def get_by_tag(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -254,7 +293,7 @@ class ConnectionsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> ConnectionGetByTagsResponse:
+    ) -> ConnectionGetByTagResponse:
         """
         Get connection details with provider and container tags
 
@@ -274,17 +313,17 @@ class ConnectionsResource(SyncAPIResource):
         return self._post(
             f"/v3/connections/{provider}/connection",
             body=maybe_transform(
-                {"container_tags": container_tags}, connection_get_by_tags_params.ConnectionGetByTagsParams
+                {"container_tags": container_tags}, connection_get_by_tag_params.ConnectionGetByTagParams
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ConnectionGetByTagsResponse,
+            cast_to=ConnectionGetByTagResponse,
         )
 
     def import_(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -322,7 +361,7 @@ class ConnectionsResource(SyncAPIResource):
 
     def list_documents(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -359,6 +398,51 @@ class ConnectionsResource(SyncAPIResource):
             cast_to=ConnectionListDocumentsResponse,
         )
 
+    def resources(
+        self,
+        connection_id: str,
+        *,
+        page: float | Omit = omit,
+        per_page: float | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConnectionResourcesResponse:
+        """
+        Fetch resources for a connection (supported providers: GitHub for now)
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not connection_id:
+            raise ValueError(f"Expected a non-empty value for `connection_id` but received {connection_id!r}")
+        return self._get(
+            f"/v3/connections/{connection_id}/resources",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                    },
+                    connection_resources_params.ConnectionResourcesParams,
+                ),
+            ),
+            cast_to=ConnectionResourcesResponse,
+        )
+
 
 class AsyncConnectionsResource(AsyncAPIResource):
     @cached_property
@@ -382,7 +466,7 @@ class AsyncConnectionsResource(AsyncAPIResource):
 
     async def create(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         document_limit: int | Omit = omit,
@@ -462,6 +546,43 @@ class AsyncConnectionsResource(AsyncAPIResource):
             cast_to=ConnectionListResponse,
         )
 
+    async def configure(
+        self,
+        connection_id: str,
+        *,
+        resources: Iterable[Dict[str, object]],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConnectionConfigureResponse:
+        """
+        Configure resources for a connection (supported providers: GitHub for now)
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not connection_id:
+            raise ValueError(f"Expected a non-empty value for `connection_id` but received {connection_id!r}")
+        return await self._post(
+            f"/v3/connections/{connection_id}/configure",
+            body=await async_maybe_transform(
+                {"resources": resources}, connection_configure_params.ConnectionConfigureParams
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=ConnectionConfigureResponse,
+        )
+
     async def delete_by_id(
         self,
         connection_id: str,
@@ -497,7 +618,7 @@ class AsyncConnectionsResource(AsyncAPIResource):
 
     async def delete_by_provider(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -568,9 +689,9 @@ class AsyncConnectionsResource(AsyncAPIResource):
             cast_to=ConnectionGetByIDResponse,
         )
 
-    async def get_by_tags(
+    async def get_by_tag(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str],
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -579,7 +700,7 @@ class AsyncConnectionsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> ConnectionGetByTagsResponse:
+    ) -> ConnectionGetByTagResponse:
         """
         Get connection details with provider and container tags
 
@@ -599,17 +720,17 @@ class AsyncConnectionsResource(AsyncAPIResource):
         return await self._post(
             f"/v3/connections/{provider}/connection",
             body=await async_maybe_transform(
-                {"container_tags": container_tags}, connection_get_by_tags_params.ConnectionGetByTagsParams
+                {"container_tags": container_tags}, connection_get_by_tag_params.ConnectionGetByTagParams
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=ConnectionGetByTagsResponse,
+            cast_to=ConnectionGetByTagResponse,
         )
 
     async def import_(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -649,7 +770,7 @@ class AsyncConnectionsResource(AsyncAPIResource):
 
     async def list_documents(
         self,
-        provider: Literal["notion", "google-drive", "onedrive", "web-crawler"],
+        provider: Literal["notion", "google-drive", "onedrive", "gmail", "github", "web-crawler", "s3"],
         *,
         container_tags: SequenceNotStr[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -686,6 +807,51 @@ class AsyncConnectionsResource(AsyncAPIResource):
             cast_to=ConnectionListDocumentsResponse,
         )
 
+    async def resources(
+        self,
+        connection_id: str,
+        *,
+        page: float | Omit = omit,
+        per_page: float | Omit = omit,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
+    ) -> ConnectionResourcesResponse:
+        """
+        Fetch resources for a connection (supported providers: GitHub for now)
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        if not connection_id:
+            raise ValueError(f"Expected a non-empty value for `connection_id` but received {connection_id!r}")
+        return await self._get(
+            f"/v3/connections/{connection_id}/resources",
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "page": page,
+                        "per_page": per_page,
+                    },
+                    connection_resources_params.ConnectionResourcesParams,
+                ),
+            ),
+            cast_to=ConnectionResourcesResponse,
+        )
+
 
 class ConnectionsResourceWithRawResponse:
     def __init__(self, connections: ConnectionsResource) -> None:
@@ -697,6 +863,9 @@ class ConnectionsResourceWithRawResponse:
         self.list = to_raw_response_wrapper(
             connections.list,
         )
+        self.configure = to_raw_response_wrapper(
+            connections.configure,
+        )
         self.delete_by_id = to_raw_response_wrapper(
             connections.delete_by_id,
         )
@@ -706,8 +875,8 @@ class ConnectionsResourceWithRawResponse:
         self.get_by_id = to_raw_response_wrapper(
             connections.get_by_id,
         )
-        self.get_by_tags = to_raw_response_wrapper(
-            connections.get_by_tags,
+        self.get_by_tag = to_raw_response_wrapper(
+            connections.get_by_tag,
         )
         self.import_ = to_raw_response_wrapper(
             connections.import_,
@@ -715,6 +884,9 @@ class ConnectionsResourceWithRawResponse:
         self.list_documents = to_raw_response_wrapper(
             connections.list_documents,
         )
+        self.resources = to_raw_response_wrapper(
+            connections.resources,
+        )
 
 
 class AsyncConnectionsResourceWithRawResponse:
@@ -727,6 +899,9 @@ class AsyncConnectionsResourceWithRawResponse:
         self.list = async_to_raw_response_wrapper(
             connections.list,
         )
+        self.configure = async_to_raw_response_wrapper(
+            connections.configure,
+        )
         self.delete_by_id = async_to_raw_response_wrapper(
             connections.delete_by_id,
         )
@@ -736,8 +911,8 @@ class AsyncConnectionsResourceWithRawResponse:
         self.get_by_id = async_to_raw_response_wrapper(
             connections.get_by_id,
         )
-        self.get_by_tags = async_to_raw_response_wrapper(
-            connections.get_by_tags,
+        self.get_by_tag = async_to_raw_response_wrapper(
+            connections.get_by_tag,
         )
         self.import_ = async_to_raw_response_wrapper(
             connections.import_,
@@ -745,6 +920,9 @@ class AsyncConnectionsResourceWithRawResponse:
         self.list_documents = async_to_raw_response_wrapper(
             connections.list_documents,
         )
+        self.resources = async_to_raw_response_wrapper(
+            connections.resources,
+        )
 
 
 class ConnectionsResourceWithStreamingResponse:
@@ -757,6 +935,9 @@ class ConnectionsResourceWithStreamingResponse:
         self.list = to_streamed_response_wrapper(
             connections.list,
         )
+        self.configure = to_streamed_response_wrapper(
+            connections.configure,
+        )
         self.delete_by_id = to_streamed_response_wrapper(
             connections.delete_by_id,
         )
@@ -766,8 +947,8 @@ class ConnectionsResourceWithStreamingResponse:
         self.get_by_id = to_streamed_response_wrapper(
             connections.get_by_id,
         )
-        self.get_by_tags = to_streamed_response_wrapper(
-            connections.get_by_tags,
+        self.get_by_tag = to_streamed_response_wrapper(
+            connections.get_by_tag,
         )
         self.import_ = to_streamed_response_wrapper(
             connections.import_,
@@ -775,6 +956,9 @@ class ConnectionsResourceWithStreamingResponse:
         self.list_documents = to_streamed_response_wrapper(
             connections.list_documents,
         )
+        self.resources = to_streamed_response_wrapper(
+            connections.resources,
+        )
 
 
 class AsyncConnectionsResourceWithStreamingResponse:
@@ -787,6 +971,9 @@ class AsyncConnectionsResourceWithStreamingResponse:
         self.list = async_to_streamed_response_wrapper(
             connections.list,
         )
+        self.configure = async_to_streamed_response_wrapper(
+            connections.configure,
+        )
         self.delete_by_id = async_to_streamed_response_wrapper(
             connections.delete_by_id,
         )
@@ -796,8 +983,8 @@ class AsyncConnectionsResourceWithStreamingResponse:
         self.get_by_id = async_to_streamed_response_wrapper(
             connections.get_by_id,
         )
-        self.get_by_tags = async_to_streamed_response_wrapper(
-            connections.get_by_tags,
+        self.get_by_tag = async_to_streamed_response_wrapper(
+            connections.get_by_tag,
         )
         self.import_ = async_to_streamed_response_wrapper(
             connections.import_,
@@ -805,3 +992,6 @@ class AsyncConnectionsResourceWithStreamingResponse:
         self.list_documents = async_to_streamed_response_wrapper(
             connections.list_documents,
         )
+        self.resources = async_to_streamed_response_wrapper(
+            connections.resources,
+        )