feat(api): add memory support

Author: stainless-app[bot]

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
   lint:
     timeout-minutes: 10
     name: lint
-    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: 'ubuntu-latest'
     if: (github.event_name == 'push' || github.event.pull_request.head.repo.fork) && (github.event_name != 'push' || github.event.head_commit.message != 'codegen metadata')
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
@@ -44,7 +44,7 @@ jobs:
     permissions:
       contents: read
       id-token: write
-    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: 'ubuntu-latest'
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
 
@@ -64,7 +64,7 @@ jobs:
 
       - name: Get GitHub OIDC Token
         if: |-
-          github.repository == 'stainless-sdks/parallel-sdk-python' &&
+          github.repository == 'parallel-web/parallel-sdk-python-staging' &&
           !startsWith(github.ref, 'refs/heads/stl/')
         id: github-oidc
         uses: actions/github-script@ed597411d8f924073f98dfc5c65a23a2325f34cd # v8.0.0
@@ -73,7 +73,7 @@ jobs:
 
       - name: Upload tarball
         if: |-
-          github.repository == 'stainless-sdks/parallel-sdk-python' &&
+          github.repository == 'parallel-web/parallel-sdk-python-staging' &&
           !startsWith(github.ref, 'refs/heads/stl/')
         env:
           URL: https://pkg.stainless.com/s
@@ -84,7 +84,7 @@ jobs:
   test:
     timeout-minutes: 10
     name: test
-    runs-on: ${{ startsWith(github.repository, 'stainless-sdks/') && 'depot-ubuntu-24.04' || 'ubuntu-latest' }}
+    runs-on: 'ubuntu-latest'
     if: github.event_name == 'push' || github.event.pull_request.head.repo.fork
     steps:
       - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 30
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/parallel-web/parallel-sdk-3001a018c926803bb713f19d99c3d5d39ed44ce3fe670bebefc326fd252b8821.yml
-openapi_spec_hash: 0883a98aa5b566818f5910d01c831c4a
-config_hash: 8572a2d7a5ede9ade3be8ad4d8aafd7d
+configured_endpoints: 33
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/parallel-web/parallel-sdk-dcb1605f4878ba899f427518b5f86d2bb79fc37820caa27392695fe8b7f5295c.yml
+openapi_spec_hash: 6c7d3db50b53be72d8b35c260f771004
+config_hash: d4f69197dba970d05757624fe3f7b373

scripts/utils/upload-artifact.sh

