mixpeek 0.21.8__py3-none-any.whl → 0.22.0__py3-none-any.whl

mixpeek/models/usermodel_input.py

@@ -4,24 +4,16 @@ from __future__ import annotations
 from .apikey import APIKey, APIKeyTypedDict
 from datetime import datetime
 from mixpeek.types import BaseModel
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class UserModelInputMetadataTypedDict(TypedDict):
-    pass
-
-
-class UserModelInputMetadata(BaseModel):
-    pass
-
-
 class UserModelInputTypedDict(TypedDict):
     email: str
     user_id: NotRequired[str]
     user_name: NotRequired[str]
     api_keys: NotRequired[List[APIKeyTypedDict]]
-    metadata: NotRequired[UserModelInputMetadataTypedDict]
+    metadata: NotRequired[Dict[str, Any]]
     created_at: NotRequired[datetime]
 
 
@@ -34,6 +26,6 @@ class UserModelInput(BaseModel):
 
     api_keys: Optional[List[APIKey]] = None
 
-    metadata: Optional[UserModelInputMetadata] = None
+    metadata: Optional[Dict[str, Any]] = None
 
     created_at: Optional[datetime] = None

mixpeek/models/usermodel_output.py

@@ -4,24 +4,16 @@ from __future__ import annotations
 from .apikey import APIKey, APIKeyTypedDict
 from datetime import datetime
 from mixpeek.types import BaseModel
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 from typing_extensions import NotRequired, TypedDict
 
 
-class UserModelOutputMetadataTypedDict(TypedDict):
-    pass
-
-
-class UserModelOutputMetadata(BaseModel):
-    pass
-
-
 class UserModelOutputTypedDict(TypedDict):
     email: str
     user_id: NotRequired[str]
     user_name: NotRequired[str]
     api_keys: NotRequired[List[APIKeyTypedDict]]
-    metadata: NotRequired[UserModelOutputMetadataTypedDict]
+    metadata: NotRequired[Dict[str, Any]]
     created_at: NotRequired[datetime]
 
 
@@ -34,6 +26,6 @@ class UserModelOutput(BaseModel):
 
     api_keys: Optional[List[APIKey]] = None
 
-    metadata: Optional[UserModelOutputMetadata] = None
+    metadata: Optional[Dict[str, Any]] = None
 
     created_at: Optional[datetime] = None

mixpeek/namespaces.py

@@ -13,13 +13,11 @@ class Namespaces(BaseSDK):
         self,
         *,
         namespace_name: str,
+        feature_extractors: Union[
+            List[models.BasicFeatureExtractor],
+            List[models.BasicFeatureExtractorTypedDict],
+        ],
         description: OptionalNullable[str] = UNSET,
