airweave-sdk 0.8.64__py3-none-any.whl → 0.8.66__py3-none-any.whl

airweave/__init__.py

@@ -33,23 +33,21 @@ if typing.TYPE_CHECKING:
         CheckoutSessionResponse,
         Collection,
         CollectionStatus,
-        CollectionUpdate,
         ConfigField,
         ConfigValues,
+        ConflictErrorResponse,
         Connection,
         ConnectionStatus,
-        CreateSubscriptionRequest,
         CursorConfig,
         CustomerPortalRequest,
         CustomerPortalResponse,
+        DeliveryAttempt,
         Destination,
         DestinationConfig,
         DestinationWithAuthenticationFields,
         DirectAuthentication,
         EmbeddingModel,
         EmbeddingModelWithAuthenticationFields,
-        EndpointOut,
-        EndpointSecretOut,
         EntityCount,
         EntityCountWithDefinition,
         EntityDefinition,
@@ -59,6 +57,8 @@ if typing.TYPE_CHECKING:
         EntitySummary,
         EntityType,
         EntityTypeStats,
+        EventMessage,
+        EventMessageWithAttempts,
         EventType,
         FeatureFlag,
         Fields,
@@ -73,13 +73,9 @@ if typing.TYPE_CHECKING:
         LegacySearchRequestSearchMethod,
         LegacySearchResponse,
         MemberResponse,
-        MessageAttemptOut,
-        MessageAttemptTriggerType,
-        MessageOut,
         MessageResponse,
-        MessageStatus,
-        MessageStatusText,
         MinuteLevelScheduleConfig,
+        NotFoundErrorResponse,
         OAuthBrowserAuthentication,
         OAuthTokenAuthentication,
         OAuthType,
@@ -88,8 +84,9 @@ if typing.TYPE_CHECKING:
         OrganizationCreate,
         OrganizationMetrics,
         OrganizationWithRole,
-        PatchSubscriptionRequest,
         QueryExpansionStrategy,
+        RateLimitErrorResponse,
+        RecoveryTask,
         ResponseType,
         RetrievalStrategy,
         S3ConfigRequest,
@@ -111,11 +108,11 @@ if typing.TYPE_CHECKING:
         SourceRateLimitResponse,
         SourceRateLimitUpdateRequest,
         SubscriptionInfo,
-        SubscriptionWithAttemptsOut,
         Sync,
         SyncConfig,
         SyncCreate,
         SyncDetails,
+        SyncEventPayload,
         SyncJob,
         SyncJobDetails,
         SyncJobStatus,
@@ -132,12 +129,15 @@ if typing.TYPE_CHECKING:
         UserCreate,
         UserOrganization,
         ValidationError,
+        ValidationErrorDetail,
         ValidationErrorLocItem,
+        ValidationErrorResponse,
+        WebhookSubscription,
     )
-    from .errors import UnprocessableEntityError
-    from . import collections, source_connections, sources
+    from .errors import ConflictError, NotFoundError, TooManyRequestsError, UnprocessableEntityError
+    from . import collections, events, source_connections, sources
     from .client import AirweaveSDK, AsyncAirweaveSDK
-    from .collections import SearchCollectionsReadableIdSearchPostRequest, SearchCollectionsReadableIdSearchPostResponse
+    from .collections import SearchCollectionsReadableIdSearchPostRequest
     from .environment import AirweaveSDKEnvironment
     from .source_connections import Authentication
     from .version import __version__
