Update registry loaders and repository

TracecatHQ · Jan 15, 2025 · cf968ca · cf968ca
1 parent 8d36f53
commit cf968ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tracecat/registry/loaders.py b/tracecat/registry/loaders.py
@@ -63,6 +63,7 @@ def get_bound_action_impl(
             display_group=action.display_group,
             doc_url=action.doc_url,
             author=action.author,
+            deprecated=action.deprecated,
             origin=action.origin,
         )
     else:
@@ -89,6 +90,7 @@ def get_bound_action_impl(
             display_group=action.display_group,
             doc_url=action.doc_url,
             author=action.author,
+            deprecated=action.deprecated,
             include_in_schema=True,
             template_action=impl.template_action,
             origin=action.origin,

diff --git a/tracecat/registry/repository.py b/tracecat/registry/repository.py
@@ -63,6 +63,7 @@ class RegisterKwargs(BaseModel):
     display_group: str | None
     doc_url: str | None
     author: str | None
+    deprecated: str | None
     secrets: list[RegistrySecret] | None
     include_in_schema: bool
 
@@ -157,6 +158,7 @@ def register_udf(
         display_group: str | None,
         doc_url: str | None,
         author: str | None,
+        deprecated: str | None,
         include_in_schema: bool,
         template_action: TemplateAction | None = None,
         origin: str = DEFAULT_REGISTRY_ORIGIN,
@@ -169,6 +171,7 @@ def register_udf(
             type=type,
             doc_url=doc_url,
             author=author,
+            deprecated=deprecated,
             secrets=secrets,
             args_cls=args_cls,
             args_docs=args_docs,
@@ -382,6 +385,7 @@ def _register_udf_from_function(
             description=validated_kwargs.description,
             doc_url=validated_kwargs.doc_url,
             author=validated_kwargs.author,
+            deprecated=validated_kwargs.deprecated,
             secrets=validated_kwargs.secrets,
             default_title=validated_kwargs.default_title,
             display_group=validated_kwargs.display_group,