feat(api): manual updates

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 17
+configured_endpoints: 18
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/supermemory--inc%2Fsupermemory-new-d52acd1a525b4bfe9f4befcc3a645f5d1289d75e7bad999cf1330e539b2ed84e.yml
 openapi_spec_hash: c34df5406cfa4d245812d30f99d28116
-config_hash: 38ef57207cda6bd69b47a02098e2fdb0
+config_hash: f305e457fd9ddd1dec2614b14d7936a5

README.md

@@ -32,7 +32,7 @@ client = Supermemory(
     api_key=os.environ.get("SUPERMEMORY_API_KEY"),  # This is the default and can be omitted
 )
 
-response = client.search.execute(
+response = client.search.documents(
     q="documents related to python",
 )
 print(response.results)
@@ -58,7 +58,7 @@ client = AsyncSupermemory(
 
 
 async def main() -> None:
-    response = await client.search.execute(
+    response = await client.search.documents(
         q="documents related to python",
     )
     print(response.results)
@@ -93,7 +93,7 @@ async def main() -> None:
         api_key="My API Key",
         http_client=DefaultAioHttpClient(),
     ) as client:
-        response = await client.search.execute(
+        response = await client.search.documents(
             q="documents related to python",
         )
         print(response.results)
@@ -111,6 +111,22 @@ Nested request parameters are [TypedDicts](https://docs.python.org/3/library/typ
 
 Typed requests and responses provide autocomplete and documentation within your editor. If you would like to see type errors in VS Code to help catch bugs earlier, set `python.analysis.typeCheckingMode` to `basic`.
 
+## Nested params
+
+Nested parameters are dictionaries, typed using `TypedDict`, for example:
+
+```python
+from supermemory import Supermemory
+
+client = Supermemory()
+
+response = client.search.memories(
+    q="machine learning concepts",
+    include={},
+)
+print(response.include)
+```
+
 ## File uploads
 
 Request parameters that correspond to file uploads can be passed as `bytes`, or a [`PathLike`](https://docs.python.org/3/library/os.html#os.PathLike) instance or a tuple of `(filename, contents, media type)`.

api.md

@@ -26,12 +26,13 @@ Methods:
 Types:
 
 ```python
-from supermemory.types import SearchExecuteResponse
+from supermemory.types import SearchDocumentsResponse, SearchMemoriesResponse
 ```
 
 Methods:
 
-- <code title="post /v3/search">client.search.<a href="./src/supermemory/resources/search.py">execute</a>(\*\*<a href="src/supermemory/types/search_execute_params.py">params</a>) -> <a href="./src/supermemory/types/search_execute_response.py">SearchExecuteResponse</a></code>
+- <code title="post /v3/search">client.search.<a href="./src/supermemory/resources/search.py">documents</a>(\*\*<a href="src/supermemory/types/search_documents_params.py">params</a>) -> <a href="./src/supermemory/types/search_documents_response.py">SearchDocumentsResponse</a></code>
+- <code title="post /v4/search">client.search.<a href="./src/supermemory/resources/search.py">memories</a>(\*\*<a href="src/supermemory/types/search_memories_params.py">params</a>) -> <a href="./src/supermemory/types/search_memories_response.py">SearchMemoriesResponse</a></code>
 
 # Settings
 

src/supermemory/resources/search.py

@@ -7,7 +7,7 @@
 
 import httpx
 
-from ..types import search_execute_params
+from ..types import search_memories_params, search_documents_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -19,7 +19,8 @@
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
-from ..types.search_execute_response import SearchExecuteResponse
+from ..types.search_memories_response import SearchMemoriesResponse
+from ..types.search_documents_response import SearchDocumentsResponse
 
 __all__ = ["SearchResource", "AsyncSearchResource"]
 
@@ -44,7 +45,7 @@ def with_streaming_response(self) -> SearchResourceWithStreamingResponse:
         """
         return SearchResourceWithStreamingResponse(self)
 
-    def execute(
+    def documents(
         self,
         *,
         q: str,
@@ -53,7 +54,7 @@ def execute(
         container_tags: List[str] | NotGiven = NOT_GIVEN,
         doc_id: str | NotGiven = NOT_GIVEN,
         document_threshold: float | NotGiven = NOT_GIVEN,
-        filters: search_execute_params.Filters | NotGiven = NOT_GIVEN,
+        filters: search_documents_params.Filters | NotGiven = NOT_GIVEN,
         include_full_docs: bool | NotGiven = NOT_GIVEN,
         include_summary: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
@@ -66,7 +67,7 @@ def execute(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SearchExecuteResponse:
+    ) -> SearchDocumentsResponse:
         """
         Search memories with advanced filtering
 
@@ -135,12 +136,82 @@ def execute(
                     "rerank": rerank,
                     "rewrite_query": rewrite_query,
                 },
-                search_execute_params.SearchExecuteParams,
+                search_documents_params.SearchDocumentsParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=SearchExecuteResponse,
+            cast_to=SearchDocumentsResponse,
+        )
+
+    def memories(
+        self,
+        *,
+        q: str,
+        container_tag: str | NotGiven = NOT_GIVEN,
+        filters: search_memories_params.Filters | NotGiven = NOT_GIVEN,
+        include: search_memories_params.Include | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        rerank: bool | NotGiven = NOT_GIVEN,
+        rewrite_query: bool | NotGiven = NOT_GIVEN,
+        threshold: float | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SearchMemoriesResponse:
+        """
+        Search memory entries - Low latency for conversational
+
+        Args:
+          q: Search query string
+
+          container_tag: Optional tag this search should be containerized by. This can be an ID for your
+              user, a project ID, or any other identifier you wish to use to filter memories.
+
+          filters: Optional filters to apply to the search
+
+          limit: Maximum number of results to return
+
+          rerank: If true, rerank the results based on the query. This is helpful if you want to
+              ensure the most relevant results are returned.
+
+          rewrite_query: If true, rewrites the query to make it easier to find documents. This increases
+              the latency by about 400ms
+
+          threshold: Threshold / sensitivity for memories selection. 0 is least sensitive (returns
+              most memories, more results), 1 is most sensitive (returns lesser memories,
+              accurate results)
+
+          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
+        """
+        return self._post(
+            "/v4/search",
+            body=maybe_transform(
+                {
+                    "q": q,
+                    "container_tag": container_tag,
+                    "filters": filters,
+                    "include": include,
+                    "limit": limit,
+                    "rerank": rerank,
+                    "rewrite_query": rewrite_query,
+                    "threshold": threshold,
+                },
+                search_memories_params.SearchMemoriesParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SearchMemoriesResponse,
         )
 
 
@@ -164,7 +235,7 @@ def with_streaming_response(self) -> AsyncSearchResourceWithStreamingResponse:
         """
         return AsyncSearchResourceWithStreamingResponse(self)
 
-    async def execute(
+    async def documents(
         self,
         *,
         q: str,
@@ -173,7 +244,7 @@ async def execute(
         container_tags: List[str] | NotGiven = NOT_GIVEN,
         doc_id: str | NotGiven = NOT_GIVEN,
         document_threshold: float | NotGiven = NOT_GIVEN,
-        filters: search_execute_params.Filters | NotGiven = NOT_GIVEN,
+        filters: search_documents_params.Filters | NotGiven = NOT_GIVEN,
         include_full_docs: bool | NotGiven = NOT_GIVEN,
         include_summary: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
@@ -186,7 +257,7 @@ async def execute(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SearchExecuteResponse:
+    ) -> SearchDocumentsResponse:
         """
         Search memories with advanced filtering
 
@@ -255,46 +326,128 @@ async def execute(
                     "rerank": rerank,
                     "rewrite_query": rewrite_query,
                 },
-                search_execute_params.SearchExecuteParams,
+                search_documents_params.SearchDocumentsParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SearchDocumentsResponse,
+        )
+
+    async def memories(
+        self,
+        *,
+        q: str,
+        container_tag: str | NotGiven = NOT_GIVEN,
+        filters: search_memories_params.Filters | NotGiven = NOT_GIVEN,
+        include: search_memories_params.Include | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        rerank: bool | NotGiven = NOT_GIVEN,
+        rewrite_query: bool | NotGiven = NOT_GIVEN,
+        threshold: float | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> SearchMemoriesResponse:
+        """
+        Search memory entries - Low latency for conversational
+
+        Args:
+          q: Search query string
+
+          container_tag: Optional tag this search should be containerized by. This can be an ID for your
+              user, a project ID, or any other identifier you wish to use to filter memories.
+
+          filters: Optional filters to apply to the search
+
+          limit: Maximum number of results to return
+
+          rerank: If true, rerank the results based on the query. This is helpful if you want to
+              ensure the most relevant results are returned.
+
+          rewrite_query: If true, rewrites the query to make it easier to find documents. This increases
+              the latency by about 400ms
+
+          threshold: Threshold / sensitivity for memories selection. 0 is least sensitive (returns
+              most memories, more results), 1 is most sensitive (returns lesser memories,
+              accurate results)
+
+          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
+        """
+        return await self._post(
+            "/v4/search",
+            body=await async_maybe_transform(
+                {
+                    "q": q,
+                    "container_tag": container_tag,
+                    "filters": filters,
+                    "include": include,
+                    "limit": limit,
+                    "rerank": rerank,
+                    "rewrite_query": rewrite_query,
+                    "threshold": threshold,
+                },
+                search_memories_params.SearchMemoriesParams,
             ),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=SearchExecuteResponse,
+            cast_to=SearchMemoriesResponse,
         )
 
 
 class SearchResourceWithRawResponse:
     def __init__(self, search: SearchResource) -> None:
         self._search = search
 
-        self.execute = to_raw_response_wrapper(
-            search.execute,
+        self.documents = to_raw_response_wrapper(
+            search.documents,
+        )
+        self.memories = to_raw_response_wrapper(
+            search.memories,
         )
 
 
 class AsyncSearchResourceWithRawResponse:
     def __init__(self, search: AsyncSearchResource) -> None:
         self._search = search
 
-        self.execute = async_to_raw_response_wrapper(
-            search.execute,
+        self.documents = async_to_raw_response_wrapper(
+            search.documents,
+        )
+        self.memories = async_to_raw_response_wrapper(
+            search.memories,
         )
 
 
 class SearchResourceWithStreamingResponse:
     def __init__(self, search: SearchResource) -> None:
         self._search = search
 
-        self.execute = to_streamed_response_wrapper(
-            search.execute,
+        self.documents = to_streamed_response_wrapper(
+            search.documents,
+        )
+        self.memories = to_streamed_response_wrapper(
+            search.memories,
         )
 
 
 class AsyncSearchResourceWithStreamingResponse:
     def __init__(self, search: AsyncSearchResource) -> None:
         self._search = search
 
-        self.execute = async_to_streamed_response_wrapper(
-            search.execute,
+        self.documents = async_to_streamed_response_wrapper(
+            search.documents,
+        )
+        self.memories = async_to_streamed_response_wrapper(
+            search.memories,
         )

src/supermemory/types/__init__.py

@@ -9,16 +9,18 @@
 from .memory_list_response import MemoryListResponse as MemoryListResponse
 from .memory_update_params import MemoryUpdateParams as MemoryUpdateParams
 from .setting_get_response import SettingGetResponse as SettingGetResponse
-from .search_execute_params import SearchExecuteParams as SearchExecuteParams
 from .setting_update_params import SettingUpdateParams as SettingUpdateParams
 from .connection_list_params import ConnectionListParams as ConnectionListParams
 from .memory_update_response import MemoryUpdateResponse as MemoryUpdateResponse
-from .search_execute_response import SearchExecuteResponse as SearchExecuteResponse
+from .search_memories_params import SearchMemoriesParams as SearchMemoriesParams
+from .search_documents_params import SearchDocumentsParams as SearchDocumentsParams
 from .setting_update_response import SettingUpdateResponse as SettingUpdateResponse
 from .connection_create_params import ConnectionCreateParams as ConnectionCreateParams
 from .connection_import_params import ConnectionImportParams as ConnectionImportParams
 from .connection_list_response import ConnectionListResponse as ConnectionListResponse
+from .search_memories_response import SearchMemoriesResponse as SearchMemoriesResponse
 from .memory_upload_file_params import MemoryUploadFileParams as MemoryUploadFileParams
+from .search_documents_response import SearchDocumentsResponse as SearchDocumentsResponse
 from .connection_create_response import ConnectionCreateResponse as ConnectionCreateResponse
 from .memory_upload_file_response import MemoryUploadFileResponse as MemoryUploadFileResponse
 from .connection_get_by_id_response import ConnectionGetByIDResponse as ConnectionGetByIDResponse