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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. airweave/__init__.py +41 -43
  2. airweave/client.py +0 -16
  3. airweave/collections/__init__.py +3 -6
  4. airweave/collections/client.py +273 -113
  5. airweave/collections/raw_client.py +633 -94
  6. airweave/collections/types/__init__.py +2 -4
  7. airweave/core/client_wrapper.py +4 -30
  8. airweave/errors/__init__.py +10 -2
  9. airweave/errors/conflict_error.py +11 -0
  10. airweave/errors/not_found_error.py +11 -0
  11. airweave/errors/too_many_requests_error.py +11 -0
  12. airweave/errors/unprocessable_entity_error.py +1 -2
  13. airweave/events/client.py +281 -500
  14. airweave/events/raw_client.py +562 -551
  15. airweave/source_connections/client.py +210 -162
  16. airweave/source_connections/raw_client.py +574 -137
  17. airweave/sources/client.py +42 -18
  18. airweave/sources/raw_client.py +118 -17
  19. airweave/types/__init__.py +33 -39
  20. airweave/types/{endpoint_secret_out.py → conflict_error_response.py} +12 -2
  21. airweave/types/delivery_attempt.py +61 -0
  22. airweave/types/event_message.py +55 -0
  23. airweave/types/event_message_with_attempts.py +59 -0
  24. airweave/types/{enable_endpoint_request.py → not_found_error_response.py} +6 -4
  25. airweave/types/{subscription_with_attempts_out.py → rate_limit_error_response.py} +9 -6
  26. airweave/types/recovery_task.py +35 -0
  27. airweave/types/search_request.py +13 -10
  28. airweave/types/search_response.py +6 -3
  29. airweave/types/source_connection.py +73 -18
  30. airweave/types/source_connection_job.py +65 -15
  31. airweave/types/source_connection_list_item.py +45 -10
  32. airweave/types/sync_event_payload.py +72 -0
  33. airweave/types/{recover_out.py → validation_error_detail.py} +19 -6
  34. airweave/types/validation_error_response.py +30 -0
  35. airweave/types/webhook_subscription.py +68 -0
  36. {airweave_sdk-0.8.65.dist-info → airweave_sdk-0.8.66.dist-info}/METADATA +1 -5
  37. {airweave_sdk-0.8.65.dist-info → airweave_sdk-0.8.66.dist-info}/RECORD +38 -38
  38. airweave/collections/types/search_collections_readable_id_search_post_response.py +0 -8
  39. airweave/types/background_task_status.py +0 -5
  40. airweave/types/background_task_type.py +0 -17
  41. airweave/types/collection_update.py +0 -35
  42. airweave/types/endpoint_out.py +0 -35
  43. airweave/types/message_attempt_out.py +0 -37
  44. airweave/types/message_attempt_trigger_type.py +0 -3
  45. airweave/types/message_out.py +0 -29
  46. airweave/types/message_status.py +0 -3
  47. airweave/types/message_status_text.py +0 -5
  48. {airweave_sdk-0.8.65.dist-info → airweave_sdk-0.8.66.dist-info}/WHEEL +0 -0
airweave/__init__.py CHANGED
@@ -20,8 +20,6 @@ if typing.TYPE_CHECKING:
20
20
  AuthProviderConnectionUpdate,
21
21
  AuthenticationDetails,
22
22
  AuthenticationMethod,
23
- BackgroundTaskStatus,
24
- BackgroundTaskType,
25
23
  BehaviorConfig,
26
24
  BillingPeriod,
27
25
  BillingPeriodStatus,
@@ -35,23 +33,21 @@ if typing.TYPE_CHECKING:
35
33
  CheckoutSessionResponse,
36
34
  Collection,
37
35
  CollectionStatus,
38
- CollectionUpdate,
39
36
  ConfigField,
40
37
  ConfigValues,
38
+ ConflictErrorResponse,
41
39
  Connection,
42
40
  ConnectionStatus,
43
41
  CursorConfig,
44
42
  CustomerPortalRequest,
45
43
  CustomerPortalResponse,
44
+ DeliveryAttempt,
46
45
  Destination,
47
46
  DestinationConfig,
48
47
  DestinationWithAuthenticationFields,
49
48
  DirectAuthentication,
50
49
  EmbeddingModel,
51
50
  EmbeddingModelWithAuthenticationFields,
52
- EnableEndpointRequest,
53
- EndpointOut,
54
- EndpointSecretOut,
55
51
  EntityCount,
