jaclang 0.5.18__py3-none-any.whl → 0.6.0__py3-none-any.whl

jaclang/core/llms/ollama.py

@@ -0,0 +1,78 @@
+"""Ollama client for MTLLM."""
+
+from .base import BaseLLM
+
+REASON_SUFFIX = """
+Reason and return the output results(s) only such that <Output> should be eval(<Output>) Compatible and reflects the
+expected output type, Follow the format below to provide the reasoning for the output result(s).
+
+[Reasoning] <Reasoning>
+[Output] <Output>
+"""
+
+NORMAL_SUFFIX = """Return the output result(s) only such that <Output> should be eval(<Output>) Compatible and
+reflects the expected output type, Follow the format below to provide the output result(s).
+
+[Output] <Output>
+"""  # noqa E501
+
+CHAIN_OF_THOUGHT_SUFFIX = """
+Generate and return the output result(s) only, adhering to the provided Type in the following format. Perform the operation in a chain of thoughts.(Think Step by Step)
+
+[Chain of Thoughts] <Chain of Thoughts>
+[Output] <Result>
+"""  # noqa E501
+
+REACT_SUFFIX = """
+"""  # noqa E501
+
+
+class Ollama(BaseLLM):
+    """Ollama API client for Large Language Models (LLMs)."""
+
+    MTLLM_METHOD_PROMPTS: dict[str, str] = {
+        "Normal": NORMAL_SUFFIX,
+        "Reason": REASON_SUFFIX,
+        "Chain-of-Thoughts": CHAIN_OF_THOUGHT_SUFFIX,
+        "ReAct": REACT_SUFFIX,
+    }
+
+    def __init__(
+        self, verbose: bool = False, max_tries: int = 10, **kwargs: dict
+    ) -> None:
+        """Initialize the Ollama API client."""
+        import ollama  # type: ignore
+
+        self.client = ollama.Client(host=kwargs.get("host", "http://localhost:11434"))
+        self.verbose = verbose
+        self.max_tries = max_tries
+        self.model_name = kwargs.get("model_name", "phi3")
+        self.default_model_params = {
+            k: v for k, v in kwargs.items() if k not in ["model_name", "host"]
+        }
+
+    def __infer__(self, meaning_in: str, **kwargs: dict) -> str:
+        """Infer a response from the input meaning."""
+        model = str(kwargs.get("model_name", self.model_name))
+        if not self.check_model(model):
+            self.download_model(model)
+        model_params = {k: v for k, v in kwargs.items() if k not in ["model_name"]}
+        messages = [{"role": "user", "content": meaning_in}]
+        output = self.client.chat(
+            model=model,
+            messages=messages,
+            options={**self.default_model_params, **model_params},
+        )
+        return output["message"]["content"]
+
+    def check_model(self, model_name: str) -> bool:
+        """Check if the model is available."""
+        try:
+            self.client.show(model_name)
+            return True
+        except Exception:
+            return False
+
+    def download_model(self, model_name: str) -> None:
+        """Download the model."""
+        self.client.pull(model_name)

jaclang/core/llms/openai.py

@@ -0,0 +1,61 @@
+"""Anthropic API client for MTLLM."""
+
+from .base import BaseLLM
+
+
+REASON_SUFFIX = """
+Reason and return the output result(s) only, adhering to the provided Type in the following format
+
+[Reasoning] <Reason>
+[Output] <Result>
+"""
+
+NORMAL_SUFFIX = """Generate and return the output result(s) only, adhering to the provided Type in the following format
+
+[Output] <result>
+"""  # noqa E501
+
+CHAIN_OF_THOUGHT_SUFFIX = """
+Generate and return the output result(s) only, adhering to the provided Type in the following format. Perform the operation in a chain of thoughts.(Think Step by Step)
+
+[Chain of Thoughts] <Chain of Thoughts>
+[Output] <Result>
+"""  # noqa E501
+
+REACT_SUFFIX = """
+"""  # noqa E501
+
+
+class OpenAI(BaseLLM):
+    """Anthropic API client for MTLLM."""
+
+    MTLLM_METHOD_PROMPTS: dict[str, str] = {
+        "Normal": NORMAL_SUFFIX,
+        "Reason": REASON_SUFFIX,
+        "Chain-of-Thoughts": CHAIN_OF_THOUGHT_SUFFIX,
+        "ReAct": REACT_SUFFIX,
+    }
+
+    def __init__(
+        self, verbose: bool = False, max_tries: int = 10, **kwargs: dict
+    ) -> None:
+        """Initialize the Anthropic API client."""
+        import openai  # type: ignore
+
+        self.client = openai.OpenAI()
+        self.verbose = verbose
+        self.max_tries = max_tries
+        self.model_name = kwargs.get("model_name", "gpt-3.5-turbo")
+        self.temperature = kwargs.get("temperature", 0.7)
+        self.max_tokens = kwargs.get("max_tokens", 1024)
+
+    def __infer__(self, meaning_in: str, **kwargs: dict) -> str:
+        """Infer a response from the input meaning."""
+        messages = [{"role": "user", "content": meaning_in}]
+        output = self.client.chat.completions.create(
+            model=kwargs.get("model_name", self.model_name),
+            temperature=kwargs.get("temperature", self.temperature),
+            max_tokens=kwargs.get("max_tokens", self.max_tokens),
+            messages=messages,
+        )
+        return output.choices[0].message.content

