databricks-agentbricks 0.2.0__tar.gz

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 (159) hide show
  1. databricks_agentbricks-0.2.0/.gitignore +22 -0
  2. databricks_agentbricks-0.2.0/NOTICE +39 -0
  3. databricks_agentbricks-0.2.0/PKG-INFO +897 -0
  4. databricks_agentbricks-0.2.0/README.md +862 -0
  5. databricks_agentbricks-0.2.0/pyproject.toml +106 -0
  6. databricks_agentbricks-0.2.0/src/databricks_agentbricks/__init__.py +1 -0
  7. databricks_agentbricks-0.2.0/src/databricks_agentbricks/_group.py +217 -0
  8. databricks_agentbricks-0.2.0/src/databricks_agentbricks/agent_project.py +657 -0
  9. databricks_agentbricks-0.2.0/src/databricks_agentbricks/app_resources.py +139 -0
  10. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/__init__.py +12 -0
  11. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/app.py +108 -0
  12. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/app_auth.py +198 -0
  13. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/auth.py +142 -0
  14. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/deploy.py +954 -0
  15. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/dev.py +310 -0
  16. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/endpoint.py +145 -0
  17. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/endpoint_examples.py +33 -0
  18. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/endpoint_output.py +53 -0
  19. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/endpoint_request.py +71 -0
  20. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/endpoint_transport.py +148 -0
  21. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/help.py +467 -0
  22. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/init.py +496 -0
  23. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/mcp.py +75 -0
  24. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/memory.py +520 -0
  25. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/sandbox.py +823 -0
  26. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/sessions.py +588 -0
  27. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/tools.py +456 -0
  28. databricks_agentbricks-0.2.0/src/databricks_agentbricks/cli/tracing.py +724 -0
  29. databricks_agentbricks-0.2.0/src/databricks_agentbricks/databricks_cli.py +38 -0
  30. databricks_agentbricks-0.2.0/src/databricks_agentbricks/errors.py +114 -0
  31. databricks_agentbricks-0.2.0/src/databricks_agentbricks/lakebase_runtime_store.py +107 -0
  32. databricks_agentbricks-0.2.0/src/databricks_agentbricks/legacy_lakebase_runtime_store.py +74 -0
  33. databricks_agentbricks-0.2.0/src/databricks_agentbricks/project_config.py +157 -0
  34. databricks_agentbricks-0.2.0/src/databricks_agentbricks/project_types.py +55 -0
  35. databricks_agentbricks-0.2.0/src/databricks_agentbricks/py.typed +0 -0
  36. databricks_agentbricks-0.2.0/src/databricks_agentbricks/render.py +349 -0
  37. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-bricks-migrate/SKILL.md +101 -0
  38. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/.env.example +36 -0
  39. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/.gitignore +199 -0
  40. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/AGENTKIT_CONTRACT.md +95 -0
  41. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/AGENTS.md +31 -0
  42. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/CLAUDE.md +1 -0
  43. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/README.md +141 -0
  44. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/__init__.py +0 -0
  45. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/agent.py +177 -0
  46. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/mcps.py +27 -0
  47. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/__init__.py +24 -0
  48. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/sample_tool.py +15 -0
  49. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/send_message.py +17 -0
  50. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/app.yaml +15 -0
  51. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/pyproject.toml +42 -0
  52. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/runtime/__init__.py +0 -0
  53. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/runtime/adapter.py +144 -0
  54. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/runtime/main.py +25 -0
  55. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/tests/__init__.py +0 -0
  56. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/tests/test_agent.py +211 -0
  57. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/tests/test_runtime.py +19 -0
  58. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-langgraph/tests/test_workspace.py +28 -0
  59. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/.env.example +37 -0
  60. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/.gitignore +199 -0
  61. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/AGENTKIT_CONTRACT.md +107 -0
  62. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/AGENTS.md +76 -0
  63. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/CLAUDE.md +1 -0
  64. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/README.md +143 -0
  65. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/__init__.py +0 -0
  66. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/agent.py +205 -0
  67. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/mcps.py +28 -0
  68. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/tools/__init__.py +24 -0
  69. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/tools/sample_tool.py +15 -0
  70. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/agent/tools/send_message.py +18 -0
  71. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/app.yaml +15 -0
  72. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/pyproject.toml +42 -0
  73. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/runtime/__init__.py +0 -0
  74. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/runtime/adapter.py +188 -0
  75. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/runtime/main.py +25 -0
  76. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/tests/__init__.py +0 -0
  77. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/tests/test_agent.py +326 -0
  78. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/tests/test_runtime.py +19 -0
  79. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/agent-openai/tests/test_workspace.py +28 -0
  80. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/.env.example +1 -0
  81. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/.gitignore +11 -0
  82. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/README.md +22 -0
  83. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/agent/__init__.py +0 -0
  84. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/agent/agent.py +27 -0
  85. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/app.yaml +1 -0
  86. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/pyproject.toml +38 -0
  87. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/runtime/__init__.py +0 -0
  88. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/runtime/main.py +31 -0
  89. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-langgraph/tests/test_runtime.py +29 -0
  90. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/.env.example +1 -0
  91. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/.gitignore +11 -0
  92. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/README.md +22 -0
  93. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/agent/__init__.py +0 -0
  94. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/agent/agent.py +29 -0
  95. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/app.yaml +1 -0
  96. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/pyproject.toml +38 -0
  97. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/runtime/__init__.py +0 -0
  98. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/runtime/main.py +31 -0
  99. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/custom-agent-openai/tests/test_runtime.py +29 -0
  100. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/sandbox_mcp.py +39 -0
  101. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/sandbox_mcp_langgraph.py +39 -0
  102. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/CHAT_APP.md +49 -0
  103. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/runtime/main.py +26 -0
  104. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/runtime/ui.py +528 -0
  105. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/tests/test_demo_ui.py +542 -0
  106. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/agentbricks-mark.png +0 -0
  107. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/app.js +1464 -0
  108. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/index.html +281 -0
  109. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/styles.css +1913 -0
  110. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/CHAT_APP.md +54 -0
  111. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/runtime/main.py +26 -0
  112. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/runtime/ui.py +521 -0
  113. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/tests/test_demo_ui.py +512 -0
  114. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/ui/agentbricks-mark.png +0 -0
  115. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/ui/app.js +1428 -0
  116. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/ui/index.html +281 -0
  117. databricks_agentbricks-0.2.0/src/databricks_agentbricks/templates/ui/agent-openai/ui/styles.css +1913 -0
  118. databricks_agentbricks-0.2.0/src/databricks_agentbricks/theme.py +72 -0
  119. databricks_agentbricks-0.2.0/src/databricks_agentkit/__init__.py +67 -0
  120. databricks_agentbricks-0.2.0/src/databricks_agentkit/_api_client.py +673 -0
  121. databricks_agentbricks-0.2.0/src/databricks_agentkit/_pagination.py +25 -0
  122. databricks_agentbricks-0.2.0/src/databricks_agentkit/client.py +31 -0
  123. databricks_agentbricks-0.2.0/src/databricks_agentkit/langgraph/__init__.py +98 -0
  124. databricks_agentbricks-0.2.0/src/databricks_agentkit/langgraph/genie.py +50 -0
  125. databricks_agentbricks-0.2.0/src/databricks_agentkit/langgraph/mcp.py +216 -0
  126. databricks_agentbricks-0.2.0/src/databricks_agentkit/langgraph/memory.py +67 -0
  127. databricks_agentbricks-0.2.0/src/databricks_agentkit/langgraph/session_store.py +485 -0
  128. databricks_agentbricks-0.2.0/src/databricks_agentkit/memory_store.py +371 -0
  129. databricks_agentbricks-0.2.0/src/databricks_agentkit/models.py +330 -0
  130. databricks_agentbricks-0.2.0/src/databricks_agentkit/openai/__init__.py +93 -0
  131. databricks_agentbricks-0.2.0/src/databricks_agentkit/openai/genie.py +50 -0
  132. databricks_agentbricks-0.2.0/src/databricks_agentkit/openai/mcp.py +189 -0
  133. databricks_agentbricks-0.2.0/src/databricks_agentkit/openai/memory.py +67 -0
  134. databricks_agentbricks-0.2.0/src/databricks_agentkit/openai/sessions.py +157 -0
  135. databricks_agentbricks-0.2.0/src/databricks_agentkit/py.typed +0 -0
  136. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/ARCHITECTURE.md +64 -0
  137. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/README.md +184 -0
  138. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/__init__.py +69 -0
  139. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/app.py +316 -0
  140. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/auth.py +149 -0
  141. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/__init__.py +1 -0
  142. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/execution.py +133 -0
  143. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/heartbeat.py +64 -0
  144. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/lakebase_runtime_store.py +648 -0
  145. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/recovery.py +70 -0
  146. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/durability/store.py +33 -0
  147. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/execution.py +160 -0
  148. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/genie.py +224 -0
  149. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/mcp_auth.py +114 -0
  150. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/model_services.py +90 -0
  151. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/runtime.py +273 -0
  152. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/session_store_client.py +120 -0
  153. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/store.py +287 -0
  154. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/tool_manifest.py +223 -0
  155. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/tracing.py +71 -0
  156. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/types.py +118 -0
  157. databricks_agentbricks-0.2.0/src/databricks_agentkit/runtime/workspace.py +27 -0
  158. databricks_agentbricks-0.2.0/src/databricks_agentkit/session_store.py +434 -0
  159. databricks_agentbricks-0.2.0/src/databricks_agentkit/timefmt.py +85 -0
