[Haik]: ckpt, swapped the HaikFix folder to be parallel to the agents folder bc ima start the full replacement process i think

2026-09-18 15:48:04 +02:00 · 2026-04-02 15:54:42 -07:00
parent 71845cc0ae
commit df07f82253
36 changed files with 111 additions and 111 deletions
@@ -0,0 +1,42 @@
+from typing import Optional, List
+from pydantic import BaseModel
+from backend.apps.HaikFix.tools.shared_structs.Tool import Tool
+from backend.apps.HaikFix.tools.shared_structs.TOOL_PERMISSIONS import TOOL_PERMISSIONS
+from typeguard import typechecked
+
+class Toolkit(BaseModel):
+    name: str
+    description: str
+    tools: Optional[List[Tool]] = None
+    nested_toolkits: Optional[List["Toolkit"]] = None
+
+    @typechecked
+    def __init__(
+        self, 
+        name: str, 
+        description: str, 
+        tools: Optional[List[Tool]] = None, 
+        nested_toolkits: Optional[List["Toolkit"]] = None
+    ) -> None:
+        super().__init__(
+            name=name, 
+            description=description, 
+            tools=tools, 
+            nested_toolkits=nested_toolkits
+        )
+        self.validate_structure(tools, nested_toolkits)
+
+    @typechecked
+    def validate_structure(self) -> None:
+        assert not ( (self.tools is None) and (self.nested_toolkits is None) ), "Either tools or nested_toolkits must be provided"
+        assert (self.tools is None) or (self.nested_toolkits is None), "Only one of tools or nested_toolkits can be provided"
+
+    @typechecked
+    def set_permission(self, permission: TOOL_PERMISSIONS) -> None:
+        self.validate_structure()
+        if self.tools is not None:
+            for tool in self.tools:
+                tool.permission = permission
+        if self.nested_toolkits is not None:
+            for toolkit in self.nested_toolkits:
+                toolkit.set_permission(permission)