56
52
  EntityCountWithDefinition,
57
53
  EntityDefinition,
@@ -61,6 +57,8 @@ if typing.TYPE_CHECKING:
61
57
  EntitySummary,
62
58
  EntityType,
63
59
  EntityTypeStats,
60
+ EventMessage,
61
+ EventMessageWithAttempts,
64
62
  EventType,
65
63
  FeatureFlag,
66
64
  Fields,
@@ -75,13 +73,9 @@ if typing.TYPE_CHECKING:
75
73
  LegacySearchRequestSearchMethod,
76
74
  LegacySearchResponse,
77
75
  MemberResponse,
78
- MessageAttemptOut,
79
- MessageAttemptTriggerType,
80
- MessageOut,
81
76
  MessageResponse,
82
- MessageStatus,
83
- MessageStatusText,
84
77
  MinuteLevelScheduleConfig,
78
+ NotFoundErrorResponse,
85
79
  OAuthBrowserAuthentication,
86
80
  OAuthTokenAuthentication,
87
81
  OAuthType,
@@ -91,7 +85,8 @@ if typing.TYPE_CHECKING:
91
85
  OrganizationMetrics,
92
86
  OrganizationWithRole,
93
87
  QueryExpansionStrategy,
94
- RecoverOut,
88
+ RateLimitErrorResponse,
89
+ RecoveryTask,
95
90
  ResponseType,
96
91
  RetrievalStrategy,
97
92
  S3ConfigRequest,
@@ -113,11 +108,11 @@ if typing.TYPE_CHECKING:
113
108
  SourceRateLimitResponse,
114
109
  SourceRateLimitUpdateRequest,
115
110
  SubscriptionInfo,
116
- SubscriptionWithAttemptsOut,
117
111
  Sync,
118
112
  SyncConfig,
119
113
  SyncCreate,
120
114
  SyncDetails,
115
+ SyncEventPayload,
121
116
  SyncJob,
122
117
  SyncJobDetails,
123
118
  SyncJobStatus,
@@ -134,12 +129,15 @@ if typing.TYPE_CHECKING:
134
129
  UserCreate,
135
130
  UserOrganization,
136
131
  ValidationError,
132
+ ValidationErrorDetail,
137
133
  ValidationErrorLocItem,
134
+ ValidationErrorResponse,
135
+ WebhookSubscription,
138
136
  )
139
- from .errors import UnprocessableEntityError
137
+ from .errors import ConflictError, NotFoundError, TooManyRequestsError, UnprocessableEntityError
140
138
  from . import collections, events, source_connections, sources
141
139
  from .client import AirweaveSDK, AsyncAirweaveSDK
142
- from .collections import SearchCollectionsReadableIdSearchPostRequest, SearchCollectionsReadableIdSearchPostResponse
140
+ from .collections import SearchCollectionsReadableIdSearchPostRequest
143
141
  from .environment import AirweaveSDKEnvironment
144
142
  from .source_connections import Authentication
145
143
  from .version import __version__