jaclang/core/llms/togetherai.py

@@ -0,0 +1,60 @@
+"""Anthropic API client for MTLLM."""
+
+from .base import BaseLLM
+
+REASON_SUFFIX = """
+Reason and return the output result(s) only, adhering to the provided Type in the following format
+
+[Reasoning] <Reason>
+[Output] <Result>
+"""
+
+NORMAL_SUFFIX = """Generate and return the output result(s) only, adhering to the provided Type in the following format
+
+[Output] <result>
+"""  # noqa E501
+
+CHAIN_OF_THOUGHT_SUFFIX = """
+Generate and return the output result(s) only, adhering to the provided Type in the following format. Perform the operation in a chain of thoughts.(Think Step by Step)
+
+[Chain of Thoughts] <Chain of Thoughts>
+[Output] <Result>
+"""  # noqa E501
+
+REACT_SUFFIX = """
+"""  # noqa E501
+
+
+class TogetherAI(BaseLLM):
+    """Anthropic API client for MTLLM."""
+
+    MTLLM_METHOD_PROMPTS: dict[str, str] = {
+        "Normal": NORMAL_SUFFIX,
+        "Reason": REASON_SUFFIX,
+        "Chain-of-Thoughts": CHAIN_OF_THOUGHT_SUFFIX,
+        "ReAct": REACT_SUFFIX,
+    }
+
+    def __init__(
+        self, verbose: bool = False, max_tries: int = 10, **kwargs: dict
+    ) -> None:
+        """Initialize the Anthropic API client."""
+        import together  # type: ignore
+
+        self.client = together.Together()
+        self.verbose = verbose
+        self.max_tries = max_tries
+        self.model_name = kwargs.get("model_name", "mistralai/Mistral-7B-Instruct-v0.3")
+        self.temperature = kwargs.get("temperature", 0.7)
+        self.max_tokens = kwargs.get("max_tokens", 1024)
+
+    def __infer__(self, meaning_in: str, **kwargs: dict) -> str:
+        """Infer a response from the input meaning."""
+        messages = [{"role": "user", "content": meaning_in}]
+        output = self.client.chat.completions.create(
+            model=kwargs.get("model_name", self.model_name),
+            temperature=kwargs.get("temperature", self.temperature),
+            max_tokens=kwargs.get("max_tokens", self.max_tokens),
+            messages=messages,
+        )
+        return output.choices[0].message.content

jaclang/core/llms/utils.py

@@ -0,0 +1,9 @@
+"""Utility functions for the LLMs module."""
+
+try:
+    from loguru import logger  # noqa F401
+except ImportError:
+    import logging
+
+    logging.basicConfig(level=logging.INFO)
+    logger = logging.getLogger(__name__)

jaclang/core/utils.py

@@ -3,7 +3,9 @@
 from __future__ import annotations
 
 import ast as ast3
-from typing import Callable, TYPE_CHECKING
+import sys
+from contextlib import contextmanager
+from typing import Callable, Iterator, TYPE_CHECKING
 
 import jaclang.compiler.absyntree as ast
 from jaclang.core.registry import SemScope
@@ -12,6 +14,19 @@ if TYPE_CHECKING:
     from jaclang.core.construct import NodeAnchor, NodeArchitype
 
 