@@ -20,7 +20,7 @@ UPLOAD_RESPONSE=$(curl -v -X PUT \
 
 if echo "$UPLOAD_RESPONSE" | grep -q "HTTP/[0-9.]* 200"; then
   echo -e "\033[32mUploaded build to Stainless storage.\033[0m"
-  echo -e "\033[32mInstallation: pip install 'https://pkg.stainless.com/s/parallel-sdk-python/$SHA/$FILENAME'\033[0m"
+  echo -e "\033[32mInstallation: pip install 'https://pkg.stainless.com/s/parallel-sdk-python-staging/$SHA/$FILENAME'\033[0m"
 else
   echo -e "\033[31mFailed to upload artifact.\033[0m"
   exit 1

src/parallel/_client.py

@@ -355,7 +355,7 @@ def search(
         advanced_settings: Optional[AdvancedSearchSettingsParam] | Omit = omit,
         client_model: Optional[str] | Omit = omit,
         max_chars_total: Optional[int] | Omit = omit,
-        mode: Optional[Literal["turbo", "basic", "advanced"]] | Omit = omit,
+        mode: Optional[Literal["turbo", "basic", "advanced", "fast"]] | Omit = omit,
         objective: Optional[str] | Omit = omit,
         session_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -388,11 +388,12 @@ def search(
 
           max_chars_total: Upper bound on total characters across excerpts from all results.
 
-          mode: Search mode preset: supported values are `turbo`, `basic`, and `advanced`. Turbo
-              mode is optimized for the fastest responses. Basic mode offers low latency and
-              works best with 2-3 high-quality search_queries. Advanced mode provides higher
-              quality with more advanced retrieval and compression. Defaults to `advanced`
-              when omitted.
+          mode: Search mode preset: supported values are `turbo`, `fast`, `basic`, and
+              `advanced`. Turbo mode is optimized for the fastest responses. Fast mode trades
+              some quality for low latency, and works best with concise, high-quality
+              objective and search_queries. Basic mode offers low latency and works best with
+              2-3 high-quality search_queries. Advanced mode provides higher quality with more
+              advanced retrieval and compression. Defaults to `advanced` when omitted.
 
           objective: Natural-language description of the underlying question or goal driving the
               search. Used together with search_queries to focus results on the most relevant
@@ -748,7 +749,7 @@ async def search(
         advanced_settings: Optional[AdvancedSearchSettingsParam] | Omit = omit,
         client_model: Optional[str] | Omit = omit,
         max_chars_total: Optional[int] | Omit = omit,
-        mode: Optional[Literal["turbo", "basic", "advanced"]] | Omit = omit,
+        mode: Optional[Literal["turbo", "basic", "advanced", "fast"]] | Omit = omit,
         objective: Optional[str] | Omit = omit,
         session_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -781,11 +782,12 @@ async def search(
 
           max_chars_total: Upper bound on total characters across excerpts from all results.
 
-          mode: Search mode preset: supported values are `turbo`, `basic`, and `advanced`. Turbo
-              mode is optimized for the fastest responses. Basic mode offers low latency and
-              works best with 2-3 high-quality search_queries. Advanced mode provides higher
-              quality with more advanced retrieval and compression. Defaults to `advanced`
-              when omitted.
+          mode: Search mode preset: supported values are `turbo`, `fast`, `basic`, and
+              `advanced`. Turbo mode is optimized for the fastest responses. Fast mode trades
+              some quality for low latency, and works best with concise, high-quality
+              objective and search_queries. Basic mode offers low latency and works best with
+              2-3 high-quality search_queries. Advanced mode provides higher quality with more
+              advanced retrieval and compression. Defaults to `advanced` when omitted.
 
           objective: Natural-language description of the underlying question or goal driving the
               search. Used together with search_queries to focus results on the most relevant

src/parallel/resources/beta/__init__.py

@@ -8,6 +8,14 @@
     BetaResourceWithStreamingResponse,
     AsyncBetaResourceWithStreamingResponse,
 )
+from .memory import (
+    MemoryResource,
+    AsyncMemoryResource,
+    MemoryResourceWithRawResponse,
+    AsyncMemoryResourceWithRawResponse,
+    MemoryResourceWithStreamingResponse,
+    AsyncMemoryResourceWithStreamingResponse,
+)
 from .findall import (
     FindAllResource,
     AsyncFindAllResource,
@@ -24,6 +32,12 @@
     "AsyncFindAllResourceWithRawResponse",
     "FindAllResourceWithStreamingResponse",
     "AsyncFindAllResourceWithStreamingResponse",
+    "MemoryResource",
+    "AsyncMemoryResource",
+    "MemoryResourceWithRawResponse",
+    "AsyncMemoryResourceWithRawResponse",
+    "MemoryResourceWithStreamingResponse",
+    "AsyncMemoryResourceWithStreamingResponse",
     "BetaResource",
     "AsyncBetaResource",
     "BetaResourceWithRawResponse",

src/parallel/resources/beta/api.md

@@ -39,3 +39,26 @@ Methods:
 - <code title="post /v1beta/findall/ingest">client.beta.findall.<a href="./src/parallel/resources/beta/findall.py">ingest</a>(\*\*<a href="src/parallel/types/beta/findall_ingest_params.py">params</a>) -> <a href="./src/parallel/types/beta/findall_schema.py">FindAllSchema</a></code>
 - <code title="get /v1beta/findall/runs/{findall_id}/result">client.beta.findall.<a href="./src/parallel/resources/beta/findall.py">result</a>(findall_id) -> <a href="./src/parallel/types/beta/findall_run_result.py">FindAllRunResult</a></code>
 - <code title="get /v1beta/findall/runs/{findall_id}/schema">client.beta.findall.<a href="./src/parallel/resources/beta/findall.py">schema</a>(findall_id) -> <a href="./src/parallel/types/beta/findall_schema.py">FindAllSchema</a></code>
+
+## Memory
+
+Types:
+
+```python
+from parallel.types.beta import (
+    FindAllMemoryResult,
+    MemoryClearRequest,
+    MemoryEvictRequest,
+    MemoryRetrieveRequest,
+    MemoryRetrieveResponse,
+    MonitorMemoryEvent,
+    MonitorMemoryResult,
+    TaskMemoryResult,
+)
+```
+
+Methods:
+
+- <code title="post /v1beta/memory/retrieve">client.beta.memory.<a href="./src/parallel/resources/beta/memory.py">retrieve</a>(\*\*<a href="src/parallel/types/beta/memory_retrieve_params.py">params</a>) -> <a href="./src/parallel/types/beta/memory_retrieve_response.py">MemoryRetrieveResponse</a></code>
+- <code title="post /v1beta/memory/clear">client.beta.memory.<a href="./src/parallel/resources/beta/memory.py">clear</a>(\*\*<a href="src/parallel/types/beta/memory_clear_params.py">params</a>) -> None</code>
+- <code title="post /v1beta/memory/evict">client.beta.memory.<a href="./src/parallel/resources/beta/memory.py">evict</a>(\*\*<a href="src/parallel/types/beta/memory_evict_params.py">params</a>) -> None</code>

src/parallel/resources/beta/beta.py

@@ -2,6 +2,14 @@
 
 from __future__ import annotations
 
+from .memory import (
+    MemoryResource,
+    AsyncMemoryResource,
+    MemoryResourceWithRawResponse,
+    AsyncMemoryResourceWithRawResponse,
+    MemoryResourceWithStreamingResponse,
+    AsyncMemoryResourceWithStreamingResponse,
+)
 from .findall import (
     FindAllResource,
     AsyncFindAllResource,
@@ -24,6 +32,13 @@ def findall(self) -> FindAllResource:
         """
         return FindAllResource(self._client)
 
+    @cached_property
+    def memory(self) -> MemoryResource:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return MemoryResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> BetaResourceWithRawResponse:
         """
@@ -52,6 +67,13 @@ def findall(self) -> AsyncFindAllResource:
         """
         return AsyncFindAllResource(self._client)
 
+    @cached_property
+    def memory(self) -> AsyncMemoryResource:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return AsyncMemoryResource(self._client)
+
     @cached_property
     def with_raw_response(self) -> AsyncBetaResourceWithRawResponse:
         """
@@ -83,6 +105,13 @@ def findall(self) -> FindAllResourceWithRawResponse:
         """
         return FindAllResourceWithRawResponse(self._beta.findall)
 
+    @cached_property
+    def memory(self) -> MemoryResourceWithRawResponse:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return MemoryResourceWithRawResponse(self._beta.memory)
+
 
 class AsyncBetaResourceWithRawResponse:
     def __init__(self, beta: AsyncBetaResource) -> None:
@@ -95,6 +124,13 @@ def findall(self) -> AsyncFindAllResourceWithRawResponse:
         """
         return AsyncFindAllResourceWithRawResponse(self._beta.findall)
 
+    @cached_property
+    def memory(self) -> AsyncMemoryResourceWithRawResponse:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return AsyncMemoryResourceWithRawResponse(self._beta.memory)
+
 
 class BetaResourceWithStreamingResponse:
     def __init__(self, beta: BetaResource) -> None:
@@ -107,6 +143,13 @@ def findall(self) -> FindAllResourceWithStreamingResponse:
         """
         return FindAllResourceWithStreamingResponse(self._beta.findall)
 
+    @cached_property
+    def memory(self) -> MemoryResourceWithStreamingResponse:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return MemoryResourceWithStreamingResponse(self._beta.memory)
+
 
 class AsyncBetaResourceWithStreamingResponse:
     def __init__(self, beta: AsyncBetaResource) -> None:
@@ -118,3 +161,10 @@ def findall(self) -> AsyncFindAllResourceWithStreamingResponse:
         The FindAll API discovers and evaluates entities that match complex criteria from natural language objectives. Submit a high-level goal and the service automatically generates structured match conditions, discovers relevant candidates, and evaluates each against the criteria. Returns comprehensive results with detailed reasoning, citations, and confidence scores for each match decision. Streaming events and webhooks are supported.
         """
         return AsyncFindAllResourceWithStreamingResponse(self._beta.findall)
+
+    @cached_property
+    def memory(self) -> AsyncMemoryResourceWithStreamingResponse:
+        """
+        The Memory API retrieves and manages memories created by Tasks, Monitors, and FindAll runs. Memories can be personal or isolated with a `memory_scope_key`.
+        """
+        return AsyncMemoryResourceWithStreamingResponse(self._beta.memory)

src/parallel/resources/beta/findall.py

@@ -75,6 +75,7 @@ def create(
         match_limit: int,
         objective: str,
         exclude_list: Optional[Iterable[findall_create_params.ExcludeList]] | Omit = omit,
+        memory_scope_key: Optional[str] | Omit = omit,
         metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
         webhook: Optional[WebhookParam] | Omit = omit,
         betas: List[ParallelBetaParam] | Omit = omit,
@@ -111,7 +112,11 @@ def create(
 
           objective: Natural language objective of the FindAll run.
 
-          exclude_list: List of entity names/IDs to exclude from results.
+          exclude_list: List of entity names/IDs to exclude from results. At most 10,000 entries are
+              allowed.
+
+          memory_scope_key: User-provided key identifying the memory scope to use. Omit to use personal
+              memory, if available.
 
           metadata: Metadata for the FindAll run.
 
@@ -148,6 +153,7 @@ def create(
                     "match_limit": match_limit,
                     "objective": objective,
                     "exclude_list": exclude_list,
+                    "memory_scope_key": memory_scope_key,
                     "metadata": metadata,
                     "webhook": webhook,
                 },
@@ -675,6 +681,7 @@ async def create(
         match_limit: int,
         objective: str,
         exclude_list: Optional[Iterable[findall_create_params.ExcludeList]] | Omit = omit,
+        memory_scope_key: Optional[str] | Omit = omit,
         metadata: Optional[Dict[str, Union[str, float, bool]]] | Omit = omit,
         webhook: Optional[WebhookParam] | Omit = omit,
         betas: List[ParallelBetaParam] | Omit = omit,
@@ -711,7 +718,11 @@ async def create(
 
           objective: Natural language objective of the FindAll run.
 
-          exclude_list: List of entity names/IDs to exclude from results.
+          exclude_list: List of entity names/IDs to exclude from results. At most 10,000 entries are
+              allowed.
+
+          memory_scope_key: User-provided key identifying the memory scope to use. Omit to use personal
+              memory, if available.
 
           metadata: Metadata for the FindAll run.
 
@@ -748,6 +759,7 @@ async def create(
                     "match_limit": match_limit,
                     "objective": objective,
                     "exclude_list": exclude_list,
+                    "memory_scope_key": memory_scope_key,
                     "metadata": metadata,
                     "webhook": webhook,
                 },