@@ -161,8 +159,6 @@ _dynamic_imports: typing.Dict[str, str] = {
161
159
  "Authentication": ".source_connections",
162
160
  "AuthenticationDetails": ".types",
163
161
  "AuthenticationMethod": ".types",
164
- "BackgroundTaskStatus": ".types",
165
- "BackgroundTaskType": ".types",
166
162
  "BehaviorConfig": ".types",
167
163
  "BillingPeriod": ".types",
168
164
  "BillingPeriodStatus": ".types",
@@ -176,23 +172,22 @@ _dynamic_imports: typing.Dict[str, str] = {
176
172
  "CheckoutSessionResponse": ".types",
177
173
  "Collection": ".types",
178
174
  "CollectionStatus": ".types",
179
- "CollectionUpdate": ".types",
180
175
  "ConfigField": ".types",
181
176
  "ConfigValues": ".types",
177
+ "ConflictError": ".errors",
178
+ "ConflictErrorResponse": ".types",
182
179
  "Connection": ".types",
183
180
  "ConnectionStatus": ".types",
184
181
  "CursorConfig": ".types",
185
182
  "CustomerPortalRequest": ".types",
186
183
  "CustomerPortalResponse": ".types",
184
+ "DeliveryAttempt": ".types",
187
185
  "Destination": ".types",
188
186
  "DestinationConfig": ".types",
189
187
  "DestinationWithAuthenticationFields": ".types",
190
188
  "DirectAuthentication": ".types",
191
189
  "EmbeddingModel": ".types",
192
190
  "EmbeddingModelWithAuthenticationFields": ".types",
193
- "EnableEndpointRequest": ".types",
194
- "EndpointOut": ".types",
195
- "EndpointSecretOut": ".types",
196
191
  "EntityCount": ".types",
197
192
  "EntityCountWithDefinition": ".types",
198
193
  "EntityDefinition": ".types",
@@ -202,6 +197,8 @@ _dynamic_imports: typing.Dict[str, str] = {
202
197
  "EntitySummary": ".types",
203
198
  "EntityType": ".types",
204
199
  "EntityTypeStats": ".types",
200
+ "EventMessage": ".types",
201
+ "EventMessageWithAttempts": ".types",
205
202
  "EventType": ".types",
206
203
  "FeatureFlag": ".types",
207
204
  "Fields": ".types",
@@ -216,13 +213,10 @@ _dynamic_imports: typing.Dict[str, str] = {
216
213
  "LegacySearchRequestSearchMethod": ".types",
217
214
  "LegacySearchResponse": ".types",
218
215
  "MemberResponse": ".types",
219
- "MessageAttemptOut": ".types",
220
- "MessageAttemptTriggerType": ".types",
221
- "MessageOut": ".types",
222
216
  "MessageResponse": ".types",
223
- "MessageStatus": ".types",
224
- "MessageStatusText": ".types",
225
217
  "MinuteLevelScheduleConfig": ".types",
218
+ "NotFoundError": ".errors",
219
+ "NotFoundErrorResponse": ".types",
226
220
  "OAuthBrowserAuthentication": ".types",
227
221
  "OAuthTokenAuthentication": ".types",
228
222
  "OAuthType": ".types",
@@ -232,7 +226,8 @@ _dynamic_imports: typing.Dict[str, str] = {
232
226
  "OrganizationMetrics": ".types",
233
227
  "OrganizationWithRole": ".types",
234
228
  "QueryExpansionStrategy": ".types",
235
- "RecoverOut": ".types",
229
+ "RateLimitErrorResponse": ".types",
230
+ "RecoveryTask": ".types",
236
231
  "ResponseType": ".types",
237
232
  "RetrievalStrategy": ".types",
238
233
  "S3ConfigRequest": ".types",
@@ -241,7 +236,6 @@ _dynamic_imports: typing.Dict[str, str] = {
241
236
  "ScheduleDetails": ".types",
242
237
  "ScheduleResponse": ".types",
243
238
  "SearchCollectionsReadableIdSearchPostRequest": ".collections",
244
- "SearchCollectionsReadableIdSearchPostResponse": ".collections",
245
239
  "SearchRequest": ".types",
246
240
  "SearchResponse": ".types",
247
241
  "SearchStatus": ".types",
@@ -256,16 +250,17 @@ _dynamic_imports: typing.Dict[str, str] = {
256
250
  "SourceRateLimitResponse": ".types",
257
251
  "SourceRateLimitUpdateRequest": ".types",
258
252
  "SubscriptionInfo": ".types",
259
- "SubscriptionWithAttemptsOut": ".types",
260
253
  "Sync": ".types",
261
254
  "SyncConfig": ".types",
262
255
  "SyncCreate": ".types",
263
256
  "SyncDetails": ".types",
257
+ "SyncEventPayload": ".types",
264
258
  "SyncJob": ".types",
265
259
  "SyncJobDetails": ".types",
266
260
  "SyncJobStatus": ".types",
267
261
  "SyncStatus": ".types",
268
262
  "SyncUpdate": ".types",
263
+ "TooManyRequestsError": ".errors",
269
264
  "Transformer": ".types",
270
265
  "TransformerCreate": ".types",
271
266
  "TransformerUpdate": ".types",
@@ -278,7 +273,10 @@ _dynamic_imports: typing.Dict[str, str] = {
278
273
  "UserCreate": ".types",
279
274
  "UserOrganization": ".types",
280
275
  "ValidationError": ".types",
276
+ "ValidationErrorDetail": ".types",
281
277
  "ValidationErrorLocItem": ".types",
278
+ "ValidationErrorResponse": ".types",
279
+ "WebhookSubscription": ".types",
282
280
  "__version__": ".version",
283
281
  "collections": ".collections",
284
282
  "events": ".events",
@@ -326,8 +324,6 @@ __all__ = [
326
324
  "Authentication",
327
325
  "AuthenticationDetails",
328
326
  "AuthenticationMethod",
329
- "BackgroundTaskStatus",
330
- "BackgroundTaskType",
331
327
  "BehaviorConfig",
332
328
  "BillingPeriod",
333
329
  "BillingPeriodStatus",
@@ -341,23 +337,22 @@ __all__ = [
341
337
  "CheckoutSessionResponse",
342
338
  "Collection",
343
339
  "CollectionStatus",
344
- "CollectionUpdate",
345
340
  "ConfigField",
346
341
  "ConfigValues",
342
+ "ConflictError",
343
+ "ConflictErrorResponse",
347
344
  "Connection",
348
345
  "ConnectionStatus",
349
346
  "CursorConfig",
350
347
  "CustomerPortalRequest",
351
348
  "CustomerPortalResponse",
349
+ "DeliveryAttempt",
352
350
  "Destination",
353
351
  "DestinationConfig",
354
352
  "DestinationWithAuthenticationFields",
355
353
  "DirectAuthentication",
356
354
  "EmbeddingModel",
357
355
  "EmbeddingModelWithAuthenticationFields",
358
- "EnableEndpointRequest",
359
- "EndpointOut",
360
- "EndpointSecretOut",
361
356
  "EntityCount",
362
357
  "EntityCountWithDefinition",
363
358
  "EntityDefinition",
@@ -367,6 +362,8 @@ __all__ = [
367
362
  "EntitySummary",
368
363
  "EntityType",
369
364
  "EntityTypeStats",
365
+ "EventMessage",
366
+ "EventMessageWithAttempts",
370
367
  "EventType",
371
368
  "FeatureFlag",
372
369
  "Fields",
@@ -381,13 +378,10 @@ __all__ = [
381
378
  "LegacySearchRequestSearchMethod",
382
379
  "LegacySearchResponse",
383
380
  "MemberResponse",
384
- "MessageAttemptOut",
385
- "MessageAttemptTriggerType",
386
- "MessageOut",
387
381
  "MessageResponse",
388
- "MessageStatus",
389
- "MessageStatusText",
390
382
  "MinuteLevelScheduleConfig",
383
+ "NotFoundError",
384
+ "NotFoundErrorResponse",
391
385
  "OAuthBrowserAuthentication",
392
386
  "OAuthTokenAuthentication",
393
387
  "OAuthType",
@@ -397,7 +391,8 @@ __all__ = [
397
391
  "OrganizationMetrics",
398
392
  "OrganizationWithRole",
399
393
  "QueryExpansionStrategy",
400
- "RecoverOut",
394
+ "RateLimitErrorResponse",
395
+ "RecoveryTask",
401
396
  "ResponseType",
402
397
  "RetrievalStrategy",
403
398
  "S3ConfigRequest",
@@ -406,7 +401,6 @@ __all__ = [
406
401
  "ScheduleDetails",
407
402
  "ScheduleResponse",
408
403
  "SearchCollectionsReadableIdSearchPostRequest",
409
- "SearchCollectionsReadableIdSearchPostResponse",
410
404
  "SearchRequest",
411
405
  "SearchResponse",
412
406
  "SearchStatus",
@@ -421,16 +415,17 @@ __all__ = [
421
415
  "SourceRateLimitResponse",
422
416
  "SourceRateLimitUpdateRequest",
423
417
  "SubscriptionInfo",
424
- "SubscriptionWithAttemptsOut",
425
418
  "Sync",
426
419
  "SyncConfig",
427
420
  "SyncCreate",
428
421
  "SyncDetails",
422
+ "SyncEventPayload",
429
423
  "SyncJob",
430
424
  "SyncJobDetails",
431
425
  "SyncJobStatus",
432
426
  "SyncStatus",
433
427
  "SyncUpdate",
428
+ "TooManyRequestsError",
434
429
  "Transformer",
435
430
  "TransformerCreate",
436
431
  "TransformerUpdate",
@@ -443,7 +438,10 @@ __all__ = [
443
438
  "UserCreate",
444
439
  "UserOrganization",
445
440
  "ValidationError",
441
+ "ValidationErrorDetail",
446
442
  "ValidationErrorLocItem",
443
+ "ValidationErrorResponse",
444
+ "WebhookSubscription",
447
445
  "__version__",
448
446
  "collections",
449
447
  "events",
airweave/client.py CHANGED
@@ -33,8 +33,6 @@ class AirweaveSDK:
33
33
 
34
34
 
35
35
 
36
- framework_name : typing.Optional[str]
37
- framework_version : typing.Optional[str]
38
36
  api_key : str
39
37
  headers : typing.Optional[typing.Dict[str, str]]
40
38
  Additional headers to send with every request.
@@ -53,8 +51,6 @@ class AirweaveSDK:
53
51
  from airweave import AirweaveSDK
54
52
 
55
53
  client = AirweaveSDK(
56
- framework_name="YOUR_FRAMEWORK_NAME",
57
- framework_version="YOUR_FRAMEWORK_VERSION",
58
54
  api_key="YOUR_API_KEY",
59
55
  )
60
56
  """
@@ -64,8 +60,6 @@ class AirweaveSDK:
64
60
  *,
65
61
  base_url: typing.Optional[str] = None,
66
62
  environment: AirweaveSDKEnvironment = AirweaveSDKEnvironment.PRODUCTION,
67
- framework_name: typing.Optional[str] = None,
68
- framework_version: typing.Optional[str] = None,
69
63
  api_key: str,
70
64
  headers: typing.Optional[typing.Dict[str, str]] = None,
71
65
  timeout: typing.Optional[float] = None,
@@ -77,8 +71,6 @@ class AirweaveSDK:
77
71
  )
78
72
  self._client_wrapper = SyncClientWrapper(
79
73
  base_url=_get_base_url(base_url=base_url, environment=environment),
80
- framework_name=framework_name,
81
- framework_version=framework_version,
82
74
  api_key=api_key,
83
75
  headers=headers,
84
76
  httpx_client=httpx_client
@@ -144,8 +136,6 @@ class AsyncAirweaveSDK:
144
136
 
145
137
 
146
138
 
147
- framework_name : typing.Optional[str]
148
- framework_version : typing.Optional[str]
149
139
  api_key : str
150
140
  headers : typing.Optional[typing.Dict[str, str]]
151
141
  Additional headers to send with every request.
@@ -164,8 +154,6 @@ class AsyncAirweaveSDK:
164
154
  from airweave import AsyncAirweaveSDK
165
155
 
166
156
  client = AsyncAirweaveSDK(
167
- framework_name="YOUR_FRAMEWORK_NAME",
168
- framework_version="YOUR_FRAMEWORK_VERSION",
169
157
  api_key="YOUR_API_KEY",
170
158
  )
171
159
  """
@@ -175,8 +163,6 @@ class AsyncAirweaveSDK:
175
163
  *,
176
164
  base_url: typing.Optional[str] = None,
177
165
  environment: AirweaveSDKEnvironment = AirweaveSDKEnvironment.PRODUCTION,
178
- framework_name: typing.Optional[str] = None,
179
- framework_version: typing.Optional[str] = None,
180
166
  api_key: str,
181
167
  headers: typing.Optional[typing.Dict[str, str]] = None,
182
168
  timeout: typing.Optional[float] = None,
@@ -188,8 +174,6 @@ class AsyncAirweaveSDK:
188
174
  )
189
175
  self._client_wrapper = AsyncClientWrapper(
190
176
  base_url=_get_base_url(base_url=base_url, environment=environment),
191
- framework_name=framework_name,
192
- framework_version=framework_version,
193
177
  api_key=api_key,
194
178
  headers=headers,
195
179
  httpx_client=httpx_client
airweave/collections/__init__.py CHANGED
@@ -6,11 +6,8 @@ import typing
6
6
  from importlib import import_module
7
7
 
8
8
  if typing.TYPE_CHECKING:
9
- from .types import SearchCollectionsReadableIdSearchPostRequest, SearchCollectionsReadableIdSearchPostResponse
10
- _dynamic_imports: typing.Dict[str, str] = {
11
- "SearchCollectionsReadableIdSearchPostRequest": ".types",
12
- "SearchCollectionsReadableIdSearchPostResponse": ".types",
13
- }
9
+ from .types import SearchCollectionsReadableIdSearchPostRequest
10
+ _dynamic_imports: typing.Dict[str, str] = {"SearchCollectionsReadableIdSearchPostRequest": ".types"}
14
11
 
15
12
 
16
13
  def __getattr__(attr_name: str) -> typing.Any:
@@ -34,4 +31,4 @@ def __dir__():
34
31
  return sorted(lazy_attrs)
35
32
 
36
33
 
37
- __all__ = ["SearchCollectionsReadableIdSearchPostRequest", "SearchCollectionsReadableIdSearchPostResponse"]
34
+ __all__ = ["SearchCollectionsReadableIdSearchPostRequest"]