dao-ai 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

dao_ai/config.py

@@ -1,5 +1,6 @@
 import atexit
 import os
+import sys
 from abc import ABC, abstractmethod
 from enum import Enum
 from os import PathLike
@@ -13,7 +14,7 @@ from typing import (
     TypeAlias,
     Union,
 )
-
+import importlib
 from databricks.sdk import WorkspaceClient
 from databricks.sdk.credentials_provider import (
     CredentialsStrategy,
@@ -28,6 +29,7 @@ from langchain_core.language_models import LanguageModelLike
 from langchain_core.runnables.base import RunnableLike
 from langchain_openai import ChatOpenAI
 from langgraph.checkpoint.base import BaseCheckpointSaver
+from langgraph.graph.state import CompiledStateGraph
 from langgraph.store.base import BaseStore
 from loguru import logger
 from mlflow.models import ModelConfig
@@ -41,6 +43,7 @@ from mlflow.models.resources import (
     DatabricksUCConnection,
     DatabricksVectorSearchIndex,
 )
+from mlflow.pyfunc import ChatModel
 from pydantic import BaseModel, ConfigDict, Field, field_serializer, model_validator
 
 
@@ -1067,6 +1070,16 @@ class AppModel(BaseModel):
 
         return self
 
+    @model_validator(mode="after")
+    def add_code_paths_to_sys_path(self):
+        for code_path in self.code_paths:
+            parent_path: str = str(Path(code_path).parent)
+            if parent_path not in sys.path:
+                sys.path.insert(0, parent_path)
+                logger.debug(f"Added code path to sys.path: {parent_path}")
+        importlib.invalidate_caches()
+        return self
+
 
 class GuidelineModel(BaseModel):
     model_config = ConfigDict(use_enum_values=True, extra="forbid")
@@ -1320,3 +1333,16 @@ class AppConfig(BaseModel):
         return [
             guardrail for guardrail in self.guardrails.values() if predicate(guardrail)
         ]
+
+    def as_graph(self) -> CompiledStateGraph:
+        from dao_ai.graph import create_dao_ai_graph
+
+        graph: CompiledStateGraph = create_dao_ai_graph(config=self)
+        return graph
+
+    def as_chat_model(self) -> ChatModel:
+        from dao_ai.models import create_agent
+
+        graph: CompiledStateGraph = self.as_graph()
+        app: ChatModel = create_agent(graph)
+        return app

dao_ai/models.py

@@ -266,7 +266,7 @@ def _process_config_messages_stream(
 def process_messages_stream(
     app: LanggraphChatModel,
     messages: Sequence[BaseMessage] | Sequence[ChatMessage] | dict[str, Any],
-    custom_inputs: dict[str, Any],
+    custom_inputs: Optional[dict[str, Any]] = None,
 ) -> Generator[ChatCompletionChunk | AIMessageChunk, None, None]:
     """
     Process messages through a ChatAgent in streaming mode.

dao_ai/providers/databricks.py

@@ -1,4 +1,5 @@
 import base64
+from importlib.metadata import version
 from pathlib import Path
 from typing import Any, Callable, Final, Sequence
 
@@ -57,7 +58,12 @@ from dao_ai.config import (
 )
 from dao_ai.models import get_latest_model_version
 from dao_ai.providers.base import ServiceProvider
-from dao_ai.utils import get_installed_packages, is_installed, normalize_name
+from dao_ai.utils import (
+    get_installed_packages,
+    is_installed,
+    is_lib_provided,
+    normalize_name,
+)
 from dao_ai.vector_search import endpoint_exists, index_exists
 
 MAX_NUM_INDEXES: Final[int] = 50
@@ -262,10 +268,6 @@ class DatabricksProvider(ServiceProvider):
         )
         logger.debug(f"auth_policy: {auth_policy}")
 
-        pip_requirements: Sequence[str] = (
-            get_installed_packages() + config.app.pip_requirements
-        )
-
         code_paths: list[str] = config.app.code_paths
         for path in code_paths:
             path = Path(path)
@@ -275,10 +277,13 @@ class DatabricksProvider(ServiceProvider):
         model_root_path: Path = Path(dao_ai.__file__).parent
         model_path: Path = model_root_path / "agent_as_code.py"
 
+        pip_requirements: Sequence[str] = config.app.pip_requirements
+
         if is_installed():
-            pip_requirements += [
-                f"dao-ai=={dao_ai.__version__}",
-            ]
+            if not is_lib_provided("dao-ai", pip_requirements):
+                pip_requirements += [
+                    f"dao-ai=={version('dao-ai')}",
+                ]
         else:
             src_path: Path = model_root_path.parent
             directories: Sequence[Path] = [d for d in src_path.iterdir() if d.is_dir()]
@@ -286,6 +291,8 @@ class DatabricksProvider(ServiceProvider):
                 directory: Path
                 code_paths.append(directory.as_posix())
 
+            pip_requirements += get_installed_packages()
+
         logger.debug(f"pip_requirements: {pip_requirements}")
         logger.debug(f"code_paths: {code_paths}")
 

dao_ai/utils.py

@@ -11,6 +11,13 @@ from loguru import logger
 import dao_ai
 
 
+def is_lib_provided(lib_name: str, pip_requirements: Sequence[str]) -> bool:
+    return any(
+        re.search(rf"\b{re.escape(lib_name)}\b", requirement)
+        for requirement in pip_requirements
+    )
+
+
 def is_installed():
     current_file = os.path.abspath(dao_ai.__file__)
     site_packages = [os.path.abspath(path) for path in site.getsitepackages()]
@@ -85,6 +92,7 @@ def load_function(function_name: str) -> Callable[..., Any]:
         >>> version = func("my_model")
     """
     logger.debug(f"Loading function: {function_name}")
+        
     try:
         # Split the FQN into module path and function name
         module_path, func_name = function_name.rsplit(".", 1)

{dao_ai-0.0.7.dist-info → dao_ai-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao-ai
-Version: 0.0.7
+Version: 0.0.9
 Summary: DAO AI: A modular, multi-agent orchestration framework for complex AI workflows. Supports agent handoff, tool integration, and dynamic configuration via YAML.
 Project-URL: Homepage, https://github.com/natefleming/dao-ai
 Project-URL: Documentation, https://natefleming.github.io/dao-ai

{dao_ai-0.0.7.dist-info → dao_ai-0.0.9.dist-info}/RECORD

@@ -3,16 +3,16 @@ dao_ai/agent_as_code.py,sha256=WjBUJzHMAEX7joMaQGgBgnofiKxjJjdQEVsBJ4VSkPA,655
 dao_ai/catalog.py,sha256=sPZpHTD3lPx4EZUtIWeQV7VQM89WJ6YH__wluk1v2lE,4947
 dao_ai/chat_models.py,sha256=uhwwOTeLyHWqoTTgHrs4n5iSyTwe4EQcLKnh3jRxPWI,8626
 dao_ai/cli.py,sha256=Aez2TQW3Q8Ho1IaIkRggt0NevDxAAVPjXkePC5GPJF0,20429
-dao_ai/config.py,sha256=R5Z0upRQMP8Owfqm-SDiQkixUIKVcQx_KVhvCeWAeIU,43277
+dao_ai/config.py,sha256=b6tSRMCLPIG4WeDagXIEJVcTHq9ldZILyGe8o_k_yaw,44196
 dao_ai/graph.py,sha256=rIm6cLsWwViB3L1dIZp9qc-U-JgFNB5ngEi22Y3iVGQ,7806
 dao_ai/guardrails.py,sha256=-Qh0f_2Db9t4Nbrrx9FM7tnpqShjMoyxepZ0HByItfU,4027
 dao_ai/messages.py,sha256=tRZQTeb5YFKu8cm1xeaCkKhidq-0tdzncNEzVePvits,6806
-dao_ai/models.py,sha256=ikfDg2U08meT-b01iGsfkqqy2u_fdlNMSFzVT78taHY,11819
+dao_ai/models.py,sha256=wME6hS7w2BTS4SUlhEYuCExMOYStnWk-YWsyatAHYyY,11836
 dao_ai/nodes.py,sha256=nusBk8tBLY4JlkmzObCKwDe5JR11A8XzYcenC-yS43o,8406
 dao_ai/prompts.py,sha256=vpmIbWs_szXUgNNDs5Gh2LcxKZti5pHDKSfoClUcgX0,1289
 dao_ai/state.py,sha256=GwbMbd1TWZx1T5iQrEOX6_rpxOitlmyeJ8dMr2o_pag,1031
 dao_ai/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dao_ai/utils.py,sha256=f4j8HhAxkvFovOfa-AqjEH_635NhyEclvnEbDxU3y1M,3949
+dao_ai/utils.py,sha256=VcYeGrXsPAKwXPNL6k5DEfzNGT3T_J9WxVJ9hDf__s8,4165
 dao_ai/vector_search.py,sha256=jlaFS_iizJ55wblgzZmswMM3UOL-qOp2BGJc0JqXYSg,2839
 dao_ai/hooks/__init__.py,sha256=LlHGIuiZt6vGW8K5AQo1XJEkBP5vDVtMhq0IdjcLrD4,417
 dao_ai/hooks/core.py,sha256=ZShHctUSoauhBgdf1cecy9-D7J6-sGn-pKjuRMumW5U,6663
@@ -22,7 +22,7 @@ dao_ai/memory/core.py,sha256=K45iCEFbqJCVxMi4m3vmBJi4c6TQ-UtKGzyugDTkPP0,4141
 dao_ai/memory/postgres.py,sha256=YILzA7xtqawPAOLFaGG_i17zW7cQxXTzTD8yd-ipe8k,12480
 dao_ai/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dao_ai/providers/base.py,sha256=-fjKypCOk28h6vioPfMj9YZSw_3Kcbi2nMuAyY7vX9k,1383
-dao_ai/providers/databricks.py,sha256=8aj2O_m4w55U3p7ZdP48m_jhbknANWv-9wiFtkygw8w,27525
+dao_ai/providers/databricks.py,sha256=vV11SptrzIBEQQyiUqrysMfdpKKPrhVPvUVeEOQxx0o,27684
 dao_ai/tools/__init__.py,sha256=ye6MHaJY7tUnJ8336YJiLxuZr55zDPNdOw6gm7j5jlc,1103
 dao_ai/tools/agent.py,sha256=_XMz6HtrybpVthhRyStADechF6vXLFyK97i01XTBhtw,1868
 dao_ai/tools/core.py,sha256=Kei33S8vrmvPOAyrFNekaWmV2jqZ-IPS1QDSvU7RZF0,1984
@@ -33,8 +33,8 @@ dao_ai/tools/python.py,sha256=XcQiTMshZyLUTVR5peB3vqsoUoAAy8gol9_pcrhddfI,1831
 dao_ai/tools/time.py,sha256=Y-23qdnNHzwjvnfkWvYsE7PoWS1hfeKy44tA7sCnNac,8759
 dao_ai/tools/unity_catalog.py,sha256=PXfLj2EgyQgaXq4Qq3t25AmTC4KyVCF_-sCtg6enens,1404
 dao_ai/tools/vector_search.py,sha256=kLveW-JGwc4IbJ7fKclJFmDXhP3h5XBrjljgab2tRD4,2559
-dao_ai-0.0.7.dist-info/METADATA,sha256=kfyy-tajwZzHOTk9L7kxgQj00basIM44nB0BBWhn22k,41169
-dao_ai-0.0.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dao_ai-0.0.7.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
-dao_ai-0.0.7.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
-dao_ai-0.0.7.dist-info/RECORD,,
+dao_ai-0.0.9.dist-info/METADATA,sha256=wpLMfCr0n0x-1tkDKnuAyiWJIBl1xbWTkNLrKCGjI_s,41169
+dao_ai-0.0.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dao_ai-0.0.9.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
+dao_ai-0.0.9.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
+dao_ai-0.0.9.dist-info/RECORD,,