[MINOR]: Address PR Comments

jaclang/cli/cli.py

@@ -444,8 +444,6 @@ def start_cli() -> None:
         args_dict = vars(args)
         args_dict.pop("command")
         args_dict.pop("version", None)
-        if command.func.__name__ != "run":
-            args_dict.pop("session")
         ret = command.call(**args_dict)
         if ret:
             print(ret)

jaclang/core/architype.py

@@ -21,7 +21,8 @@
 
 from jaclang.compiler.constant import EdgeDir
 from jaclang.core.utils import collect_node_connections
-from jaclang.vendor.orjson import dumps
+
+from orjson import dumps
 
 GENERIC_ID_REGEX = compile(
     r"^(g|n|e|w):([^:]*):([0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12})$",
@@ -196,7 +197,7 @@ def sync(self, node: Optional["NodeAnchor"] = None) -> Optional[Architype]:
 
         from .context import ExecutionContext
 
-        jsrc = ExecutionContext.get().datasource
+        jsrc = ExecutionContext.get_or_create().datasource
         anchor = jsrc.find_one(self.id)
 
         if anchor and (node or self).has_read_access(anchor):
@@ -208,7 +209,7 @@ def allocate(self) -> None:
         """Allocate hashes and memory."""
         from .context import ExecutionContext
 
-        jctx = ExecutionContext.get()
+        jctx = ExecutionContext.get_or_create()
         self.root = jctx.root.id
         jctx.datasource.set(self, True)
 
@@ -228,7 +229,7 @@ def access_level(self, to: Anchor) -> int:
         """Access validation."""
         from .context import ExecutionContext
 
-        jctx = ExecutionContext.get()
+        jctx = ExecutionContext.get_or_create()
         jroot = jctx.root
         to.current_access_level = -1
 
@@ -347,7 +348,7 @@ def ref(cls, ref_id: str) -> Optional[NodeAnchor]:
     def _save(self) -> None:
         from .context import ExecutionContext
 
-        jsrc = ExecutionContext.get().datasource
+        jsrc = ExecutionContext.get_or_create().datasource
 
         for edge in self.edges:
             edge.save()
@@ -359,7 +360,7 @@ def destroy(self) -> None:
         if self.architype and self.current_access_level > 1:
             from .context import ExecutionContext
 
-            jsrc = ExecutionContext.get().datasource
+            jsrc = ExecutionContext.get_or_create().datasource
             for edge in self.edges:
                 edge.destroy()
 
@@ -507,7 +508,7 @@ def ref(cls, ref_id: str) -> Optional[EdgeAnchor]:
     def _save(self) -> None:
         from .context import ExecutionContext
 
-        jsrc = ExecutionContext.get().datasource
+        jsrc = ExecutionContext.get_or_create().datasource
 
         if source := self.source:
             source.save()
@@ -522,7 +523,7 @@ def destroy(self) -> None:
         if self.architype and self.current_access_level == 1:
             from .context import ExecutionContext
 
-            jsrc = ExecutionContext.get().datasource
+            jsrc = ExecutionContext.get_or_create().datasource
 
             source = self.source
             target = self.target
@@ -601,14 +602,14 @@ def ref(cls, ref_id: str) -> Optional[WalkerAnchor]:
     def _save(self) -> None:
         from .context import ExecutionContext
 
-        ExecutionContext.get().datasource.set(self)
+        ExecutionContext.get_or_create().datasource.set(self)
 
     def destroy(self) -> None:
         """Delete Anchor."""
         if self.architype and self.current_access_level > 1:
             from .context import ExecutionContext
 
-            ExecutionContext.get().datasource.remove(self)
+            ExecutionContext.get_or_create().datasource.remove(self)
 
     def sync(self, node: Optional["NodeAnchor"] = None) -> Optional[WalkerArchitype]:
         """Retrieve the Architype from db and return."""

jaclang/core/context.py

@@ -63,7 +63,7 @@ def validate_access(self) -> bool:
         return self.root.has_read_access(self.entry)
 
     @staticmethod
-    def get(options: Optional[dict[str, Any]] = None) -> ExecutionContext:
+    def get_or_create(options: Optional[dict[str, Any]] = None) -> ExecutionContext:
         """Get or create execution context."""
         if not isinstance(ctx := EXECUTION_CONTEXT.get(None), ExecutionContext):
             EXECUTION_CONTEXT.set(ctx := ExecutionContext(**options or {}))

jaclang/plugin/default.py

@@ -68,7 +68,7 @@ class JacFeatureDefaults:
     @hookimpl
     def context(options: Optional[dict[str, Any]]) -> ExecutionContext:
         """Get the execution context."""
-        return ExecutionContext.get(options)
+        return ExecutionContext.get_or_create(options)
 
     @staticmethod
     @hookimpl