Include Sema4ai actions when refreshing the agent spec

Sema4AI · Oct 23, 2024 · b0b7d39 · b0b7d39
1 parent 04b7b49
commit b0b7d39
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sema4ai/src/sema4ai_code/refresh_agent_spec_helper.py b/sema4ai/src/sema4ai_code/refresh_agent_spec_helper.py
@@ -6,6 +6,7 @@
     from sema4ai_code.agents.list_actions_from_agent import ActionPackageInFilesystem
 
 SEMA4AI = "sema4ai"
+MYACTIONS = "myactions"
 
 
 class ActionPackage(TypedDict, total=False):
@@ -48,7 +49,7 @@ def _update_agent_spec_with_actions(
             "whitelist": "",
         }
         for action_package in action_packages_in_filesystem
-        if action_package.organization.replace(".", "").lower() != SEMA4AI
+        if action_package.organization.replace(".", "").lower() in [SEMA4AI, MYACTIONS]
     ]
 
     missing = []