[MINOR]: Remove JacFeature import inside method; fix FastAPI.enable t…

…rigger
Jaseci-Labs · Oct 2, 2024 · 1837dac · 1837dac
1 parent 16c10db
commit 1837dac
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/jac-cloud/jac_cloud/core/architype.py b/jac-cloud/jac_cloud/core/architype.py
@@ -19,6 +19,7 @@
 
 from bson import ObjectId
 
+from jaclang.plugin.feature import JacFeature as Jac
 from jaclang.runtimelib.architype import (
  Access as _Access,
  AccessLevel,
@@ -422,8 +423,6 @@ def build_query(
  bulk_write: BulkWrite,
  ) -> None:
  """Save Anchor."""
- from jaclang.plugin.feature import JacFeature as Jac
-
  if self.state.deleted is False and Jac.check_write_access(self): # type: ignore[arg-type]
  self.state.deleted = True
  self.delete(bulk_write)
@@ -470,7 +469,6 @@ def update(self, bulk_write: BulkWrite, propagate: bool = False) -> None:
  ############################################################
  # POPULATE CONTEXT #
  ############################################################
- from jaclang.plugin.feature import JacFeature as Jac
 
  if Jac.check_write_access(self): # type: ignore[arg-type]
  set_architype = changes.pop("$set", {})

diff --git a/jac-cloud/jac_cloud/core/memory.py b/jac-cloud/jac_cloud/core/memory.py
@@ -6,6 +6,7 @@
 
 from bson import ObjectId
 
+from jaclang.plugin.feature import JacFeature as Jac
 from jaclang.runtimelib.memory import Memory
 
 
@@ -115,8 +116,6 @@ def close(self) -> None:
 
  def get_bulk_write(self) -> BulkWrite:
  """Sync memory to database."""
- from jaclang.plugin.feature import JacFeature as Jac
-
  bulk_write = BulkWrite()
 
  for anchor in self.__gc__:

diff --git a/jac-cloud/jac_cloud/plugin/jaseci.py b/jac-cloud/jac_cloud/plugin/jaseci.py
@@ -391,7 +391,7 @@ def get_edges(
  target_obj: list[NodeArchitype] | None,
  ) -> list[EdgeArchitype]:
  """Get edges connected to this node."""
- if FastAPI.enable():
+ if FastAPI.is_enabled():
  JaseciContext.get().mem.populate_data(node.edges)
 
  return JacFeatureImpl.get_edges(
@@ -407,7 +407,7 @@ def edges_to_nodes(
  target_obj: list[NodeArchitype] | None,
  ) -> list[NodeArchitype]:
  """Get set of nodes connected to this node."""
- if FastAPI.enable():
+ if FastAPI.is_enabled():
  JaseciContext.get().mem.populate_data(node.edges)
 
  return JacFeatureImpl.edges_to_nodes(