-        feature_extractors: Optional[
-            Union[
-                List[models.BasicFeatureExtractor],
-                List[models.BasicFeatureExtractorTypedDict],
-            ]
-        ] = None,
         payload_indexes: OptionalNullable[
             Union[
                 List[models.PayloadIndexConfig],
@@ -33,12 +31,12 @@ class Namespaces(BaseSDK):
     ) -> models.NamespaceResponse:
         r"""Create Namespace
 
-        Creates a new namespace with the specified configuration
+        Creates a new namespace with specified feature extractors and payload indexes.
 
         :param namespace_name: Name of the namespace to create
+        :param feature_extractors: List of feature extractors to use. At least one feature extractor must be provided.
         :param description: Description of the namespace
-        :param feature_extractors: List of feature extractors to use
-        :param payload_indexes: List of payload index configurations
+        :param payload_indexes: Optional list of custom payload index configurations. Indexes required by selected feature extractors will be added automatically.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -58,7 +56,7 @@ class Namespaces(BaseSDK):
             namespace_name=namespace_name,
             description=description,
             feature_extractors=utils.get_pydantic_model(
-                feature_extractors, Optional[List[models.BasicFeatureExtractor]]
+                feature_extractors, List[models.BasicFeatureExtractor]
             ),
             payload_indexes=utils.get_pydantic_model(
                 payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
@@ -94,6 +92,7 @@ class Namespaces(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_namespace_v1_namespaces_post",
                 oauth2_scopes=[],
@@ -107,7 +106,7 @@ class Namespaces(BaseSDK):
         )
 
         response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
+        if utils.match_response(http_res, "201", "application/json"):
             return utils.unmarshal_json(http_res.text, models.NamespaceResponse)
         if utils.match_response(
             http_res, ["400", "401", "403", "404"], "application/json"
@@ -150,13 +149,11 @@ class Namespaces(BaseSDK):
         self,
         *,
         namespace_name: str,
+        feature_extractors: Union[
+            List[models.BasicFeatureExtractor],
+            List[models.BasicFeatureExtractorTypedDict],
+        ],
         description: OptionalNullable[str] = UNSET,
-        feature_extractors: Optional[
-            Union[
-                List[models.BasicFeatureExtractor],
-                List[models.BasicFeatureExtractorTypedDict],
-            ]
-        ] = None,
         payload_indexes: OptionalNullable[
             Union[
                 List[models.PayloadIndexConfig],
@@ -170,12 +167,12 @@ class Namespaces(BaseSDK):
     ) -> models.NamespaceResponse:
         r"""Create Namespace
 
-        Creates a new namespace with the specified configuration
+        Creates a new namespace with specified feature extractors and payload indexes.
 
         :param namespace_name: Name of the namespace to create
+        :param feature_extractors: List of feature extractors to use. At least one feature extractor must be provided.
         :param description: Description of the namespace
-        :param feature_extractors: List of feature extractors to use
-        :param payload_indexes: List of payload index configurations
+        :param payload_indexes: Optional list of custom payload index configurations. Indexes required by selected feature extractors will be added automatically.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -195,7 +192,7 @@ class Namespaces(BaseSDK):
             namespace_name=namespace_name,
             description=description,
             feature_extractors=utils.get_pydantic_model(
-                feature_extractors, Optional[List[models.BasicFeatureExtractor]]
+                feature_extractors, List[models.BasicFeatureExtractor]
             ),
             payload_indexes=utils.get_pydantic_model(
                 payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
@@ -231,6 +228,7 @@ class Namespaces(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_namespace_v1_namespaces_post",
                 oauth2_scopes=[],
@@ -244,7 +242,7 @@ class Namespaces(BaseSDK):
         )
 
         response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
+        if utils.match_response(http_res, "201", "application/json"):
             return utils.unmarshal_json(http_res.text, models.NamespaceResponse)
         if utils.match_response(
             http_res, ["400", "401", "403", "404"], "application/json"
@@ -335,6 +333,7 @@ class Namespaces(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_namespaces_v1_namespaces_get",
                 oauth2_scopes=[],
@@ -439,6 +438,7 @@ class Namespaces(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_namespaces_v1_namespaces_get",
                 oauth2_scopes=[],
@@ -550,6 +550,7 @@ class Namespaces(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_namespace_v1_namespaces__namespace__delete",
                 oauth2_scopes=[],
@@ -661,6 +662,7 @@ class Namespaces(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_namespace_v1_namespaces__namespace__delete",
                 oauth2_scopes=[],
@@ -794,6 +796,7 @@ class Namespaces(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update_namespace_v1_namespaces__namespace__put",
                 oauth2_scopes=[],
@@ -927,6 +930,7 @@ class Namespaces(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update_namespace_v1_namespaces__namespace__put",
                 oauth2_scopes=[],
@@ -1038,6 +1042,7 @@ class Namespaces(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_namespace_v1_namespaces__namespace__get",
                 oauth2_scopes=[],
@@ -1149,6 +1154,7 @@ class Namespaces(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_namespace_v1_namespaces__namespace__get",
                 oauth2_scopes=[],

mixpeek/organizationnotifications.py

@@ -59,6 +59,7 @@ class OrganizationNotifications(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="send_notification_v1_organizations_notifications_send_post",
                 oauth2_scopes=[],
@@ -161,6 +162,7 @@ class OrganizationNotifications(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="send_notification_v1_organizations_notifications_send_post",
                 oauth2_scopes=[],

mixpeek/organizations.py

@@ -6,7 +6,7 @@ from mixpeek import models, utils
 from mixpeek._hooks import HookContext
 from mixpeek.types import OptionalNullable, UNSET
 from mixpeek.utils import get_security_from_env
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class Organizations(BaseSDK):
@@ -63,6 +63,7 @@ class Organizations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_organization_v1_organizations_get",
                 oauth2_scopes=[],
@@ -168,6 +169,7 @@ class Organizations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_organization_v1_organizations_get",
                 oauth2_scopes=[],
@@ -229,9 +231,7 @@ class Organizations(BaseSDK):
         api_keys: Optional[
             Union[List[models.APIKey], List[models.APIKeyTypedDict]]
         ] = None,
-        metadata: Optional[
-            Union[models.UserModelInputMetadata, models.UserModelInputMetadataTypedDict]
-        ] = None,
+        metadata: Optional[Dict[str, Any]] = None,
         created_at: Optional[datetime] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -269,9 +269,7 @@ class Organizations(BaseSDK):
             user_name=user_name,
             email=email,
             api_keys=utils.get_pydantic_model(api_keys, Optional[List[models.APIKey]]),
-            metadata=utils.get_pydantic_model(
-                metadata, Optional[models.UserModelInputMetadata]
-            ),
+            metadata=metadata,
             created_at=created_at,
         )
 
@@ -304,6 +302,7 @@ class Organizations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="add_user_v1_organizations_users_post",
                 oauth2_scopes=[],
@@ -365,9 +364,7 @@ class Organizations(BaseSDK):
         api_keys: Optional[
             Union[List[models.APIKey], List[models.APIKeyTypedDict]]
         ] = None,
-        metadata: Optional[
-            Union[models.UserModelInputMetadata, models.UserModelInputMetadataTypedDict]
-        ] = None,
+        metadata: Optional[Dict[str, Any]] = None,
         created_at: Optional[datetime] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -405,9 +402,7 @@ class Organizations(BaseSDK):
             user_name=user_name,
             email=email,
             api_keys=utils.get_pydantic_model(api_keys, Optional[List[models.APIKey]]),
-            metadata=utils.get_pydantic_model(
-                metadata, Optional[models.UserModelInputMetadata]
-            ),
+            metadata=metadata,
             created_at=created_at,
         )
 
@@ -440,6 +435,7 @@ class Organizations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="add_user_v1_organizations_users_post",
                 oauth2_scopes=[],
@@ -556,6 +552,7 @@ class Organizations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_api_key_v1_organizations_users__user_email__api_keys__key_name__delete",
                 oauth2_scopes=[],
@@ -672,6 +669,7 @@ class Organizations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_api_key_v1_organizations_users__user_email__api_keys__key_name__delete",
                 oauth2_scopes=[],
@@ -799,6 +797,7 @@ class Organizations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update_api_key_v1_organizations_users__user_email__api_keys__key_name__patch",
                 oauth2_scopes=[],
@@ -926,6 +925,7 @@ class Organizations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="update_api_key_v1_organizations_users__user_email__api_keys__key_name__patch",
                 oauth2_scopes=[],

mixpeek/organizationsusage.py

@@ -71,6 +71,7 @@ class OrganizationsUsage(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_usage_v1_organizations_usage_post",
                 oauth2_scopes=[],
@@ -185,6 +186,7 @@ class OrganizationsUsage(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_usage_v1_organizations_usage_post",
                 oauth2_scopes=[],

mixpeek/research.py

@@ -66,6 +66,7 @@ class Research(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_research_v1_research_get",
                 oauth2_scopes=[],
@@ -175,6 +176,7 @@ class Research(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_research_v1_research_get",
                 oauth2_scopes=[],

mixpeek/retrieverinteractions.py

@@ -5,7 +5,7 @@ from mixpeek import models, utils
 from mixpeek._hooks import HookContext
 from mixpeek.types import OptionalNullable, UNSET
 from mixpeek.utils import get_security_from_env
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, Dict, List, Mapping, Optional
 
 
 class RetrieverInteractions(BaseSDK):
@@ -16,12 +16,7 @@ class RetrieverInteractions(BaseSDK):
         interaction_type: List[models.InteractionType],
         x_namespace: OptionalNullable[str] = UNSET,
         position: OptionalNullable[int] = UNSET,
-        metadata: OptionalNullable[
-            Union[
-                models.SearchInteractionMetadata,
-                models.SearchInteractionMetadataTypedDict,
-            ]
-        ] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         user_id: OptionalNullable[str] = UNSET,
         session_id: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -64,9 +59,7 @@ class RetrieverInteractions(BaseSDK):
                 feature_id=feature_id,
                 interaction_type=interaction_type,
                 position=position,
-                metadata=utils.get_pydantic_model(
-                    metadata, OptionalNullable[models.SearchInteractionMetadata]
-                ),
+                metadata=metadata,
                 user_id=user_id,
                 session_id=session_id,
             ),
@@ -105,6 +98,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_interaction_v1_retrievers_interactions_post",
                 oauth2_scopes=[],
@@ -164,12 +158,7 @@ class RetrieverInteractions(BaseSDK):
         interaction_type: List[models.InteractionType],
         x_namespace: OptionalNullable[str] = UNSET,
         position: OptionalNullable[int] = UNSET,
-        metadata: OptionalNullable[
-            Union[
-                models.SearchInteractionMetadata,
-                models.SearchInteractionMetadataTypedDict,
-            ]
-        ] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         user_id: OptionalNullable[str] = UNSET,
         session_id: OptionalNullable[str] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
@@ -212,9 +201,7 @@ class RetrieverInteractions(BaseSDK):
                 feature_id=feature_id,
                 interaction_type=interaction_type,
                 position=position,
-                metadata=utils.get_pydantic_model(
-                    metadata, OptionalNullable[models.SearchInteractionMetadata]
-                ),
+                metadata=metadata,
                 user_id=user_id,
                 session_id=session_id,
             ),
@@ -253,6 +240,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_interaction_v1_retrievers_interactions_post",
                 oauth2_scopes=[],
@@ -382,6 +370,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_interactions_v1_retrievers_interactions_get",
                 oauth2_scopes=[],
@@ -511,6 +500,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="list_interactions_v1_retrievers_interactions_get",
                 oauth2_scopes=[],
@@ -628,6 +618,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_interaction_v1_retrievers_interactions__interaction_id__get",
                 oauth2_scopes=[],
@@ -745,6 +736,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_interaction_v1_retrievers_interactions__interaction_id__get",
                 oauth2_scopes=[],
@@ -864,6 +856,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_interaction_v1_retrievers_interactions__interaction_id__delete",
                 oauth2_scopes=[],
@@ -983,6 +976,7 @@ class RetrieverInteractions(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="delete_interaction_v1_retrievers_interactions__interaction_id__delete",
                 oauth2_scopes=[],

mixpeek/retrievers.py

@@ -5,7 +5,7 @@ from mixpeek import models, utils
 from mixpeek._hooks import HookContext
 from mixpeek.types import OptionalNullable, UNSET
 from mixpeek.utils import get_security_from_env
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class Retrievers(BaseSDK):
@@ -22,12 +22,7 @@ class Retrievers(BaseSDK):
         ],
         x_namespace: OptionalNullable[str] = UNSET,
         description: OptionalNullable[str] = UNSET,
-        metadata: Optional[
-            Union[
-                models.CreateRetrieverRequestMetadata,
-                models.CreateRetrieverRequestMetadataTypedDict,
-            ]
-        ] = None,
+        metadata: Optional[Dict[str, Any]] = None,
         cache_config: OptionalNullable[
             Union[models.RetrieverCacheConfig, models.RetrieverCacheConfigTypedDict]
         ] = UNSET,
@@ -71,9 +66,7 @@ class Retrievers(BaseSDK):
                 ),
                 collection_ids=collection_ids,
                 stages=utils.get_pydantic_model(stages, List[models.StageConfigInput]),
-                metadata=utils.get_pydantic_model(
-                    metadata, Optional[models.CreateRetrieverRequestMetadata]
-                ),
+                metadata=metadata,
                 cache_config=utils.get_pydantic_model(
                     cache_config, OptionalNullable[models.RetrieverCacheConfig]
                 ),
@@ -113,6 +106,7 @@ class Retrievers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_retriever_v1_retrievers_retrievers_post",
                 oauth2_scopes=[],
@@ -178,12 +172,7 @@ class Retrievers(BaseSDK):
         ],
         x_namespace: OptionalNullable[str] = UNSET,
         description: OptionalNullable[str] = UNSET,
-        metadata: Optional[
-            Union[
-                models.CreateRetrieverRequestMetadata,
-                models.CreateRetrieverRequestMetadataTypedDict,
-            ]
-        ] = None,
+        metadata: Optional[Dict[str, Any]] = None,
         cache_config: OptionalNullable[
             Union[models.RetrieverCacheConfig, models.RetrieverCacheConfigTypedDict]
         ] = UNSET,
@@ -227,9 +216,7 @@ class Retrievers(BaseSDK):
                 ),
                 collection_ids=collection_ids,
                 stages=utils.get_pydantic_model(stages, List[models.StageConfigInput]),
-                metadata=utils.get_pydantic_model(
-                    metadata, Optional[models.CreateRetrieverRequestMetadata]
-                ),
+                metadata=metadata,
                 cache_config=utils.get_pydantic_model(
                     cache_config, OptionalNullable[models.RetrieverCacheConfig]
                 ),
@@ -269,6 +256,7 @@ class Retrievers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="create_retriever_v1_retrievers_retrievers_post",
                 oauth2_scopes=[],
@@ -381,6 +369,7 @@ class Retrievers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_retriever_v1_retrievers_retrievers__retriever_id__get",
                 oauth2_scopes=[],
@@ -493,6 +482,7 @@ class Retrievers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="get_retriever_v1_retrievers_retrievers__retriever_id__get",
                 oauth2_scopes=[],
@@ -549,10 +539,7 @@ class Retrievers(BaseSDK):
         self,
         *,
         retriever_id: str,
-        inputs: Union[
-            models.RetrieverQueryRequestInputs,
-            models.RetrieverQueryRequestInputsTypedDict,
-        ],
+        inputs: Dict[str, Any],
         x_namespace: OptionalNullable[str] = UNSET,
         filters: OptionalNullable[
             Union[models.LogicalOperatorInput, models.LogicalOperatorInputTypedDict]
@@ -602,9 +589,7 @@ class Retrievers(BaseSDK):
                 retriever_id=retriever_id,
                 x_namespace=x_namespace,
                 retriever_query_request=models.RetrieverQueryRequest(
-                    inputs=utils.get_pydantic_model(
-                        inputs, models.RetrieverQueryRequestInputs
-                    ),
+                    inputs=inputs,
                     filters=utils.get_pydantic_model(
                         filters, OptionalNullable[models.LogicalOperatorInput]
                     ),
@@ -653,6 +638,7 @@ class Retrievers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="execute_retriever_v1_retrievers_retrievers__retriever_id__execute_post",
                 oauth2_scopes=[],
@@ -709,10 +695,7 @@ class Retrievers(BaseSDK):
         self,
         *,
         retriever_id: str,
-        inputs: Union[
-            models.RetrieverQueryRequestInputs,
-            models.RetrieverQueryRequestInputsTypedDict,
-        ],
+        inputs: Dict[str, Any],
         x_namespace: OptionalNullable[str] = UNSET,
         filters: OptionalNullable[
             Union[models.LogicalOperatorInput, models.LogicalOperatorInputTypedDict]
@@ -762,9 +745,7 @@ class Retrievers(BaseSDK):
                 retriever_id=retriever_id,
                 x_namespace=x_namespace,
                 retriever_query_request=models.RetrieverQueryRequest(
-                    inputs=utils.get_pydantic_model(
-                        inputs, models.RetrieverQueryRequestInputs
-                    ),
+                    inputs=inputs,
                     filters=utils.get_pydantic_model(
                         filters, OptionalNullable[models.LogicalOperatorInput]
                     ),
@@ -813,6 +794,7 @@ class Retrievers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="execute_retriever_v1_retrievers_retrievers__retriever_id__execute_post",
                 oauth2_scopes=[],