@@ -172,23 +172,22 @@ _dynamic_imports: typing.Dict[str, str] = {
     "CheckoutSessionResponse": ".types",
     "Collection": ".types",
     "CollectionStatus": ".types",
-    "CollectionUpdate": ".types",
     "ConfigField": ".types",
     "ConfigValues": ".types",
+    "ConflictError": ".errors",
+    "ConflictErrorResponse": ".types",
     "Connection": ".types",
     "ConnectionStatus": ".types",
-    "CreateSubscriptionRequest": ".types",
     "CursorConfig": ".types",
     "CustomerPortalRequest": ".types",
     "CustomerPortalResponse": ".types",
+    "DeliveryAttempt": ".types",
     "Destination": ".types",
     "DestinationConfig": ".types",
     "DestinationWithAuthenticationFields": ".types",
     "DirectAuthentication": ".types",
     "EmbeddingModel": ".types",
     "EmbeddingModelWithAuthenticationFields": ".types",
-    "EndpointOut": ".types",
-    "EndpointSecretOut": ".types",
     "EntityCount": ".types",
     "EntityCountWithDefinition": ".types",
     "EntityDefinition": ".types",
@@ -198,6 +197,8 @@ _dynamic_imports: typing.Dict[str, str] = {
     "EntitySummary": ".types",
     "EntityType": ".types",
     "EntityTypeStats": ".types",
+    "EventMessage": ".types",
+    "EventMessageWithAttempts": ".types",
     "EventType": ".types",
     "FeatureFlag": ".types",
     "Fields": ".types",
@@ -212,13 +213,10 @@ _dynamic_imports: typing.Dict[str, str] = {
     "LegacySearchRequestSearchMethod": ".types",
     "LegacySearchResponse": ".types",
     "MemberResponse": ".types",
-    "MessageAttemptOut": ".types",
-    "MessageAttemptTriggerType": ".types",
-    "MessageOut": ".types",
     "MessageResponse": ".types",
-    "MessageStatus": ".types",
-    "MessageStatusText": ".types",
     "MinuteLevelScheduleConfig": ".types",
+    "NotFoundError": ".errors",
+    "NotFoundErrorResponse": ".types",
     "OAuthBrowserAuthentication": ".types",
     "OAuthTokenAuthentication": ".types",
     "OAuthType": ".types",
@@ -227,8 +225,9 @@ _dynamic_imports: typing.Dict[str, str] = {
     "OrganizationCreate": ".types",
     "OrganizationMetrics": ".types",
     "OrganizationWithRole": ".types",
-    "PatchSubscriptionRequest": ".types",
     "QueryExpansionStrategy": ".types",
+    "RateLimitErrorResponse": ".types",
+    "RecoveryTask": ".types",
     "ResponseType": ".types",
     "RetrievalStrategy": ".types",
     "S3ConfigRequest": ".types",
@@ -237,7 +236,6 @@ _dynamic_imports: typing.Dict[str, str] = {
     "ScheduleDetails": ".types",
     "ScheduleResponse": ".types",
     "SearchCollectionsReadableIdSearchPostRequest": ".collections",
-    "SearchCollectionsReadableIdSearchPostResponse": ".collections",
     "SearchRequest": ".types",
     "SearchResponse": ".types",
     "SearchStatus": ".types",
@@ -252,16 +250,17 @@ _dynamic_imports: typing.Dict[str, str] = {
     "SourceRateLimitResponse": ".types",
     "SourceRateLimitUpdateRequest": ".types",
     "SubscriptionInfo": ".types",
-    "SubscriptionWithAttemptsOut": ".types",
     "Sync": ".types",
     "SyncConfig": ".types",
     "SyncCreate": ".types",
     "SyncDetails": ".types",
+    "SyncEventPayload": ".types",
     "SyncJob": ".types",
     "SyncJobDetails": ".types",
     "SyncJobStatus": ".types",
     "SyncStatus": ".types",
     "SyncUpdate": ".types",
+    "TooManyRequestsError": ".errors",
     "Transformer": ".types",
     "TransformerCreate": ".types",
     "TransformerUpdate": ".types",
@@ -274,9 +273,13 @@ _dynamic_imports: typing.Dict[str, str] = {
     "UserCreate": ".types",
     "UserOrganization": ".types",
     "ValidationError": ".types",
+    "ValidationErrorDetail": ".types",
     "ValidationErrorLocItem": ".types",
+    "ValidationErrorResponse": ".types",
+    "WebhookSubscription": ".types",
     "__version__": ".version",
     "collections": ".collections",
+    "events": ".events",
     "source_connections": ".source_connections",
     "sources": ".sources",
 }
@@ -334,23 +337,22 @@ __all__ = [
     "CheckoutSessionResponse",
     "Collection",
     "CollectionStatus",
-    "CollectionUpdate",
     "ConfigField",
     "ConfigValues",
+    "ConflictError",
+    "ConflictErrorResponse",
     "Connection",
     "ConnectionStatus",
-    "CreateSubscriptionRequest",
     "CursorConfig",
     "CustomerPortalRequest",
     "CustomerPortalResponse",
+    "DeliveryAttempt",
     "Destination",
     "DestinationConfig",
     "DestinationWithAuthenticationFields",
     "DirectAuthentication",
     "EmbeddingModel",
     "EmbeddingModelWithAuthenticationFields",
-    "EndpointOut",
-    "EndpointSecretOut",
     "EntityCount",
     "EntityCountWithDefinition",
     "EntityDefinition",
@@ -360,6 +362,8 @@ __all__ = [
     "EntitySummary",
     "EntityType",
     "EntityTypeStats",
+    "EventMessage",
+    "EventMessageWithAttempts",
     "EventType",
     "FeatureFlag",
     "Fields",
@@ -374,13 +378,10 @@ __all__ = [
     "LegacySearchRequestSearchMethod",
     "LegacySearchResponse",
     "MemberResponse",
-    "MessageAttemptOut",
-    "MessageAttemptTriggerType",
-    "MessageOut",
     "MessageResponse",
-    "MessageStatus",
-    "MessageStatusText",
     "MinuteLevelScheduleConfig",
+    "NotFoundError",
+    "NotFoundErrorResponse",
     "OAuthBrowserAuthentication",
     "OAuthTokenAuthentication",
     "OAuthType",
@@ -389,8 +390,9 @@ __all__ = [
     "OrganizationCreate",
     "OrganizationMetrics",
     "OrganizationWithRole",
-    "PatchSubscriptionRequest",
     "QueryExpansionStrategy",
+    "RateLimitErrorResponse",
+    "RecoveryTask",
     "ResponseType",
     "RetrievalStrategy",
     "S3ConfigRequest",
@@ -399,7 +401,6 @@ __all__ = [
     "ScheduleDetails",
     "ScheduleResponse",
     "SearchCollectionsReadableIdSearchPostRequest",
-    "SearchCollectionsReadableIdSearchPostResponse",
     "SearchRequest",
     "SearchResponse",
     "SearchStatus",
@@ -414,16 +415,17 @@ __all__ = [
     "SourceRateLimitResponse",
     "SourceRateLimitUpdateRequest",
     "SubscriptionInfo",
-    "SubscriptionWithAttemptsOut",
     "Sync",
     "SyncConfig",
     "SyncCreate",
     "SyncDetails",
+    "SyncEventPayload",
     "SyncJob",
     "SyncJobDetails",
     "SyncJobStatus",
     "SyncStatus",
     "SyncUpdate",
+    "TooManyRequestsError",
     "Transformer",
     "TransformerCreate",
     "TransformerUpdate",
@@ -436,9 +438,13 @@ __all__ = [
     "UserCreate",
     "UserOrganization",
     "ValidationError",
+    "ValidationErrorDetail",
     "ValidationErrorLocItem",
+    "ValidationErrorResponse",
+    "WebhookSubscription",
     "__version__",
     "collections",
+    "events",
     "source_connections",
     "sources",
 ]

airweave/client.py

@@ -10,6 +10,7 @@ from .environment import AirweaveSDKEnvironment
 
 if typing.TYPE_CHECKING:
     from .collections.client import AsyncCollectionsClient, CollectionsClient
+    from .events.client import AsyncEventsClient, EventsClient
     from .source_connections.client import AsyncSourceConnectionsClient, SourceConnectionsClient
     from .sources.client import AsyncSourcesClient, SourcesClient
 
@@ -32,8 +33,6 @@ class AirweaveSDK:
 
 
 
-    framework_name : typing.Optional[str]
-    framework_version : typing.Optional[str]
     api_key : str
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
@@ -52,8 +51,6 @@ class AirweaveSDK:
     from airweave import AirweaveSDK
 
     client = AirweaveSDK(
-        framework_name="YOUR_FRAMEWORK_NAME",
-        framework_version="YOUR_FRAMEWORK_VERSION",
         api_key="YOUR_API_KEY",
     )
     """
@@ -63,8 +60,6 @@ class AirweaveSDK:
         *,
         base_url: typing.Optional[str] = None,
         environment: AirweaveSDKEnvironment = AirweaveSDKEnvironment.PRODUCTION,
-        framework_name: typing.Optional[str] = None,
-        framework_version: typing.Optional[str] = None,
         api_key: str,
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
@@ -76,8 +71,6 @@ class AirweaveSDK:
         )
         self._client_wrapper = SyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            framework_name=framework_name,
-            framework_version=framework_version,
             api_key=api_key,
             headers=headers,
             httpx_client=httpx_client
@@ -90,6 +83,7 @@ class AirweaveSDK:
         self._sources: typing.Optional[SourcesClient] = None
         self._collections: typing.Optional[CollectionsClient] = None
         self._source_connections: typing.Optional[SourceConnectionsClient] = None
+        self._events: typing.Optional[EventsClient] = None
 
     @property
     def sources(self):
@@ -115,6 +109,14 @@ class AirweaveSDK:
             self._source_connections = SourceConnectionsClient(client_wrapper=self._client_wrapper)
         return self._source_connections
 
+    @property
+    def events(self):
+        if self._events is None:
+            from .events.client import EventsClient  # noqa: E402
+
+            self._events = EventsClient(client_wrapper=self._client_wrapper)
+        return self._events
+
 
 class AsyncAirweaveSDK:
     """
@@ -134,8 +136,6 @@ class AsyncAirweaveSDK:
 
 
 
-    framework_name : typing.Optional[str]
-    framework_version : typing.Optional[str]
     api_key : str
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
@@ -154,8 +154,6 @@ class AsyncAirweaveSDK:
     from airweave import AsyncAirweaveSDK
 
     client = AsyncAirweaveSDK(
-        framework_name="YOUR_FRAMEWORK_NAME",
-        framework_version="YOUR_FRAMEWORK_VERSION",
         api_key="YOUR_API_KEY",
     )
     """
@@ -165,8 +163,6 @@ class AsyncAirweaveSDK:
         *,
         base_url: typing.Optional[str] = None,
         environment: AirweaveSDKEnvironment = AirweaveSDKEnvironment.PRODUCTION,
-        framework_name: typing.Optional[str] = None,
-        framework_version: typing.Optional[str] = None,
         api_key: str,
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
@@ -178,8 +174,6 @@ class AsyncAirweaveSDK:
         )
         self._client_wrapper = AsyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
-            framework_name=framework_name,
-            framework_version=framework_version,
             api_key=api_key,
             headers=headers,
             httpx_client=httpx_client
@@ -192,6 +186,7 @@ class AsyncAirweaveSDK:
         self._sources: typing.Optional[AsyncSourcesClient] = None
         self._collections: typing.Optional[AsyncCollectionsClient] = None
         self._source_connections: typing.Optional[AsyncSourceConnectionsClient] = None
+        self._events: typing.Optional[AsyncEventsClient] = None
 
     @property
     def sources(self):
@@ -217,6 +212,14 @@ class AsyncAirweaveSDK:
             self._source_connections = AsyncSourceConnectionsClient(client_wrapper=self._client_wrapper)
         return self._source_connections
 
+    @property
+    def events(self):
+        if self._events is None:
+            from .events.client import AsyncEventsClient  # noqa: E402
+
+            self._events = AsyncEventsClient(client_wrapper=self._client_wrapper)
+        return self._events
+
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AirweaveSDKEnvironment) -> str:
     if base_url is not None:

airweave/collections/__init__.py

@@ -6,11 +6,8 @@ import typing
 from importlib import import_module
 
 if typing.TYPE_CHECKING:
-    from .types import SearchCollectionsReadableIdSearchPostRequest, SearchCollectionsReadableIdSearchPostResponse
-_dynamic_imports: typing.Dict[str, str] = {
-    "SearchCollectionsReadableIdSearchPostRequest": ".types",
-    "SearchCollectionsReadableIdSearchPostResponse": ".types",
-}
+    from .types import SearchCollectionsReadableIdSearchPostRequest
+_dynamic_imports: typing.Dict[str, str] = {"SearchCollectionsReadableIdSearchPostRequest": ".types"}
 
 
 def __getattr__(attr_name: str) -> typing.Any:
@@ -34,4 +31,4 @@ def __dir__():
     return sorted(lazy_attrs)
 
 
-__all__ = ["SearchCollectionsReadableIdSearchPostRequest", "SearchCollectionsReadableIdSearchPostResponse"]
+__all__ = ["SearchCollectionsReadableIdSearchPostRequest"]