airweave-sdk 0.8.63__py3-none-any.whl → 0.8.65__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.
- airweave/__init__.py +15 -7
- airweave/client.py +19 -0
- airweave/core/client_wrapper.py +2 -2
- airweave/events/__init__.py +4 -0
- airweave/events/client.py +1138 -0
- airweave/events/raw_client.py +1424 -0
- airweave/types/__init__.py +12 -6
- airweave/types/background_task_status.py +5 -0
- airweave/types/background_task_type.py +17 -0
- airweave/types/{create_subscription_request.py → enable_endpoint_request.py} +4 -6
- airweave/types/{patch_subscription_request.py → recover_out.py} +6 -8
- {airweave_sdk-0.8.63.dist-info → airweave_sdk-0.8.65.dist-info}/METADATA +1 -1
- {airweave_sdk-0.8.63.dist-info → airweave_sdk-0.8.65.dist-info}/RECORD +14 -9
- {airweave_sdk-0.8.63.dist-info → airweave_sdk-0.8.65.dist-info}/WHEEL +0 -0
airweave/__init__.py
CHANGED
|
@@ -20,6 +20,8 @@ if typing.TYPE_CHECKING:
|
|
|
20
20
|
AuthProviderConnectionUpdate,
|
|
21
21
|
AuthenticationDetails,
|
|
22
22
|
AuthenticationMethod,
|
|
23
|
+
BackgroundTaskStatus,
|
|
24
|
+
BackgroundTaskType,
|
|
23
25
|
BehaviorConfig,
|
|
24
26
|
BillingPeriod,
|
|
25
27
|
BillingPeriodStatus,
|
|
@@ -38,7 +40,6 @@ if typing.TYPE_CHECKING:
|
|
|
38
40
|
ConfigValues,
|
|
39
41
|
Connection,
|
|
40
42
|
ConnectionStatus,
|
|
41
|
-
CreateSubscriptionRequest,
|
|
42
43
|
CursorConfig,
|
|
43
44
|
CustomerPortalRequest,
|
|
44
45
|
CustomerPortalResponse,
|
|
@@ -48,6 +49,7 @@ if typing.TYPE_CHECKING:
|
|
|
48
49
|
DirectAuthentication,
|
|
49
50
|
EmbeddingModel,
|
|
50
51
|
EmbeddingModelWithAuthenticationFields,
|
|
52
|
+
EnableEndpointRequest,
|
|
51
53
|
EndpointOut,
|
|
52
54
|
EndpointSecretOut,
|
|
53
55
|
EntityCount,
|
|
@@ -88,8 +90,8 @@ if typing.TYPE_CHECKING:
|
|
|
88
90
|
OrganizationCreate,
|
|
89
91
|
OrganizationMetrics,
|
|
90
92
|
OrganizationWithRole,
|
|
91
|
-
PatchSubscriptionRequest,
|
|
92
93
|
QueryExpansionStrategy,
|
|
94
|
+
RecoverOut,
|
|
93
95
|
ResponseType,
|
|
94
96
|
RetrievalStrategy,
|
|
95
97
|
S3ConfigRequest,
|
|
@@ -135,7 +137,7 @@ if typing.TYPE_CHECKING:
|
|
|
135
137
|
ValidationErrorLocItem,
|
|
136
138
|
)
|
|
137
139
|
from .errors import UnprocessableEntityError
|
|
138
|
-
from . import collections, source_connections, sources
|
|
140
|
+
from . import collections, events, source_connections, sources
|
|
139
141
|
from .client import AirweaveSDK, AsyncAirweaveSDK
|
|
140
142
|
from .collections import SearchCollectionsReadableIdSearchPostRequest, SearchCollectionsReadableIdSearchPostResponse
|
|
141
143
|
from .environment import AirweaveSDKEnvironment
|
|
@@ -159,6 +161,8 @@ _dynamic_imports: typing.Dict[str, str] = {
|
|
|
159
161
|
"Authentication": ".source_connections",
|
|
160
162
|
"AuthenticationDetails": ".types",
|
|
161
163
|
"AuthenticationMethod": ".types",
|
|
164
|
+
"BackgroundTaskStatus": ".types",
|
|
165
|
+
"BackgroundTaskType": ".types",
|
|
162
166
|
"BehaviorConfig": ".types",
|
|
163
167
|
"BillingPeriod": ".types",
|
|
164
168
|
"BillingPeriodStatus": ".types",
|
|
@@ -177,7 +181,6 @@ _dynamic_imports: typing.Dict[str, str] = {
|
|
|
177
181
|
"ConfigValues": ".types",
|
|
178
182
|
"Connection": ".types",
|
|
179
183
|
"ConnectionStatus": ".types",
|
|
180
|
-
"CreateSubscriptionRequest": ".types",
|
|
181
184
|
"CursorConfig": ".types",
|
|
182
185
|
"CustomerPortalRequest": ".types",
|
|
183
186
|
"CustomerPortalResponse": ".types",
|
|
@@ -187,6 +190,7 @@ _dynamic_imports: typing.Dict[str, str] = {
|
|
|
187
190
|
"DirectAuthentication": ".types",
|
|
188
191
|
"EmbeddingModel": ".types",
|
|
189
192
|
"EmbeddingModelWithAuthenticationFields": ".types",
|
|
193
|
+
"EnableEndpointRequest": ".types",
|
|
190
194
|
"EndpointOut": ".types",
|
|
191
195
|
"EndpointSecretOut": ".types",
|
|
192
196
|
"EntityCount": ".types",
|
|
@@ -227,8 +231,8 @@ _dynamic_imports: typing.Dict[str, str] = {
|
|
|
227
231
|
"OrganizationCreate": ".types",
|
|
228
232
|
"OrganizationMetrics": ".types",
|
|
229
233
|
"OrganizationWithRole": ".types",
|
|
230
|
-
"PatchSubscriptionRequest": ".types",
|
|
231
234
|
"QueryExpansionStrategy": ".types",
|
|
235
|
+
"RecoverOut": ".types",
|
|
232
236
|
"ResponseType": ".types",
|
|
233
237
|
"RetrievalStrategy": ".types",
|
|
234
238
|
"S3ConfigRequest": ".types",
|
|
@@ -277,6 +281,7 @@ _dynamic_imports: typing.Dict[str, str] = {
|
|
|
277
281
|
"ValidationErrorLocItem": ".types",
|
|
278
282
|
"__version__": ".version",
|
|
279
283
|
"collections": ".collections",
|
|
284
|
+
"events": ".events",
|
|
280
285
|
"source_connections": ".source_connections",
|
|
281
286
|
"sources": ".sources",
|
|
282
287
|
}
|
|
@@ -321,6 +326,8 @@ __all__ = [
|
|
|
321
326
|
"Authentication",
|
|
322
327
|
"AuthenticationDetails",
|
|
323
328
|
"AuthenticationMethod",
|
|
329
|
+
"BackgroundTaskStatus",
|
|
330
|
+
"BackgroundTaskType",
|
|
324
331
|
"BehaviorConfig",
|
|
325
332
|
"BillingPeriod",
|
|
326
333
|
"BillingPeriodStatus",
|
|
@@ -339,7 +346,6 @@ __all__ = [
|
|
|
339
346
|
"ConfigValues",
|
|
340
347
|
"Connection",
|
|
341
348
|
"ConnectionStatus",
|
|
342
|
-
"CreateSubscriptionRequest",
|
|
343
349
|
"CursorConfig",
|
|
344
350
|
"CustomerPortalRequest",
|
|
345
351
|
"CustomerPortalResponse",
|
|
@@ -349,6 +355,7 @@ __all__ = [
|
|
|
349
355
|
"DirectAuthentication",
|
|
350
356
|
"EmbeddingModel",
|
|
351
357
|
"EmbeddingModelWithAuthenticationFields",
|
|
358
|
+
"EnableEndpointRequest",
|
|
352
359
|
"EndpointOut",
|
|
353
360
|
"EndpointSecretOut",
|
|
354
361
|
"EntityCount",
|
|
@@ -389,8 +396,8 @@ __all__ = [
|
|
|
389
396
|
"OrganizationCreate",
|
|
390
397
|
"OrganizationMetrics",
|
|
391
398
|
"OrganizationWithRole",
|
|
392
|
-
"PatchSubscriptionRequest",
|
|
393
399
|
"QueryExpansionStrategy",
|
|
400
|
+
"RecoverOut",
|
|
394
401
|
"ResponseType",
|
|
395
402
|
"RetrievalStrategy",
|
|
396
403
|
"S3ConfigRequest",
|
|
@@ -439,6 +446,7 @@ __all__ = [
|
|
|
439
446
|
"ValidationErrorLocItem",
|
|
440
447
|
"__version__",
|
|
441
448
|
"collections",
|
|
449
|
+
"events",
|
|
442
450
|
"source_connections",
|
|
443
451
|
"sources",
|
|
444
452
|
]
|
airweave/client.py
CHANGED
|
@@ -10,6 +10,7 @@ from .environment import AirweaveSDKEnvironment
|
|
|
10
10
|
|
|
11
11
|
if typing.TYPE_CHECKING:
|
|
12
12
|
from .collections.client import AsyncCollectionsClient, CollectionsClient
|
|
13
|
+
from .events.client import AsyncEventsClient, EventsClient
|
|
13
14
|
from .source_connections.client import AsyncSourceConnectionsClient, SourceConnectionsClient
|
|
14
15
|
from .sources.client import AsyncSourcesClient, SourcesClient
|
|
15
16
|
|
|
@@ -90,6 +91,7 @@ class AirweaveSDK:
|
|
|
90
91
|
self._sources: typing.Optional[SourcesClient] = None
|
|
91
92
|
self._collections: typing.Optional[CollectionsClient] = None
|
|
92
93
|
self._source_connections: typing.Optional[SourceConnectionsClient] = None
|
|
94
|
+
self._events: typing.Optional[EventsClient] = None
|
|
93
95
|
|
|
94
96
|
@property
|
|
95
97
|
def sources(self):
|
|
@@ -115,6 +117,14 @@ class AirweaveSDK:
|
|
|
115
117
|
self._source_connections = SourceConnectionsClient(client_wrapper=self._client_wrapper)
|
|
116
118
|
return self._source_connections
|
|
117
119
|
|
|
120
|
+
@property
|
|
121
|
+
def events(self):
|
|
122
|
+
if self._events is None:
|
|
123
|
+
from .events.client import EventsClient # noqa: E402
|
|
124
|
+
|
|
125
|
+
self._events = EventsClient(client_wrapper=self._client_wrapper)
|
|
126
|
+
return self._events
|
|
127
|
+
|
|
118
128
|
|
|
119
129
|
class AsyncAirweaveSDK:
|
|
120
130
|
"""
|
|
@@ -192,6 +202,7 @@ class AsyncAirweaveSDK:
|
|
|
192
202
|
self._sources: typing.Optional[AsyncSourcesClient] = None
|
|
193
203
|
self._collections: typing.Optional[AsyncCollectionsClient] = None
|
|
194
204
|
self._source_connections: typing.Optional[AsyncSourceConnectionsClient] = None
|
|
205
|
+
self._events: typing.Optional[AsyncEventsClient] = None
|
|
195
206
|
|
|
196
207
|
@property
|
|
197
208
|
def sources(self):
|
|
@@ -217,6 +228,14 @@ class AsyncAirweaveSDK:
|
|
|
217
228
|
self._source_connections = AsyncSourceConnectionsClient(client_wrapper=self._client_wrapper)
|
|
218
229
|
return self._source_connections
|
|
219
230
|
|
|
231
|
+
@property
|
|
232
|
+
def events(self):
|
|
233
|
+
if self._events is None:
|
|
234
|
+
from .events.client import AsyncEventsClient # noqa: E402
|
|
235
|
+
|
|
236
|
+
self._events = AsyncEventsClient(client_wrapper=self._client_wrapper)
|
|
237
|
+
return self._events
|
|
238
|
+
|
|
220
239
|
|
|
221
240
|
def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AirweaveSDKEnvironment) -> str:
|
|
222
241
|
if base_url is not None:
|
airweave/core/client_wrapper.py
CHANGED
|
@@ -26,10 +26,10 @@ class BaseClientWrapper:
|
|
|
26
26
|
|
|
27
27
|
def get_headers(self) -> typing.Dict[str, str]:
|
|
28
28
|
headers: typing.Dict[str, str] = {
|
|
29
|
-
"User-Agent": "airweave-sdk/v0.8.
|
|
29
|
+
"User-Agent": "airweave-sdk/v0.8.65",
|
|
30
30
|
"X-Fern-Language": "Python",
|
|
31
31
|
"X-Fern-SDK-Name": "airweave-sdk",
|
|
32
|
-
"X-Fern-SDK-Version": "v0.8.
|
|
32
|
+
"X-Fern-SDK-Version": "v0.8.65",
|
|
33
33
|
**(self.get_custom_headers() or {}),
|
|
34
34
|
}
|
|
35
35
|
if self._framework_name is not None:
|