+@contextmanager
+def sys_path_context(path: str) -> Iterator[None]:
+    """Add a path to sys.path temporarily."""
+    novel_path = path not in sys.path
+    try:
+        if novel_path:
+            sys.path.append(path)
+        yield
+    finally:
+        if novel_path:
+            sys.path.remove(path)
+
+
 def collect_node_connections(
     current_node: NodeAnchor,
     visited_nodes: set,

jaclang/plugin/default.py

@@ -19,7 +19,6 @@ from jaclang.core.aott import (
     get_all_type_explanations,
     get_info_types,
     get_object_string,
-    get_reasoning_output,
     get_type_annotation,
 )
 from jaclang.core.construct import (
@@ -428,10 +427,16 @@ class JacFeatureDefaults:
 
     @staticmethod
     @hookimpl
-    def get_root() -> Architype:
+    def get_root() -> Root:
         """Jac's assign comprehension feature."""
         return root
 
+    @staticmethod
+    @hookimpl
+    def get_root_type() -> Type[Root]:
+        """Jac's root getter."""
+        return Root
+
     @staticmethod
     @hookimpl
     def build_edge(
@@ -587,9 +592,15 @@ class JacFeatureDefaults:
         _scope = SemScope.get_scope_from_str(scope)
         assert _scope is not None
 
-        reason = False
-        if "reason" in model_params:
-            reason = model_params.pop("reason")
+        method = model_params.pop("method") if "method" in model_params else "Normal"
+        available_methods = model.MTLLM_METHOD_PROMPTS.keys()
+        assert (
+            method in available_methods
+        ), f"Invalid method: {method}. Select from {available_methods}"
+
+        context = (
+            "\n".join(model_params.pop("context")) if "context" in model_params else ""
+        )
 
         type_collector: list = []
         information, collected_types = get_info_types(_scope, mod_registry, incl_info)
@@ -605,22 +616,34 @@ class JacFeatureDefaults:
 
         output_information = f"{outputs[0]} ({outputs[1]})"
         type_collector.extend(extract_non_primary_type(outputs[1]))
+        output_type_explanations = "\n".join(
+            list(
+                get_all_type_explanations(
+                    extract_non_primary_type(outputs[1]), mod_registry
+                ).values()
+            )
+        )
 
         type_explanations_list = list(
             get_all_type_explanations(type_collector, mod_registry).values()
         )
         type_explanations = "\n".join(type_explanations_list)
 
-        meaning_in = aott_raise(
-            information,
-            inputs_information,
-            output_information,
-            type_explanations,
-            action,
-            reason,
+        meaning_out = aott_raise(
+            model=model,
+            information=information,
+            inputs_information=inputs_information,
+            output_information=output_information,
+            type_explanations=type_explanations,
+            action=action,
+            context=context,
+            method=method,
+            tools=[],
+            model_params=model_params,
+        )
+        output = model.resolve_output(
+            meaning_out, outputs[0], outputs[1], output_type_explanations
         )
-        meaning_out = model.__infer__(meaning_in, **model_params)
-        reasoning, output = get_reasoning_output(meaning_out)
         return output
 
 

jaclang/plugin/feature.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import types
-from typing import Any, Callable, Optional, Type, TypeAlias, Union
+from typing import Any, Callable, Optional, Type, Union
 
 from jaclang.compiler.absyntree import Module
 from jaclang.core.construct import (
@@ -31,8 +31,6 @@ class JacFeature:
     from jaclang.plugin.spec import DSFunc
     from jaclang.compiler.constant import EdgeDir
 
-    RootType: TypeAlias = Root
-
     @staticmethod
     def make_architype(
         cls: type,
@@ -82,7 +80,7 @@ class JacFeature:
         mdl_alias: Optional[str] = None,
         override_name: Optional[str] = None,
         mod_bundle: Optional[Module] = None,
-        lng: Optional[str] = None,
+        lng: Optional[str] = "jac",
         items: Optional[dict[str, Union[str, bool]]] = None,
     ) -> Optional[types.ModuleType]:
         """Core Import Process."""
@@ -218,10 +216,15 @@ class JacFeature:
         return pm.hook.assign_compr(target=target, attr_val=attr_val)
 
     @staticmethod
-    def get_root() -> Architype:
-        """Jac's assign comprehension feature."""
+    def get_root() -> Root:
+        """Jac's root getter."""
         return pm.hook.get_root()
 
+    @staticmethod
+    def get_root_type() -> Type[Root]:
+        """Jac's root type getter."""
+        return pm.hook.get_root_type()
+
     @staticmethod
     def build_edge(
         is_undirected: bool,

jaclang/plugin/spec.py

@@ -12,6 +12,7 @@ from jaclang.plugin.default import (
     EdgeArchitype,
     EdgeDir,
     NodeArchitype,
+    Root,
     WalkerArchitype,
 )
 
@@ -198,7 +199,13 @@ class JacFeatureSpec:
 
     @staticmethod
     @hookspec(firstresult=True)
-    def get_root() -> Architype:
+    def get_root() -> Root:
+        """Jac's root getter."""
+        raise NotImplementedError
+
+    @staticmethod
+    @hookspec(firstresult=True)
+    def get_root_type() -> Type[Root]:
         """Jac's root getter."""
         raise NotImplementedError
 

jaclang/settings.py

@@ -10,7 +10,7 @@ class Settings:
     """Main settings of Jac lang."""
 
     fuse_type_info_debug: bool = False
-    jac_proc_debug: bool = False
+    py_raise: bool = False
 
     def __post_init__(self) -> None:
         """Initialize settings."""

jaclang/utils/helpers.py

@@ -131,7 +131,6 @@ def import_target_to_relative_path(
     import_level: int,
     import_target: str,
     base_path: Optional[str] = None,
-    file_extension: str = ".jac",
 ) -> str:
     """Convert an import target string into a relative file path."""
     if not base_path:
@@ -141,7 +140,12 @@ def import_target_to_relative_path(
     actual_parts = parts[traversal_levels:]
     for _ in range(traversal_levels):
         base_path = os.path.dirname(base_path)
-    relative_path = os.path.join(base_path, *actual_parts) + file_extension
+    relative_path = os.path.join(base_path, *actual_parts)
+    relative_path = (
+        relative_path + ".jac"
+        if os.path.exists(relative_path + ".jac")
+        else relative_path
+    )
     return relative_path
 
 

jaclang/utils/treeprinter.py

@@ -88,16 +88,19 @@ def print_ast_tree(
     from jaclang.compiler.absyntree import AstSymbolNode, Token
 
     def __node_repr_in_tree(node: AstNode) -> str:
+        access = (
+            f"Access: {node.access.tag.value}"
+            if isinstance(node, ast.AstAccessNode) and node.access is not None
+            else ""
+        )
         if isinstance(node, Token) and isinstance(node, AstSymbolNode):
-            return (
-                f"{node.__class__.__name__} - {node.value} - Type: {node.sym_info.typ}"
-            )
+            return f"{node.__class__.__name__} - {node.value} - Type: {node.sym_info.typ}, {access}"
         elif isinstance(node, Token):
-            return f"{node.__class__.__name__} - {node.value}"
+            return f"{node.__class__.__name__} - {node.value}, {access}"
         elif isinstance(node, AstSymbolNode):
-            return f"{node.__class__.__name__} - {node.sym_name} - Type: {node.sym_info.typ}"
+            return f"{node.__class__.__name__} - {node.sym_name} - Type: {node.sym_info.typ}, {access}"
         else:
-            return f"{node.__class__.__name__}"
+            return f"{node.__class__.__name__}, {access}"
 
     def __node_repr_in_py_tree(node: ast3.AST) -> str:
         if isinstance(node, ast3.Constant):

jaclang-0.6.0.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.1
+Name: jaclang
+Version: 0.6.0
+Home-page: https://github.com/Jaseci-Labs/jaclang
+Author: Jason Mars
+Author-email: jason@jaseci.org
+Provides-Extra: llms
+Requires-Dist: transformers ; extra == 'llms'
+Requires-Dist: accelerator ; extra == 'llms'
+Requires-Dist: torch ; extra == 'llms'
+Requires-Dist: ollama ; extra == 'llms'
+Requires-Dist: anthropic ; extra == 'llms'
+Requires-Dist: groq ; extra == 'llms'
+Requires-Dist: openai ; extra == 'llms'
+Requires-Dist: together ; extra == 'llms'
+Requires-Dist: loguru ; extra == 'llms'
+

{jaclang-0.5.18.dist-info → jaclang-0.6.0.dist-info}/RECORD

@@ -1,40 +1,41 @@
 jaclang/__init__.py,sha256=vTvt9LIz5RNKrLRueH3gUM3KTAjsqoanHQ_Pn9m4i9g,675
-jaclang/settings.py,sha256=qu17a7AenlX1vUnZTHWnaKvwqwKQxWiYueWjLr6MWKY,3162
+jaclang/settings.py,sha256=2bLDsF1xiFI9VL8vEIPZ5A5Q-XQ9hfK9lIszH99nKkc,3156
 jaclang/cli/__init__.py,sha256=7aaPgYddIAHBvkdv36ngbfwsimMnfGaTDcaHYMg_vf4,23
-jaclang/cli/cli.py,sha256=_z3PmgTvLS_zkFe-BC7uvmKKJ2ZQqDhQAQpAAlebZOg,10687
+jaclang/cli/cli.py,sha256=O4obezeiQKGsYqt4_ZN3pGBL7xjA2d_wcxgEVqZ959g,10806
 jaclang/cli/cmdreg.py,sha256=bn2UdOkNbE-4zfbomO2j8rTtkXhsltH4jE5rKqA5HbY,7862
-jaclang/compiler/__init__.py,sha256=X_n8G4KrjWePCGNC1Leo3d1is4MBer4xCHc0A4983-g,2937
-jaclang/compiler/absyntree.py,sha256=quVRgeMZl0AcZ5It5qwKOudLFl7MqOzHcHiERIjUBEo,126927
+jaclang/compiler/__init__.py,sha256=Fq2rBjvbFxTbolEC872X5CicOjyJmKo0yZlnUcChg6k,3119
+jaclang/compiler/absyntree.py,sha256=VBqAZZQ9zRS73M55nZaATBoHzlDtRv86ogwyM4fjPZ0,126939
 jaclang/compiler/codeloc.py,sha256=KMwf5OMQu3_uDjIdH7ta2piacUtXNPgUV1t8OuLjpzE,2828
 jaclang/compiler/compile.py,sha256=fS6Uvor93EavESKrwadqp7bstcpMRRACvBkqbr4En04,2682
 jaclang/compiler/constant.py,sha256=C8nOgWLAg-fRg8Qax_jRcYp6jGyWSSA1gwzk9Zdy7HE,6534
-jaclang/compiler/jac.lark,sha256=Xz6xlZ2-YVzshh6WJPh35AU8QispGy1zebr1htclf_Y,16995
-jaclang/compiler/parser.py,sha256=kwA6qJXxaZOnhkmNuXav03CH57RJXdohY-bw-jsjDWo,136544
-jaclang/compiler/symtable.py,sha256=SRYSwZtLHXLIOkE9CfdWDkkwx0vDLXvMeYiFfh6-wP4,5769
-jaclang/compiler/workspace.py,sha256=8r7vNhIKgct2eE7sXaSrfy3NzrE9V2rhnaIXDEArgc8,7381
+jaclang/compiler/jac.lark,sha256=Kvq5zZbQMJ-1bdXLVuCafRVS9w91UfN-0nLnEq0gOSw,17019
+jaclang/compiler/parser.py,sha256=EtASMImnMyrKXixKpHYStBlhdDxVwtsIX306EWC8AHg,138177
+jaclang/compiler/symtable.py,sha256=f-9c3eQJTDhEG22kYC3Vd4i5gRWf77_CsqFm7t8Wsng,6031
+jaclang/compiler/workspace.py,sha256=I1MF2jaSylDOcQQV4Ez3svi4FdOyfmbNupuZT7q9aFQ,7797
 jaclang/compiler/generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaclang/compiler/generated/jac_parser.py,sha256=wwJA422L2G8eor8eqgv04pshhDxsLnYyi5MAE4lDKQE,334286
+jaclang/compiler/generated/jac_parser.py,sha256=Dn9D__4gJTGRYcfXgbkcwICHXe1qwMs2x0MkSuN3U_U,338234
 jaclang/compiler/passes/__init__.py,sha256=0Tw0d130ZjzA05jVcny9cf5NfLjlaM70PKqFnY4zqn4,69
 jaclang/compiler/passes/ir_pass.py,sha256=gh1Zd8ouu79FoxerW1sMxktmfeC9gHyp4H_MoAviYP8,5504
-jaclang/compiler/passes/transform.py,sha256=6t-bbX_s615i7_naOIBjT4wvAkvLFM4niRHYyF4my8A,2086
-jaclang/compiler/passes/main/__init__.py,sha256=jT0AZ3-Cp4VIoJUdUAfwEdjgB0mtMRaqMIw4cjlrLvs,905
+jaclang/compiler/passes/transform.py,sha256=Z3TuHMJ7TyyBHt0XmRWEjMk8-t6logKs7XG_OTDJIvo,2162
+jaclang/compiler/passes/main/__init__.py,sha256=m8vZ0F6EiwOm0e0DWKRYgc8JcgiT2ayBrY6GojZkTfY,945
+jaclang/compiler/passes/main/access_modifier_pass.py,sha256=1tQJ9bNlOVX3sgL0DK4jv643Pu5xl68oL8LNsaHf-MY,4864
 jaclang/compiler/passes/main/def_impl_match_pass.py,sha256=_KgAVVzMAatigKDp1vAnhyY2GcWf0rRoD8MkfYg-POU,3297
 jaclang/compiler/passes/main/def_use_pass.py,sha256=d2REmfme2HjUj8avDcXUuPbv3yKfvYHqpL49sxniLhQ,8576
 jaclang/compiler/passes/main/fuse_typeinfo_pass.py,sha256=9WrF78r4cNWnDqofx225b9cUZafTi3HwCffXINmgoZ0,15968
-jaclang/compiler/passes/main/import_pass.py,sha256=EgMoPDpDNucbh7s4WRkMEfOdEtXM6dtsN0Qorwh489A,6384
-jaclang/compiler/passes/main/pyast_gen_pass.py,sha256=q1GM4zXA-33MsqMlvJhGa8z1ydcGA9hMRHVZle9O_aU,138177
-jaclang/compiler/passes/main/pyast_load_pass.py,sha256=WXq7Ahaev2ZIsPtmXnPgHn6sIGOpRJaAMHtNrJUSHRU,87133
+jaclang/compiler/passes/main/import_pass.py,sha256=JdDcAI2MmdJbisSSsgWHcKnuEmE_PmywSV7G9OzPras,6740
+jaclang/compiler/passes/main/pyast_gen_pass.py,sha256=L24aRhoXIGoXu1FwwW2X3HoGWwjLmOAtbLCGqOrix0g,138795
+jaclang/compiler/passes/main/pyast_load_pass.py,sha256=UDbosNQ88CcfSW1wKBYgLRz3ZBxpJ8cGwpn_KZaYvvk,89725
 jaclang/compiler/passes/main/pybc_gen_pass.py,sha256=CjA9AqyMO3Pv_b5Hh0YI6JmCqIru2ASonO6rhrkau-M,1336
-jaclang/compiler/passes/main/pyout_pass.py,sha256=jw-ApCvVyAqynBFCArPQ20wq2ou0s7lTCGqcUyxrJWI,3018
-jaclang/compiler/passes/main/registry_pass.py,sha256=VEaQKNNZZjRhgMf809X1HpytNzV2W9HyViB2SwQvxNI,4421
-jaclang/compiler/passes/main/schedules.py,sha256=Tn_jr6Lunm3t5tAluxvE493qfa4lF2kgIQPN0J42TEE,1048
+jaclang/compiler/passes/main/pyout_pass.py,sha256=clJkASFiBiki5w0MQXYIwcdvahYETgqAjKlYOYVYKNQ,3137
+jaclang/compiler/passes/main/registry_pass.py,sha256=Lp_EqlflXkSgQuJe_EVBtzC16YAJx70bDO6X0nJIX5U,4579
+jaclang/compiler/passes/main/schedules.py,sha256=Mvwt3aphVh4rhs8Mf5itK4GCIPM00rQ2a4tDzjmAzUk,1167
 jaclang/compiler/passes/main/sub_node_tab_pass.py,sha256=fGgK0CBWsuD6BS4BsLwXPl1b5UC2N2YEOGa6jNMu8N8,1332
-jaclang/compiler/passes/main/sym_tab_build_pass.py,sha256=xLlcsntcVHVZ7YEyp1Jh_tKw93dQP2uqK5D5JFQtOyI,41678
+jaclang/compiler/passes/main/sym_tab_build_pass.py,sha256=TXM_oxTYIyKEXm2NDh6UgKw5KGPgYC1ONEqS4nN5PMQ,41769
 jaclang/compiler/passes/main/type_check_pass.py,sha256=-f41Ukr3B1w4rkQdZ2xJy6nozymgnVKjJ8E1fn7qJmI,3339
 jaclang/compiler/passes/main/tests/__init__.py,sha256=UBAATLUEH3yuBN4LYKnXXV79kokRc4XB-rX12qfu0ds,28
 jaclang/compiler/passes/main/tests/test_decl_def_match_pass.py,sha256=QDsvTEpJt3AuS3GRwfq-1mezRCpx7rwEEyg1AbusaBs,1374
 jaclang/compiler/passes/main/tests/test_def_use_pass.py,sha256=dmyZgXugOCN-_YY7dxkNHgt8vqey8qzNJf8sWyLLZ5w,1013
-jaclang/compiler/passes/main/tests/test_import_pass.py,sha256=KDTIhUnAVyp0iQ64lC6RQvOkdS7CNVtVbfPbG1aX1L8,580
+jaclang/compiler/passes/main/tests/test_import_pass.py,sha256=iFyrdmNuoi1C25qkL7kyUXd55Exl3Cb0iP8vRZQKsC0,586
 jaclang/compiler/passes/main/tests/test_pyast_build_pass.py,sha256=LIT4TP-nhtftRtY5rNySRQlim-dWMSlkfUvkhZTk4pc,1383
 jaclang/compiler/passes/main/tests/test_pyast_gen_pass.py,sha256=fNL_FS26AQGlRCvwWfl-Qyt7iW2_A99GIHOAnnkpw9A,4731
 jaclang/compiler/passes/main/tests/test_pybc_gen_pass.py,sha256=If8PE4exN5g9o1NRElNC0XdfIwJAp7M7f69rzmYRYUQ,655
@@ -45,40 +46,48 @@ jaclang/compiler/passes/main/tests/test_type_check_pass.py,sha256=bzrVcsBZra2HkQ
 jaclang/compiler/passes/main/tests/test_typeinfo_pass.py,sha256=ehC0_giLg7NwB7fR10nW5Te8mZ76qmUFxkK1bEjtmrw,129
 jaclang/compiler/passes/tool/__init__.py,sha256=xekCOXysHIcthWm8NRmQoA1Ah1XV8vFbkfeHphJtUdc,223
 jaclang/compiler/passes/tool/fuse_comments_pass.py,sha256=N9a84qArNuTXX1iaXsBzqcufx6A3zYq2p-1ieH6FmHc,3133
-jaclang/compiler/passes/tool/jac_formatter_pass.py,sha256=PKG6jvJeqzPj3loytwymYwTXT-Dhq5opLmfFKXvwj30,83271
+jaclang/compiler/passes/tool/jac_formatter_pass.py,sha256=oYXDWWlxYj0FGXxXUOPnmwnkFPmayd2fRoWngvDgXwc,86113
 jaclang/compiler/passes/tool/schedules.py,sha256=kmbsCazAMizGAdQuZpFky5BPlYlMXqNw7wOUzdi_wBo,432
 jaclang/compiler/passes/tool/tests/__init__.py,sha256=AeOaZjA1rf6VAr0JqIit6jlcmOzW7pxGr4U1fOwgK_Y,24
 jaclang/compiler/passes/tool/tests/test_fuse_comments_pass.py,sha256=ZeWHsm7VIyyS8KKpoB2SdlHM4jF22fMfSrfTfxt2MQw,398
-jaclang/compiler/passes/tool/tests/test_jac_format_pass.py,sha256=Ii5lDsAlS2cbTQPdU-BHL1Vbd_679_lcu4_IrNqBM1k,5869
+jaclang/compiler/passes/tool/tests/test_jac_format_pass.py,sha256=P4vBRbwVAkb8svFSAOf4VKioZS8BkoaVfYdn-LCrRqA,6144
 jaclang/compiler/passes/tool/tests/test_unparse_validate.py,sha256=Tg9k7BOSkWngAZLvSHmPt95ILJzdIvdHxKBT__mgpS0,2910
 jaclang/compiler/passes/utils/__init__.py,sha256=UsI5rUopTUiStAzup4kbPwIwrnC5ofCrqWBCBbM2-k4,35
 jaclang/compiler/passes/utils/mypy_ast_build.py,sha256=0EJGhsdcIQoXeUbHlqZrPMa9XD69BZ5RcamrfOLKdk4,26057
 jaclang/compiler/tests/__init__.py,sha256=qiXa5UNRBanGOcplFKTT9a_9GEyiv7goq1OzuCjDCFE,27
-jaclang/compiler/tests/test_importer.py,sha256=QSzwYYsAfyU-dcEsvKp1KWZ-gEanWdjIBy5bGY3vUfw,1334
+jaclang/compiler/tests/test_importer.py,sha256=JNmte5FsHhnng9jzw7N5BenflAFCasuhezN1sytDVyg,1739
 jaclang/compiler/tests/test_parser.py,sha256=C81mUo8EGwypPTTLRVS9BglP0Dyye9xaPSQtw7cwnnI,4814
 jaclang/compiler/tests/test_workspace.py,sha256=SEBcvz_daTbonrLHK9FbjiH86TUbOtVGH-iZ3xkJSMk,3184
 jaclang/compiler/tests/fixtures/__init__.py,sha256=udQ0T6rajpW_nMiYKJNckqP8izZ-pH3P4PNTJEln2NU,36
 jaclang/compiler/tests/fixtures/activity.py,sha256=fSvxYDKufsPeQIrbuh031zHw_hdbRv5iK9mS7dD8E54,263
 jaclang/core/__init__.py,sha256=jDDYBCV82qPhmcDVk3NIvHbhng0ebSrXD3xrojg0-eo,34
-jaclang/core/aott.py,sha256=_CdK1l37BK5ZCfA7NjOouH-xveWWbsBCRFz24dvsQZA,7716
+jaclang/core/aott.py,sha256=vqnoEwmAaLlasUDVI6k3d67OtqMquJ1YtgFELvLnKqk,7068
 jaclang/core/construct.py,sha256=_mYHEbUPY3Dn0rcl21eJXLqAGYH_0_C-x1K9HDa2RKk,13633
-jaclang/core/importer.py,sha256=8aNHfUOx_Pw6Knox__n3Gq8uub0pWWmXWsW3kGPFv8k,5437
-jaclang/core/llms.py,sha256=bs0EoQcIZc_45HSxDHSEwUgyw-YhawfX4kQYM16UNnk,4139
+jaclang/core/importer.py,sha256=gVJhQz7nKlIAzx7FF19cgmFQd_iRQuZ19Yz2SShLCPU,6020
 jaclang/core/registry.py,sha256=4uuXahPN4SMVEWwJ6Gjm5LM14rePMGoyjQtSbDQmQZs,4178
-jaclang/core/utils.py,sha256=Li35s4HuDreuNpsE1blv8RL0okK9wXFUqoF-XVgS6ro,7374
+jaclang/core/utils.py,sha256=5e4DyjtSweFxKeIuEyeYGV2MsuTtfnncfIwRdoyFdys,7736
+jaclang/core/llms/__init__.py,sha256=hxuxduH6ahhkXgJ9eUkhfn8LCS1KTz1ESuYgtspu_c0,371
+jaclang/core/llms/anthropic.py,sha256=JmgKLUTk27Hg221rt4QBLRjonxE_qLkkSgZuLFZUrCg,1979
+jaclang/core/llms/base.py,sha256=IOQWOP3zyPq05PVq7iYmh5wTd62m2W0husljvRtLYJo,6082
+jaclang/core/llms/groq.py,sha256=jdcz8DMz4ResI2_0pfJB1rJ8Z8r9f0Av66D3ha0fCB4,2144
+jaclang/core/llms/huggingface.py,sha256=8NqPScBFhodjONc4SbJ1OHyftQ1y9q_a7QevprDeArQ,2656
+jaclang/core/llms/ollama.py,sha256=mlNf844qgiBizVRRBfQt-pM8sN6dX7ESqdo2SXu8aE0,2691
+jaclang/core/llms/openai.py,sha256=uG4BIW8x9NhWMhy9taqb4esIa3POO_cezsZ7ZclLan8,1975
+jaclang/core/llms/togetherai.py,sha256=MTPz7QasO2MjDn8hVCjOVU3SO6H6pF9Qoh7ZHmx46yw,2005
+jaclang/core/llms/utils.py,sha256=mTpROnh4qS8fE26TkcotSt-mAqWoSeEeRBOu0SgoPmA,219
 jaclang/plugin/__init__.py,sha256=qLbQ2KjwyTvlV50Q7JTIznyIzuGDPjuwM_ZJjdk-avo,182
 jaclang/plugin/builtin.py,sha256=D1R4GGNHX96P-wZnGm9OI4dMeCJvuXuHkCw2Cl5vaAU,1201
-jaclang/plugin/default.py,sha256=o6sDfuahxK-MzLSXTJ5rfui1xXiK_g6s_6O54FpC7zs,23654
-jaclang/plugin/feature.py,sha256=mkU1YwlTTvTuZ4FpsLQZCUr3mQtlp5ICAuvEft2K5QA,8782
-jaclang/plugin/spec.py,sha256=AiJ-YJ1UiDNO7qjwdc9vK4aua2LYR7GprpVLze0O7to,7806
+jaclang/plugin/default.py,sha256=QuhAIMZ3Flhn4UVFm8vWVmAfAbqEFx_ekhqb84VO8og,24458
+jaclang/plugin/feature.py,sha256=u4UCLIaZ8DKhtpfKclfC3TUmx-SHfRttUdtgLrHo3Io,8853
+jaclang/plugin/spec.py,sha256=tPeekVbwxyBE2xlkff3FXxfQkAaFsyPKPX_5FjUwfPs,7968
 jaclang/plugin/tests/__init__.py,sha256=rn_tNG8jCHWwBc_rx4yFkGc4N1GISb7aPuTFVRTvrTk,38
 jaclang/plugin/tests/test_features.py,sha256=uNQ52HHc0GiOGg5xUZk-ddafdtud0IHN4H_EUAs4er4,3547
 jaclang/utils/__init__.py,sha256=86LQ_LDyWV-JFkYBpeVHpLaVxkqwFDP60XpWXOFZIQk,46
-jaclang/utils/helpers.py,sha256=RILfJ8Xm8XksHdsY1xQEQDvbw9Wk6jkE74vp6EvngKI,5746
+jaclang/utils/helpers.py,sha256=6tfu2paCSSAUzJMZlMVl4ftbf9HOWN4TaJz7U-S0G1g,5831
 jaclang/utils/lang_tools.py,sha256=odv66Os8bJvw9JIUUkccGWvEPL98c-60veAiOiNyTy4,9956
 jaclang/utils/log.py,sha256=G8Y_DnETgTh9xzvlW5gh9zqJ1ap4YY_MDTwIMu5Uc0Y,262
 jaclang/utils/test.py,sha256=7-2Te0GQcKY3RJfNKa3cEJjG-RMC1r-0fHuOcFXrmE8,5383
-jaclang/utils/treeprinter.py,sha256=RV7mmnlUdla_dYxIrwxsV2F32ncgEnB9PdA3ZnKLxPE,10934
+jaclang/utils/treeprinter.py,sha256=wRWUaEVxnNdcbTYyQ5PJiSJ2PI4UoNTZzHadwSZ_eUE,11117
 jaclang/utils/tests/__init__.py,sha256=8uwVqMsc6cxBAM1DuHLuNuTnzLXqOqM-WRa4ixOMF6w,23
 jaclang/utils/tests/test_lang_tools.py,sha256=YmV1AWieSoqcxWJYnxX1Lz3dZAR0rRvz6billsGJdHY,4144
 jaclang/vendor/__init__.py,sha256=pbflH5hBfDiKfTJEjIXId44Eof8CVmXQlZwFYDG6_4E,35
@@ -410,8 +419,8 @@ jaclang/vendor/pluggy/_manager.py,sha256=fcYU7VER0CplRym4jAJ7RCFYl6cfDSeVM589YHH
 jaclang/vendor/pluggy/_result.py,sha256=wXDoVy68bOaOrBzQ0bWFb3HTbpqhvdQMgYwaZvfzxfA,3239
 jaclang/vendor/pluggy/_tracing.py,sha256=kSBr25F_rNklV2QhLD6h1jx6Z1kcKDRbuYvF5jv35pU,2089
 jaclang/vendor/pluggy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaclang-0.5.18.dist-info/METADATA,sha256=NTG7UE-l9XL9qa5ezDByBq7fSgiPYU9ffU6ZMLR5JEg,153
-jaclang-0.5.18.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-jaclang-0.5.18.dist-info/entry_points.txt,sha256=Z-snS3MDBdV1Z4znXDZTenyyndovaNjqRkDW1t2kYSs,50
-jaclang-0.5.18.dist-info/top_level.txt,sha256=ZOAoLpE67ozkUJd-v3C59wBNXiteRD8IWPbsYB3M08g,8
-jaclang-0.5.18.dist-info/RECORD,,
+jaclang-0.6.0.dist-info/METADATA,sha256=oFL9x1Tyx0ZnVcb1BKmakhCSneRit3Qtabu7QOGiQXk,546
+jaclang-0.6.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+jaclang-0.6.0.dist-info/entry_points.txt,sha256=Z-snS3MDBdV1Z4znXDZTenyyndovaNjqRkDW1t2kYSs,50
+jaclang-0.6.0.dist-info/top_level.txt,sha256=ZOAoLpE67ozkUJd-v3C59wBNXiteRD8IWPbsYB3M08g,8
+jaclang-0.6.0.dist-info/RECORD,,