feat(api): enable ZDR on remaining AI endpoints via OpenAI (#1099)

Stainless-Generated-From: cdbd5ca

Author: stlc-bot

src/context/dev/resources/ai.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing_extensions import overload
+from typing_extensions import Literal, overload
 
 import httpx
 
@@ -51,6 +51,7 @@ def extract_product(
         max_age_ms: int | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: ai_extract_product_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -75,6 +76,12 @@ def extract_product(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -91,6 +98,7 @@ def extract_product(
                     "max_age_ms": max_age_ms,
                     "tags": tags,
                     "timeout_opts": timeout_opts,
+                    "zdr": zdr,
                 },
                 ai_extract_product_params.AIExtractProductParams,
             ),
@@ -261,6 +269,7 @@ async def extract_product(
         max_age_ms: int | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: ai_extract_product_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -285,6 +294,12 @@ async def extract_product(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -301,6 +316,7 @@ async def extract_product(
                     "max_age_ms": max_age_ms,
                     "tags": tags,
                     "timeout_opts": timeout_opts,
+                    "zdr": zdr,
                 },
                 ai_extract_product_params.AIExtractProductParams,
             ),

src/context/dev/resources/industry.py

@@ -52,6 +52,7 @@ def retrieve_naics(
         min_results: int | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: industry_retrieve_naics_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -79,6 +80,12 @@ def retrieve_naics(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -101,6 +108,7 @@ def retrieve_naics(
                         "min_results": min_results,
                         "tags": tags,
                         "timeout_opts": timeout_opts,
+                        "zdr": zdr,
                     },
                     industry_retrieve_naics_params.IndustryRetrieveNaicsParams,
                 ),
@@ -117,6 +125,7 @@ def retrieve_sic(
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: industry_retrieve_sic_params.TimeoutOpts | Omit = omit,
         type: Literal["original_sic", "latest_sec"] | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -149,6 +158,12 @@ def retrieve_sic(
               Industrial Classification system; `latest_sec` uses the current SIC list as
               published by the SEC. Defaults to `original_sic`.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -172,6 +187,7 @@ def retrieve_sic(
                         "tags": tags,
                         "timeout_opts": timeout_opts,
                         "type": type,
+                        "zdr": zdr,
                     },
                     industry_retrieve_sic_params.IndustryRetrieveSicParams,
                 ),
@@ -208,6 +224,7 @@ async def retrieve_naics(
         min_results: int | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: industry_retrieve_naics_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -235,6 +252,12 @@ async def retrieve_naics(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -257,6 +280,7 @@ async def retrieve_naics(
                         "min_results": min_results,
                         "tags": tags,
                         "timeout_opts": timeout_opts,
+                        "zdr": zdr,
                     },
                     industry_retrieve_naics_params.IndustryRetrieveNaicsParams,
                 ),
@@ -273,6 +297,7 @@ async def retrieve_sic(
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: industry_retrieve_sic_params.TimeoutOpts | Omit = omit,
         type: Literal["original_sic", "latest_sec"] | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -305,6 +330,12 @@ async def retrieve_sic(
               Industrial Classification system; `latest_sec` uses the current SIC list as
               published by the SEC. Defaults to `original_sic`.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -328,6 +359,7 @@ async def retrieve_sic(
                         "tags": tags,
                         "timeout_opts": timeout_opts,
                         "type": type,
+                        "zdr": zdr,
                     },
                     industry_retrieve_sic_params.IndustryRetrieveSicParams,
                 ),

src/context/dev/resources/parse.py

@@ -180,9 +180,10 @@ def handle(
           use_main_content_only: Extract only the main content from HTML-like inputs
 
           zdr: Set to enabled to bypass shared caches and omit request and response content
-              from retained usage logs. Requires zero data retention to be enabled for your
-              organization (contact support@context.dev), otherwise the request fails with
-              ZDR_NOT_ENABLED. Successful ZDR responses include X-Context-ZDR: true.
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
 
           extra_headers: Send extra headers
 
@@ -364,9 +365,10 @@ async def handle(
           use_main_content_only: Extract only the main content from HTML-like inputs
 
           zdr: Set to enabled to bypass shared caches and omit request and response content
-              from retained usage logs. Requires zero data retention to be enabled for your
-              organization (contact support@context.dev), otherwise the request fails with
-              ZDR_NOT_ENABLED. Successful ZDR responses include X-Context-ZDR: true.
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
 
           extra_headers: Send extra headers
 

src/context/dev/resources/people.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from typing import Iterable
+from typing_extensions import Literal
 
 import httpx
 
@@ -54,6 +55,7 @@ def enrich(
         social_urls: SequenceNotStr[str] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: person_enrich_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -74,6 +76,12 @@ def enrich(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -94,6 +102,7 @@ def enrich(
                     "social_urls": social_urls,
                     "tags": tags,
                     "timeout_opts": timeout_opts,
+                    "zdr": zdr,
                 },
                 person_enrich_params.PersonEnrichParams,
             ),
@@ -135,6 +144,7 @@ async def enrich(
         social_urls: SequenceNotStr[str] | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
         timeout_opts: person_enrich_params.TimeoutOpts | Omit = omit,
+        zdr: Literal["enabled", "disabled"] | 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,
@@ -155,6 +165,12 @@ async def enrich(
               timeoutOpts[milliseconds]=30000&timeoutOpts[behavior]=fail or a JSON-encoded
               timeoutOpts object.
 
+          zdr: Set to enabled to bypass shared caches and omit request and response content
+              from retained usage logs. Asset uploads are skipped, so hosted image URLs are
+              omitted. Requires zero data retention to be enabled for your organization
+              (contact support@context.dev), otherwise the request fails with ZDR_NOT_ENABLED.
+              Successful ZDR responses include X-Context-ZDR: true.
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -175,6 +191,7 @@ async def enrich(
                     "social_urls": social_urls,
                     "tags": tags,
                     "timeout_opts": timeout_opts,
+                    "zdr": zdr,
                 },
                 person_enrich_params.PersonEnrichParams,
             ),