omnata-plugin-runtime 0.5.7a131__py3-none-any.whl → 0.5.7a132__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.
- omnata_plugin_runtime/omnata_plugin.py +2 -40
- {omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/METADATA +1 -1
- {omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/RECORD +5 -5
- {omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/LICENSE +0 -0
- {omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/WHEEL +0 -0
@@ -2250,16 +2250,6 @@ def consumer_udtf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2250
2250
|
if name != 'cls' and name != 'connection_parameters'
|
2251
2251
|
}
|
2252
2252
|
|
2253
|
-
DynamicModel = create_model('DynamicModel', **fields)
|
2254
|
-
|
2255
|
-
# Attach the documentation to the function
|
2256
|
-
original_process.__doc__ = original_process.__doc__ or ""
|
2257
|
-
original_process.__doc__ += "\n\nArgs:\n"
|
2258
|
-
for name, param in params.items():
|
2259
|
-
if name != 'self':
|
2260
|
-
doc = param_docs.get(name, {})
|
2261
|
-
original_process.__doc__ += f" {name} ({param.annotation.__name__}): {doc.get('description', 'No description provided.')}\n"
|
2262
|
-
|
2263
2253
|
@wraps(original_process)
|
2264
2254
|
def wrapped_process(self, connection_parameters, *args, **kwargs):
|
2265
2255
|
if connection_parameters is None:
|
@@ -2268,12 +2258,6 @@ def consumer_udtf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2268
2258
|
if not isinstance(connection_parameters, Dict):
|
2269
2259
|
raise ValueError("The first argument must be an object, the result of calling PLUGIN_CONNECTION.")
|
2270
2260
|
|
2271
|
-
try:
|
2272
|
-
# Validate the arguments using the dynamic Pydantic model
|
2273
|
-
validated_args = DynamicModel(**kwargs)
|
2274
|
-
except ValidationError as e:
|
2275
|
-
raise ValueError(f"Argument validation error: {e}")
|
2276
|
-
|
2277
2261
|
# convert the connection parameters dictionary to a ConnectionConfigurationParameters object which includes the real secrets
|
2278
2262
|
if 'other_secrets_name' in connection_parameters:
|
2279
2263
|
# this is the new way, where the sync engine only passes the name of the secret
|
@@ -2290,7 +2274,7 @@ def consumer_udtf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2290
2274
|
parameters = ConnectionConfigurationParameters.model_validate(connection_parameters)
|
2291
2275
|
|
2292
2276
|
# Pass the validated arguments to the function
|
2293
|
-
return original_process(self, parameters, *args, **
|
2277
|
+
return original_process(self, parameters, *args, **kwargs)
|
2294
2278
|
# Replace the original 'process' method with the wrapped version
|
2295
2279
|
setattr(cls, 'process', wrapped_process)
|
2296
2280
|
|
@@ -2326,22 +2310,6 @@ def consumer_udf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2326
2310
|
if list(params.keys())[0] != 'connection_parameters':
|
2327
2311
|
raise ValueError("The first argument should be 'connection_parameters'.")
|
2328
2312
|
|
2329
|
-
# Create a Pydantic model based on the function signature and the provided documentation
|
2330
|
-
fields = {
|
2331
|
-
name: (param.annotation, param_docs.get(name, {}).get("default", param.default))
|
2332
|
-
for name, param in params.items()
|
2333
|
-
if name != 'connection_parameters'
|
2334
|
-
}
|
2335
|
-
|
2336
|
-
DynamicModel = create_model('DynamicModel', **fields)
|
2337
|
-
|
2338
|
-
# Attach the documentation to the function
|
2339
|
-
func.__doc__ = func.__doc__ or ""
|
2340
|
-
func.__doc__ += "\n\nArgs:\n"
|
2341
|
-
for name, param in params.items():
|
2342
|
-
doc = param_docs.get(name, {})
|
2343
|
-
func.__doc__ += f" {name} ({param.annotation.__name__}): {doc.get('description', 'No description provided.')}\n"
|
2344
|
-
|
2345
2313
|
@wraps(func)
|
2346
2314
|
def wrapper(self, connection_parameters, *args, **kwargs):
|
2347
2315
|
if connection_parameters is None:
|
@@ -2350,12 +2318,6 @@ def consumer_udf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2350
2318
|
if not isinstance(connection_parameters, Dict):
|
2351
2319
|
raise ValueError("The first argument must be an object, the result of calling PLUGIN_CONNECTION.")
|
2352
2320
|
|
2353
|
-
try:
|
2354
|
-
# Validate the arguments using the dynamic Pydantic model
|
2355
|
-
validated_args = DynamicModel(**kwargs)
|
2356
|
-
except ValidationError as e:
|
2357
|
-
raise ValueError(f"Argument validation error: {e}")
|
2358
|
-
|
2359
2321
|
# convert the connection parameters dictionary to a ConnectionConfigurationParameters object which includes the real secrets
|
2360
2322
|
if 'other_secrets_name' in connection_parameters:
|
2361
2323
|
# this is the new way, where the sync engine only passes the name of the secret
|
@@ -2372,7 +2334,7 @@ def consumer_udf(param_docs: Dict[str, Dict[str, Any]]):
|
|
2372
2334
|
parameters = ConnectionConfigurationParameters.model_validate(connection_parameters)
|
2373
2335
|
|
2374
2336
|
# Pass the validated arguments to the function
|
2375
|
-
return func(parameters, *args, **
|
2337
|
+
return func(parameters, *args, **kwargs)
|
2376
2338
|
|
2377
2339
|
wrapper._is_omnata_consumer_udf = True
|
2378
2340
|
return wrapper
|
{omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/RECORD
RENAMED
@@ -3,10 +3,10 @@ omnata_plugin_runtime/api.py,sha256=FxzTqri4no8ClkOm7vZADG8aD47jcGBCTTQDEORmOJM,
|
|
3
3
|
omnata_plugin_runtime/configuration.py,sha256=TI6GaVFhewVawBCaYN34GujY57qEP6q2nik4YpSEk5s,38100
|
4
4
|
omnata_plugin_runtime/forms.py,sha256=GzSPEwcijsoPCXEO1mHiE8ylvX_KSE5TkhwqkymA2Ss,19755
|
5
5
|
omnata_plugin_runtime/logging.py,sha256=bn7eKoNWvtuyTk7RTwBS9UARMtqkiICtgMtzq3KA2V0,3272
|
6
|
-
omnata_plugin_runtime/omnata_plugin.py,sha256=
|
6
|
+
omnata_plugin_runtime/omnata_plugin.py,sha256=eV1QKOqdxktkVB3ZEQPS7A9a9q7UOsyEhfyZ-JwV5iw,117231
|
7
7
|
omnata_plugin_runtime/plugin_entrypoints.py,sha256=PFSLsYEVnWHVvSoOYTtTK2JY6pp6_8_eYP53WqLRiPE,27975
|
8
8
|
omnata_plugin_runtime/rate_limiting.py,sha256=DVQ_bc-mVLBkrU1PTns1MWXhHiLpSB5HkWCcdePtJ2A,25611
|
9
|
-
omnata_plugin_runtime-0.5.
|
10
|
-
omnata_plugin_runtime-0.5.
|
11
|
-
omnata_plugin_runtime-0.5.
|
12
|
-
omnata_plugin_runtime-0.5.
|
9
|
+
omnata_plugin_runtime-0.5.7a132.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
|
10
|
+
omnata_plugin_runtime-0.5.7a132.dist-info/METADATA,sha256=YHhi631jWmg1ZBJuNsry--DCdhMxWVNynL2ZreqnS7w,1985
|
11
|
+
omnata_plugin_runtime-0.5.7a132.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
12
|
+
omnata_plugin_runtime-0.5.7a132.dist-info/RECORD,,
|
{omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/LICENSE
RENAMED
File without changes
|
{omnata_plugin_runtime-0.5.7a131.dist-info → omnata_plugin_runtime-0.5.7a132.dist-info}/WHEEL
RENAMED
File without changes
|