dao-ai 0.0.28__py3-none-any.whl → 0.0.29__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. dao_ai/providers/databricks.py +7 -0
  2. dao_ai/tools/genie.py +0 -3
  3. {dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/METADATA +3 -2
  4. {dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/RECORD +7 -7
  5. {dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/WHEEL +0 -0
  6. {dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/entry_points.txt +0 -0
  7. {dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/licenses/LICENSE +0 -0
dao_ai/providers/databricks.py CHANGED
@@ -219,6 +219,13 @@ class DatabricksProvider(ServiceProvider):
219
219
  logger.debug("Creating agent...")
220
220
  mlflow.set_registry_uri("databricks-uc")
221
221
 
222
+ # Set up experiment for proper tracking
223
+ experiment: Experiment = self.get_or_create_experiment(config)
224
+ mlflow.set_experiment(experiment_id=experiment.experiment_id)
225
+ logger.debug(
226
+ f"Using experiment: {experiment.name} (ID: {experiment.experiment_id})"
227
+ )
228
+
222
229
  llms: Sequence[LLMModel] = list(config.resources.llms.values())
223
230
  vector_indexes: Sequence[IndexModel] = list(
224
231
  config.resources.vector_stores.values()
dao_ai/tools/genie.py CHANGED
@@ -298,9 +298,6 @@ def create_genie_tool(
298
298
  if isinstance(genie_room, dict):
299
299
  genie_room = GenieRoomModel(**genie_room)
300
300
 
301
- space_id: AnyVariable = genie_room.space_id or os.environ.get(
302
- "DATABRICKS_GENIE_SPACE_ID"
303
- )
304
301
  space_id: AnyVariable = genie_room.space_id or os.environ.get(
305
302
  "DATABRICKS_GENIE_SPACE_ID"
306
303
  )
{dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dao-ai
3
- Version: 0.0.28
3
+ Version: 0.0.29
4
4
  Summary: DAO AI: A modular, multi-agent orchestration framework for complex AI workflows. Supports agent handoff, tool integration, and dynamic configuration via YAML.
5
5
  Project-URL: Homepage, https://github.com/natefleming/dao-ai
6
6
  Project-URL: Documentation, https://natefleming.github.io/dao-ai
@@ -18,12 +18,13 @@ Classifier: Intended Audience :: Science/Research
18
18
  Classifier: License :: OSI Approved :: MIT License
19
19
  Classifier: Operating System :: OS Independent
20
20
  Classifier: Programming Language :: Python :: 3
21
+ Classifier: Programming Language :: Python :: 3.11
21
22
  Classifier: Programming Language :: Python :: 3.12
22
23
  Classifier: Programming Language :: Python :: 3.13
23
24
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
24
25
  Classifier: Topic :: Software Development :: Libraries :: Python Modules
25
26
  Classifier: Topic :: System :: Distributed Computing
26
- Requires-Python: >=3.12
27
+ Requires-Python: >=3.11
27
28
  Requires-Dist: databricks-agents>=1.7.0
28
29
  Requires-Dist: databricks-langchain>=0.8.1
29
30
  Requires-Dist: databricks-mcp>=0.3.0
{dao_ai-0.0.28.dist-info → dao_ai-0.0.29.dist-info}/RECORD RENAMED
@@ -22,11 +22,11 @@ dao_ai/memory/core.py,sha256=DnEjQO3S7hXr3CDDd7C2eE7fQUmcCS_8q9BXEgjPH3U,4271
22
22
  dao_ai/memory/postgres.py,sha256=vvI3osjx1EoU5GBA6SCUstTBKillcmLl12hVgDMjfJY,15346
23
23
  dao_ai/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
24
  dao_ai/providers/base.py,sha256=-fjKypCOk28h6vioPfMj9YZSw_3Kcbi2nMuAyY7vX9k,1383
25
- dao_ai/providers/databricks.py,sha256=W_lXSMbPTULMAx-KW7zBJfP7LtkcPGRnEfGcSYuu708,65824
25
+ dao_ai/providers/databricks.py,sha256=QwQS0Mad6nZNOB2k_h3lTrwLRzI5ApLQjxOz4CpvhlI,66130
26
26
  dao_ai/tools/__init__.py,sha256=G5-5Yi6zpQOH53b5IzLdtsC6g0Ep6leI5GxgxOmgw7Q,1203
27
27
  dao_ai/tools/agent.py,sha256=WbQnyziiT12TLMrA7xK0VuOU029tdmUBXbUl-R1VZ0Q,1886
28
28
  dao_ai/tools/core.py,sha256=Kei33S8vrmvPOAyrFNekaWmV2jqZ-IPS1QDSvU7RZF0,1984
29
- dao_ai/tools/genie.py,sha256=R9Vl72s6NUtfTrQcdfhNtJFOw-BrKs_5JGt-3dAQGjA,15034
29
+ dao_ai/tools/genie.py,sha256=wt6pVykukNStOUlrTyjWUj-S2Wb47IuMr5HztEcN3Wg,14925
30
30
  dao_ai/tools/human_in_the_loop.py,sha256=yk35MO9eNETnYFH-sqlgR-G24TrEgXpJlnZUustsLkI,3681
31
31
  dao_ai/tools/mcp.py,sha256=5aQoRtx2z4xm6zgRslc78rSfEQe-mfhqov2NsiybYfc,8416
32
32
  dao_ai/tools/python.py,sha256=XcQiTMshZyLUTVR5peB3vqsoUoAAy8gol9_pcrhddfI,1831
@@ -34,8 +34,8 @@ dao_ai/tools/slack.py,sha256=SCvyVcD9Pv_XXPXePE_fSU1Pd8VLTEkKDLvoGTZWy2Y,4775
34
34
  dao_ai/tools/time.py,sha256=Y-23qdnNHzwjvnfkWvYsE7PoWS1hfeKy44tA7sCnNac,8759
35
35
  dao_ai/tools/unity_catalog.py,sha256=uX_h52BuBAr4c9UeqSMI7DNz3BPRLeai5tBVW4sJqRI,13113
36
36
  dao_ai/tools/vector_search.py,sha256=h6yCgEtOA3h-anJG0hGB3VvcmC3Os7_qnhz8xxRjv1E,11346
37
- dao_ai-0.0.28.dist-info/METADATA,sha256=xVgC8iAxjfXmuRz_FzITQTXgZgbTcmEBAiY-DYvGWSk,42727
38
- dao_ai-0.0.28.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
39
- dao_ai-0.0.28.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
40
- dao_ai-0.0.28.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
41
- dao_ai-0.0.28.dist-info/RECORD,,
37
+ dao_ai-0.0.29.dist-info/METADATA,sha256=ze54m_V26Rh451YBjImiAAf_ZgZ3j41XscMLMoSrHGc,42778
38
+ dao_ai-0.0.29.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
39
+ dao_ai-0.0.29.dist-info/entry_points.txt,sha256=Xa-UFyc6gWGwMqMJOt06ZOog2vAfygV_DSwg1AiP46g,43
40
+ dao_ai-0.0.29.dist-info/licenses/LICENSE,sha256=YZt3W32LtPYruuvHE9lGk2bw6ZPMMJD8yLrjgHybyz4,1069
41
+ dao_ai-0.0.29.dist-info/RECORD,,