fix(backend): enable sdk schema cache only for task workers

Signed-off-by: Fatih Acar <[email protected]>

Author: fatih-acar

backend/infrahub/core/schema/manager.py

@@ -5,6 +5,7 @@
 from infrahub_sdk.schema import BranchSchema as SDKBranchSchema
 
 from infrahub import lock
+from infrahub.components import ComponentType
 from infrahub.core.manager import NodeManager
 from infrahub.core.models import (
     HashableModelDiff,
@@ -26,6 +27,7 @@
 from infrahub.core.utils import parse_node_kind
 from infrahub.exceptions import SchemaNotFoundError
 from infrahub.log import get_logger
+from infrahub.workers.dependencies import get_component_type
 
 from .constants import IGNORE_FOR_NODE
 from .schema_branch import SchemaBranch
@@ -42,7 +44,8 @@ class SchemaManager(NodeManager):
     def __init__(self) -> None:
         self._cache: dict[int, Any] = {}
         self._branches: dict[str, SchemaBranch] = {}
-        self._sdk_branches: dict[str, SDKBranchSchema] = {}
+        if get_component_type() == ComponentType.GIT_AGENT:
+            self._sdk_branches: dict[str, SDKBranchSchema] = {}
 
     def _get_from_cache(self, key: int) -> Any:
         return self._cache[key]
@@ -147,12 +150,16 @@ def get_schema_branch(self, name: str) -> SchemaBranch:
         return self._branches[name]
 
     def get_sdk_schema_branch(self, name: str) -> SDKBranchSchema:
-        return self._sdk_branches[name]
+        if get_component_type() == ComponentType.GIT_AGENT:
+            return self._sdk_branches[name]
+
+        raise Exception("get_sdk_schema_branch outside of task worker")
 
     def set_schema_branch(self, name: str, schema: SchemaBranch) -> None:
         schema.name = name
         self._branches[name] = schema
-        self._sdk_branches[name] = SDKBranchSchema.from_api_response(data=schema.to_dict_api_schema_object())
+        if get_component_type() == ComponentType.GIT_AGENT:
+            self._sdk_branches[name] = SDKBranchSchema.from_api_response(data=schema.to_dict_api_schema_object())
 
     def process_schema_branch(self, name: str) -> None:
         schema_branch = self.get_schema_branch(name=name)
@@ -771,6 +778,8 @@ def purge_inactive_branches(self, active_branches: list[str]) -> list[str]:
         for branch_name in list(self._branches.keys()):
             if branch_name not in active_branches:
                 del self._branches[branch_name]
+                if get_component_type() == ComponentType.GIT_AGENT:
+                    del self._sdk_branches[branch_name]
                 removed_branches.append(branch_name)
 
         for hash_key in list(self._cache.keys()):