databricks_agentbricks-0.2.0/.gitignore ADDED
@@ -0,0 +1,22 @@
1
+ # test
2
+ .pytest_cache/
3
+ mlruns/
4
+
5
+ # Byte-compiled files
6
+ __pycache__
7
+ *.pyc
8
+
9
+ # Editor files
10
+ *.vscode
11
+ .idea
12
+
13
+ # Mac
14
+ .DS_Store
15
+
16
+ # packaging
17
+ dist/
18
+ *.egg-info/
19
+ Pipfile
20
+ docs/build
21
+
22
+ **/uv.lock
databricks_agentbricks-0.2.0/NOTICE ADDED
@@ -0,0 +1,39 @@
1
+ Copyright (c) 2026 Databricks, Inc. All rights reserved.
2
+
3
+ This Software is the property of Databricks, Inc. ("Databricks") and no license, right, or authorization to install, use, copy, modify, or distribute this Software is granted except pursuant to Databricks' prior written authorization. Any use of this Software without such written authorization from Databricks is strictly prohibited.
4
+
5
+ # MIT License
6
+
7
+ This Software contains code from the following open source projects, licensed under the MIT license (https://opensource.org/licenses/MIT).
8
+
9
+ Package: PyYAML
10
+ Project URL: https://github.com/yaml/pyyaml
11
+ Copyright: 2017-2021 Ingy döt Net, 2006-2016 Kirill Simonov
12
+
13
+ Package: rich
14
+ Project URL: https://github.com/Textualize/rich
15
+ Copyright: 2020 Will McGugan
16
+
17
+ Package: tomli
18
+ Project URL: https://github.com/hukkin/tomli
19
+ Copyright: 2021 Taneli Hukkinen
20
+
21
+ Package: tomlkit
22
+ Project URL: https://github.com/python-poetry/tomlkit
23
+ Copyright: 2018 Sébastien Eustace
24
+
25
+ # Apache License 2.0
26
+
27
+ This Software contains code from the following open source projects, licensed under the Apache License 2.0 license (https://www.apache.org/licenses/LICENSE-2.0).
28
+
29
+ Package: databricks-sdk
30
+ Project URL: https://github.com/databricks/databricks-sdk-py
31
+ Copyright: 2023 Databricks, Inc.
32
+
33
+ # BSD 3-Clause License
34
+
35
+ This Software contains code from the following open source projects, licensed under the BSD 3-Clause License (https://opensource.org/license/bsd-3-clause).
36
+
37
+ Package: click
38
+ Project URL: https://github.com/pallets/click
39
+ Copyright: 2014 Pallets