databricks-agentbricks 0.2.0__tar.gz → 0.3.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 (161) hide show
  1. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/PKG-INFO +115 -115
  2. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/README.md +114 -114
  3. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/pyproject.toml +2 -2
  4. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/_group.py +2 -2
  5. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/agent_project.py +6 -6
  6. databricks_agentbricks-0.3.0/src/databricks_agentbricks/app_resources.py +198 -0
  7. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/__init__.py +1 -1
  8. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/app.py +5 -5
  9. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/auth.py +5 -5
  10. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/deploy.py +77 -42
  11. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/dev.py +29 -29
  12. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/endpoint_examples.py +1 -1
  13. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/help.py +144 -97
  14. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/init.py +14 -12
  15. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/mcp.py +2 -2
  16. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/memory.py +12 -12
  17. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/sandbox.py +1 -1
  18. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/sessions.py +13 -13
  19. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/tools.py +14 -14
  20. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/tracing.py +270 -81
  21. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/errors.py +1 -1
  22. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/lakebase_runtime_store.py +1 -1
  23. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/project_config.py +3 -3
  24. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-bricks-migrate/SKILL.md +3 -3
  25. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/AGENTS.md +2 -2
  26. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/README.md +8 -8
  27. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/app.yaml +2 -2
  28. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/pyproject.toml +1 -1
  29. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/runtime/main.py +1 -1
  30. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/AGENTS.md +5 -5
  31. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/README.md +7 -7
  32. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/app.yaml +2 -2
  33. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/pyproject.toml +1 -1
  34. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/runtime/main.py +1 -1
  35. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/README.md +3 -3
  36. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/pyproject.toml +1 -1
  37. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/runtime/main.py +1 -1
  38. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/README.md +3 -3
  39. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/pyproject.toml +1 -1
  40. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/runtime/main.py +1 -1
  41. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/CHAT_APP.md +1 -1
  42. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/runtime/main.py +1 -1
  43. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/runtime/ui.py +2 -2
  44. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/tests/test_demo_ui.py +1 -1
  45. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/app.js +4 -4
  46. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/CHAT_APP.md +1 -1
  47. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/runtime/main.py +1 -1
  48. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/runtime/ui.py +2 -2
  49. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/tests/test_demo_ui.py +1 -1
  50. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/ui/app.js +4 -4
  51. databricks_agentbricks-0.3.0/src/databricks_agentbricks/trace_tables.py +36 -0
  52. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/_api_client.py +2 -2
  53. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/langgraph/memory.py +1 -1
  54. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/langgraph/session_store.py +1 -1
  55. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/openai/memory.py +1 -1
  56. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/openai/sessions.py +1 -1
  57. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/README.md +14 -14
  58. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/app.py +1 -1
  59. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/lakebase_runtime_store.py +2 -2
  60. databricks_agentbricks-0.2.0/src/databricks_agentbricks/app_resources.py +0 -139
  61. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/.gitignore +0 -0
  62. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/NOTICE +0 -0
  63. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/__init__.py +0 -0
  64. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/app_auth.py +0 -0
  65. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/endpoint.py +0 -0
  66. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/endpoint_output.py +0 -0
  67. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/endpoint_request.py +0 -0
  68. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/cli/endpoint_transport.py +0 -0
  69. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/databricks_cli.py +0 -0
  70. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/legacy_lakebase_runtime_store.py +0 -0
  71. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/project_types.py +0 -0
  72. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/py.typed +0 -0
  73. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/render.py +0 -0
  74. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/.env.example +0 -0
  75. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/.gitignore +0 -0
  76. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/AGENTKIT_CONTRACT.md +0 -0
  77. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/CLAUDE.md +0 -0
  78. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/__init__.py +0 -0
  79. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/agent.py +0 -0
  80. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/mcps.py +0 -0
  81. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/__init__.py +0 -0
  82. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/sample_tool.py +0 -0
  83. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/agent/tools/send_message.py +0 -0
  84. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/runtime/__init__.py +0 -0
  85. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/runtime/adapter.py +0 -0
  86. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/tests/__init__.py +0 -0
  87. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/tests/test_agent.py +0 -0
  88. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/tests/test_runtime.py +0 -0
  89. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-langgraph/tests/test_workspace.py +0 -0
  90. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/.env.example +0 -0
  91. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/.gitignore +0 -0
  92. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/AGENTKIT_CONTRACT.md +0 -0
  93. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/CLAUDE.md +0 -0
  94. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/__init__.py +0 -0
  95. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/agent.py +0 -0
  96. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/mcps.py +0 -0
  97. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/tools/__init__.py +0 -0
  98. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/tools/sample_tool.py +0 -0
  99. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/agent/tools/send_message.py +0 -0
  100. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/runtime/__init__.py +0 -0
  101. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/runtime/adapter.py +0 -0
  102. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/tests/__init__.py +0 -0
  103. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/tests/test_agent.py +0 -0
  104. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/tests/test_runtime.py +0 -0
  105. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/agent-openai/tests/test_workspace.py +0 -0
  106. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/.env.example +0 -0
  107. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/.gitignore +0 -0
  108. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/agent/__init__.py +0 -0
  109. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/agent/agent.py +0 -0
  110. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/app.yaml +0 -0
  111. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/runtime/__init__.py +0 -0
  112. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-langgraph/tests/test_runtime.py +0 -0
  113. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/.env.example +0 -0
  114. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/.gitignore +0 -0
  115. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/agent/__init__.py +0 -0
  116. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/agent/agent.py +0 -0
  117. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/app.yaml +0 -0
  118. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/runtime/__init__.py +0 -0
  119. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/custom-agent-openai/tests/test_runtime.py +0 -0
  120. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/sandbox_mcp.py +0 -0
  121. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/sandbox_mcp_langgraph.py +0 -0
  122. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/agentbricks-mark.png +0 -0
  123. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/index.html +0 -0
  124. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-langgraph/ui/styles.css +0 -0
  125. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/ui/agentbricks-mark.png +0 -0
  126. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/ui/index.html +0 -0
  127. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/templates/ui/agent-openai/ui/styles.css +0 -0
  128. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentbricks/theme.py +0 -0
  129. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/__init__.py +0 -0
  130. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/_pagination.py +0 -0
  131. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/client.py +0 -0
  132. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/langgraph/__init__.py +0 -0
  133. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/langgraph/genie.py +0 -0
  134. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/langgraph/mcp.py +0 -0
  135. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/memory_store.py +0 -0
  136. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/models.py +0 -0
  137. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/openai/__init__.py +0 -0
  138. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/openai/genie.py +0 -0
  139. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/openai/mcp.py +0 -0
  140. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/py.typed +0 -0
  141. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/ARCHITECTURE.md +0 -0
  142. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/__init__.py +0 -0
  143. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/auth.py +0 -0
  144. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/__init__.py +0 -0
  145. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/execution.py +0 -0
  146. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/heartbeat.py +0 -0
  147. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/recovery.py +0 -0
  148. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/durability/store.py +0 -0
  149. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/execution.py +0 -0
  150. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/genie.py +0 -0
  151. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/mcp_auth.py +0 -0
  152. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/model_services.py +0 -0
  153. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/runtime.py +0 -0
  154. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/session_store_client.py +0 -0
  155. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/store.py +0 -0
  156. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/tool_manifest.py +0 -0
  157. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/tracing.py +0 -0
  158. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/types.py +0 -0
  159. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/runtime/workspace.py +0 -0
  160. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/session_store.py +0 -0
  161. {databricks_agentbricks-0.2.0 → databricks_agentbricks-0.3.0}/src/databricks_agentkit/timefmt.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.5
2
2
  Name: databricks-agentbricks
3
- Version: 0.2.0
3
+ Version: 0.3.0
4
4
  Summary: Agent Bricks CLI and AgentKit SDK for custom agents on Databricks
5
5
  Author-email: Databricks <agent-feedback@databricks.com>
6
6
  License-File: NOTICE
@@ -33,7 +33,7 @@ Requires-Dist: openai-agents>=0.8.0; extra == 'openai'
33
33
  Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.25.0; extra == 'openai'
34
34
  Description-Content-Type: text/markdown
35
35
 
36
- # Agent Bricks CLI (`ab`)
36
+ # Agent Bricks CLI (`agentbricks`)
37
37
 
38
38
  Agent Bricks CLI is an experimental command-line interface for building and deploying custom
39
39
  agents on Databricks. It manages memory, sessions, tracing, and deployments from one authenticated
@@ -60,17 +60,17 @@ invocation protocol to design yourself. Bring your own agent, or start from a te
60
60
 
61
61
  ![Deployment: from a blank directory to a running service](docs/deployment.svg)
62
62
 
63
- - **Agent project** - `ab init` scaffolds a deployable project from a framework template
63
+ - **Agent project** - `agentbricks init` scaffolds a deployable project from a framework template
64
64
  (LangGraph or OpenAI Agents) with the runtime, tests, and an optional chat UI wired up; you edit
65
65
  the application code (model, tools, prompts).
66
66
  - **`agent.toml`** - the declarative source of truth for the Databricks-managed infrastructure your
67
67
  agent depends on: tool bindings (data sandbox, managed MCP services, Unity Catalog functions) and
68
- memory, session, and durability resources. `ab deploy` reads it to provision and wire everything
68
+ memory, session, and durability resources. `agentbricks deploy` reads it to provision and wire everything
69
69
  up (detailed under [Agent tools](#agent-tools)).
70
- - **`ab deploy`** - provisions the bound stores, grants the app's service principal access to
70
+ - **`agentbricks deploy`** - provisions the bound stores, grants the app's service principal access to
71
71
  them, provisions the durable-runtime database when durability is on, configures tracing, and rolls
72
- out the app. `ab deployments` covers the lifecycle (list, get, logs, start, stop, delete).
73
- - **`ab dev`** - runs your agent from the same manifest the deployment uses, so local behavior
72
+ out the app. `agentbricks deployments` covers the lifecycle (list, get, logs, start, stop, delete).
73
+ - **`agentbricks dev`** - runs your agent from the same manifest the deployment uses, so local behavior
74
74
  matches what ships.
75
75
 
76
76
  **Runtime**
@@ -83,28 +83,28 @@ The two ways to run an agent:
83
83
  invocation contract (synchronous, streaming, background). Enable the durable runtime so
84
84
  long-running and background work survives restarts, redeploys, and crashes. The framework
85
85
  templates are thin layers over `DurableAgentServer` (HTTP contract detailed under [Runtime](#runtime)).
86
- - **Custom server - generic, full control.** `ab init --server custom` scaffolds a minimal FastAPI
86
+ - **Custom server - generic, full control.** `agentbricks init --server custom` scaffolds a minimal FastAPI
87
87
  server with no `DurableAgentServer`: you define your own endpoints, request/response shapes, and protocol.
88
- `ab dev` and `ab deploy` run and ship it the same way.
88
+ `agentbricks dev` and `agentbricks deploy` run and ship it the same way.
89
89
 
90
90
  ## Prerequisites
91
91
 
92
- - **Python ≥3.10** — the `ab` CLI installs and runs on any Python 3.10+. The
93
- `memory`, `sessions`, `tools`, and `ab tracing bind`/`unbind` commands need nothing else.
92
+ - **Python ≥3.10** — the `agentbricks` CLI installs and runs on any Python 3.10+. The
93
+ `memory`, `sessions`, `tools`, and `agentbricks tracing bind`/`unbind` commands need nothing else.
94
94
  - **[`uv`](https://docs.astral.sh/uv/)** — needed to scaffold, run, and deploy an
95
- agent (`ab init` → `ab dev` → `ab deploy`): the scaffolded project builds
95
+ agent (`agentbricks init` → `agentbricks dev` → `agentbricks deploy`): the scaffolded project builds
96
96
  its environment and launches with `uv run`, both locally and in the deployed Apps
97
- runtime. The store/session/tools commands and `ab tracing bind`/`unbind` don't need it;
98
- `ab tracing list`/`get` do, to read `ab dev`'s local trace store.
97
+ runtime. The store/session/tools commands and `agentbricks tracing bind`/`unbind` don't need it;
98
+ `agentbricks tracing list`/`get` do, to read `agentbricks dev`'s local trace store.
99
99
  - **[Databricks CLI](https://docs.databricks.com/dev-tools/cli/)** — needed for
100
- browser-based `ab login`. If a profile is already authenticated, the CLI uses it
100
+ browser-based `agentbricks login`. If a profile is already authenticated, the CLI uses it
101
101
  directly and the Databricks CLI is optional.
102
102
 
103
103
  ## Installation
104
104
 
105
105
  From PyPI:
106
106
 
107
- The `databricks-agentbricks` Python distribution installs the `ab` command and AgentKit.
107
+ The `databricks-agentbricks` Python distribution installs the `agentbricks` command and AgentKit.
108
108
 
109
109
  ```sh
110
110
  pip install databricks-agentbricks
@@ -122,7 +122,7 @@ declare their framework dependencies automatically.
122
122
  ## Shell completion
123
123
  Add this to `~/.zshrc`:
124
124
  ```sh
125
- eval "$(_AB_COMPLETE=zsh_source ab)"
125
+ eval "$(_AGENTBRICKS_COMPLETE=zsh_source agentbricks)"
126
126
  ```
127
127
 
128
128
  ## Authentication
@@ -131,57 +131,57 @@ Agent Bricks CLI uses [Databricks authentication](https://docs.databricks.com/aw
131
131
  Ask the CLI to authenticate and remember a named profile:
132
132
 
133
133
  ```sh
134
- ab login --profile <profile>
135
- ab sessions stores list
134
+ agentbricks login --profile <profile>
135
+ agentbricks sessions stores list
136
136
  ```
137
137
 
138
- `ab login` validates existing credentials first. If credentials are missing or rejected in
138
+ `agentbricks login` validates existing credentials first. If credentials are missing or rejected in
139
139
  an interactive terminal, the CLI runs `databricks auth login --profile <profile>`, revalidates the
140
140
  profile, and stores the selection in the existing `~/.agentbricks/config.json` state file. This
141
141
  browser-based setup requires the Databricks CLI. In non-interactive environments, authenticate the
142
- profile before running `ab`. `ab logout` forgets the saved selection without revoking the underlying
142
+ profile before running `agentbricks`. `agentbricks logout` forgets the saved selection without revoking the underlying
143
143
  credentials.
144
144
 
145
- If Databricks SDK default authentication is already configured, you can skip `ab login`.
145
+ If Databricks SDK default authentication is already configured, you can skip `agentbricks login`.
146
146
  You can also pass the global `--profile/-p` option before an individual command, for example
147
- `ab --profile <profile> tools list`. Use `--output json` for scripting.
147
+ `agentbricks --profile <profile> tools list`. Use `--output json` for scripting.
148
148
 
149
149
  ## Quickstart
150
150
 
151
151
  The shortest path from a blank directory to a running and deployed agent:
152
152
 
153
153
  ```sh
154
- ab login --profile <profile>
155
- ab init my-agent
154
+ agentbricks login --profile <profile>
155
+ agentbricks init my-agent
156
156
  cd my-agent
157
- ab dev # run locally
158
- ab deploy my-agent # deploy to Databricks
157
+ agentbricks dev # run locally
158
+ agentbricks deploy my-agent # deploy to Databricks
159
159
  ```
160
160
 
161
- `ab dev` runs the agent locally on `http://localhost:8000`, wrapping the Databricks Apps
161
+ `agentbricks dev` runs the agent locally on `http://localhost:8000`, wrapping the Databricks Apps
162
162
  local runtime so local behavior matches a deployment.
163
163
 
164
- `ab deploy my-agent` deploys a Databricks App named `agent-bricks-my-agent`, provisions the
164
+ `agentbricks deploy my-agent` deploys a Databricks App named `agent-bricks-my-agent`, provisions the
165
165
  stores declared in `agent.toml`, and grants the app's service principal access to them. Use
166
- `ab deployments list` to find deployed apps, and `ab deployments get agent-bricks-my-agent` to
166
+ `agentbricks deployments list` to find deployed apps, and `agentbricks deployments get agent-bricks-my-agent` to
167
167
  print an app's URL and status.
168
168
 
169
- `ab init` declares default memory and session stores in `agent.toml`, so the deployed agent has
169
+ `agentbricks init` declares default memory and session stores in `agent.toml`, so the deployed agent has
170
170
  long-term memory and durable conversation history. It creates `<name>-<6-letter-token>-memory` and
171
171
  `<name>-<6-letter-token>-sessions`, and records both names in `agent.toml`.
172
- `ab deploy` creates them if they don't exist yet. Point the agent at stores you already have with
173
- `ab memory bind <name>` / `ab sessions bind <name>`, or scaffold without stores using
174
- `ab init --server custom` (see [Initialize the chat app demo](#initialize-the-chat-app-demo)).
175
-
176
- To exercise the agent (locally under `ab dev` or once deployed), `ab endpoint invoke` sends
177
- it an HTTP request. MLflow tracing is on by default (`ab init` binds a default
178
- `/Shared/agentbricks_traces/<project>` experiment): `ab dev` traces to a local MLflow server under
179
- `.agentbricks/` and `ab deploy` to the bound workspace experiment; `ab tracing list` shows the
172
+ `agentbricks deploy` creates them if they don't exist yet. Point the agent at stores you already have with
173
+ `agentbricks memory bind <name>` / `agentbricks sessions bind <name>`, or scaffold without stores using
174
+ `agentbricks init --server custom` (see [Initialize the chat app demo](#initialize-the-chat-app-demo)).
175
+
176
+ To exercise the agent (locally under `agentbricks dev` or once deployed), `agentbricks endpoint invoke` sends
177
+ it an HTTP request. MLflow tracing is on by default (`agentbricks init` binds a default
178
+ `/Shared/agentbricks_traces/<project>` experiment): `agentbricks dev` traces to a local MLflow server under
179
+ `.agentbricks/` and `agentbricks deploy` to the bound workspace experiment; `agentbricks tracing list` shows the
180
180
  available traces.
181
181
 
182
182
  ## Public names
183
183
 
184
- Use `ab` for the CLI and `AgentKitClient` from `databricks_agentkit` for the Python SDK. New
184
+ Use `agentbricks` for the CLI and `AgentKitClient` from `databricks_agentkit` for the Python SDK. New
185
185
  projects store local state under `.agentbricks/` and `~/.agentbricks/`, use
186
186
  `server = "agentbricks"` in `agent.toml`, and deploy apps with the `agent-bricks-` prefix.
187
187
 
@@ -248,11 +248,11 @@ and return a JSON result. You can also add your own FastAPI endpoints.
248
248
  Start from a template, edit the agent code in `agent/`, and run it locally before deploying:
249
249
 
250
250
  ```sh
251
- ab init my-agent --framework langgraph --server agentbricks --profile <profile>
251
+ agentbricks init my-agent --framework langgraph --server agentbricks --profile <profile>
252
252
  cd my-agent
253
- ab dev
253
+ agentbricks dev
254
254
  # Stop the local server when ready to deploy.
255
- ab --profile <profile> deploy my-agent
255
+ agentbricks --profile <profile> deploy my-agent
256
256
  ```
257
257
 
258
258
  Use `--framework openai` for OpenAI Agents. Templates keep agent code separate from the runtime
@@ -278,8 +278,8 @@ Each managed run is an **invocation**. Send a client-generated UUID `id` and you
278
278
  The UUID also acts as an idempotency key: repeating the same request reuses the existing invocation
279
279
  while its record is retained; using the ID for a different request returns `409`.
280
280
 
281
- `ab dev` keeps execution state in process and loses it on restart. For projects with
282
- `[agent].server = "agentbricks"`, `ab deploy` provisions a persistent Runtime Store for requests,
281
+ `agentbricks dev` keeps execution state in process and loses it on restart. For projects with
282
+ `[agent].server = "agentbricks"`, `agentbricks deploy` provisions a persistent Runtime Store for requests,
283
283
  status, events, and results. Register `@app.recover` to restart interrupted app-auth work after
284
284
  worker failures. Recovery is at-least-once, so external side effects must be idempotent. Session
285
285
  and Memory Stores separately preserve the state used by your agent.
@@ -289,7 +289,7 @@ workspace's shared Lakebase project and give the app SP ownership. The managed r
289
289
  tables; no manual Lakebase grant or Postgres app-resource attachment is needed. Backend selection
290
290
  is an internal rollout detail, not a user-facing setting; the current implementation retains the legacy
291
291
  per-app Lakebase project by default. Once enabled, redeploy reads the stored backend and verifies
292
- the app identity, and `ab deployments delete` removes the managed store before deleting the app.
292
+ the app identity, and `agentbricks deployments delete` removes the managed store before deleting the app.
293
293
  The switch does not migrate existing deployments between backends. Managed cleanup errors retain
294
294
  the app for retry. Direct app deletion bypasses managed store cleanup.
295
295
 
@@ -300,7 +300,7 @@ is never written to the Runtime Store. A replacement attempt after failure recov
300
300
 
301
301
  Use `server = "custom"` to deploy your own HTTP server without provisioning a Runtime Store.
302
302
  Changing the server type of an existing deployment is not supported. To use a different server,
303
- scaffold a new project with the desired `ab init --server` option and deploy it under a new name.
303
+ scaffold a new project with the desired `agentbricks init --server` option and deploy it under a new name.
304
304
  See the [runtime guide](src/databricks_agentkit/runtime/README.md) for agent hooks, full API examples,
305
305
  and recovery behavior.
306
306
 
@@ -319,7 +319,7 @@ fully managed store for each, both backed by Lakebase and usable from agents bui
319
319
  later, separate conversations, retrieved by semantic search.
320
320
 
321
321
  The examples below use the [`AgentKitClient` Python SDK](#agentkit-sdk); the same operations are available
322
- as `ab sessions` / `ab memory` CLI commands.
322
+ as `agentbricks sessions` / `agentbricks memory` CLI commands.
323
323
 
324
324
  ![Sessions and memory: the agent reads and appends one conversation's transcript in the session store, and recalls and saves durable facts in the memory store, which outlive any single conversation.](docs/sessions_and_memory.png)
325
325
 
@@ -415,8 +415,8 @@ directly.
415
415
  In an agent configured with `server = "agentbricks"`, add the memory tools so the model can read and write memory during a run.
416
416
  `memory_tools(actor)` exposes `remember` and `recall` bound to one actor's partition; it resolves the
417
417
  store from the `[memory_store]` binding, carried to the runtime by the `AGENT_MEMORY_STORE` env var
418
- that `ab deploy` injects, and returns no tools when no store is set, so the agent runs unchanged.
419
- That "no store set" path is also how it runs under `ab dev`, which runs locally: memory is off
418
+ that `agentbricks deploy` injects, and returns no tools when no store is set, so the agent runs unchanged.
419
+ That "no store set" path is also how it runs under `agentbricks dev`, which runs locally: memory is off
420
420
  there (the store is provisioned and used only at deploy). The OpenAI Agents adapter exposes the same as
421
421
  `memory_tools()`:
422
422
 
@@ -431,27 +431,27 @@ agent = create_agent(model=..., tools=[*your_tools, *memory_tools(actor)])
431
431
  > actor's entries. For strict isolation between tenants or users, use a separate store per boundary.
432
432
  > Grant another principal — such as your app's service principal — access with
433
433
  > `session_store.grant_permission(principal_id)` or `memory_store.grant_permission(principal_id)`;
434
- > `ab deploy` does this for the deployed app automatically.
434
+ > `agentbricks deploy` does this for the deployed app automatically.
435
435
 
436
436
  ### Declaring and provisioning stores
437
437
 
438
- For a deployed agent, `agent.toml` declares which stores it uses and `ab deploy` provisions them —
439
- you don't create stores by hand. `ab init` declares a default memory and session store named from
438
+ For a deployed agent, `agent.toml` declares which stores it uses and `agentbricks deploy` provisions them —
439
+ you don't create stores by hand. `agentbricks init` declares a default memory and session store named from
440
440
  the project; override those names, point at stores you already have, or let `deploy` create them:
441
441
 
442
442
  ```sh
443
443
  # Scaffold a project with default memory and session stores declared in agent.toml.
444
- ab init my-agent
444
+ agentbricks init my-agent
445
445
 
446
446
  # Override the declared store names at init time.
447
- ab init my-agent --memory-store support-agent-memory --session-store support-agent-sessions
447
+ agentbricks init my-agent --memory-store support-agent-memory --session-store support-agent-sessions
448
448
 
449
449
  # Or point an existing project at specific stores (edits agent.toml only; creates nothing).
450
- ab sessions bind support-agent-sessions
451
- ab memory bind support-agent-memory
450
+ agentbricks sessions bind support-agent-sessions
451
+ agentbricks memory bind support-agent-memory
452
452
 
453
453
  # deploy creates any declared-but-missing store and grants the app's service principal access.
454
- ab deploy my-agent
454
+ agentbricks deploy my-agent
455
455
  ```
456
456
 
457
457
  Memory and session stores are independent resources: deleting one never affects the other.
@@ -462,7 +462,7 @@ For the full command reference - every command, subcommand, argument, and option
462
462
  see [`cli.md`](cli.md). The tree below is a quick overview.
463
463
 
464
464
  ```text
465
- ab [-p <profile>] [-o text|json]
465
+ agentbricks [-p <profile>] [-o text|json]
466
466
  login [--profile P]
467
467
  logout
468
468
  init [--framework openai|langgraph] [--server agentbricks|custom]
@@ -504,8 +504,8 @@ ab [-p <profile>] [-o text|json]
504
504
  From the existing project, prepare a migration for your coding agent:
505
505
 
506
506
  ```sh
507
- ab init --framework langgraph --existing .
508
- ab init --framework openai --existing .
507
+ agentbricks init --framework langgraph --existing .
508
+ agentbricks init --framework openai --existing .
509
509
  ```
510
510
 
511
511
  This writes `agent-bricks-migrate/` containing a skill, a prompt to paste into your coding agent,
@@ -536,17 +536,17 @@ application. Migration supports LangGraph and the OpenAI Agents SDK with the man
536
536
 
537
537
  ## Invoke HTTP endpoints
538
538
 
539
- `ab endpoint invoke` is a low-level HTTP command. It resolves and authenticates a deployed
539
+ `agentbricks endpoint invoke` is a low-level HTTP command. It resolves and authenticates a deployed
540
540
  Databricks App, or targets localhost and arbitrary servers through `--url`. It does not assume an
541
541
  agent protocol: provide the method, path, query parameters, and complete JSON body required by the
542
542
  server.
543
543
 
544
544
  ```sh
545
- ab --profile <profile> endpoint invoke agent-bricks-my-agent \
545
+ agentbricks --profile <profile> endpoint invoke agent-bricks-my-agent \
546
546
  --path /api/invocations \
547
547
  --json '{"id":"00000000-0000-4000-8000-000000000001","input":[{"role":"user","content":"Hello"}]}'
548
548
 
549
- ab endpoint invoke --url http://localhost:8000 \
549
+ agentbricks endpoint invoke --url http://localhost:8000 \
550
550
  --path /api/invocations \
551
551
  --json '{"id":"00000000-0000-4000-8000-000000000001","input":[{"role":"user","content":"Hello"}]}'
552
552
  ```
@@ -557,11 +557,11 @@ a client-generated invocation ID, and streaming servers require their own stream
557
557
 
558
558
  ```sh
559
559
  INVOCATION_ID=$(uuidgen)
560
- ab --profile <profile> endpoint invoke agent-bricks-my-agent \
560
+ agentbricks --profile <profile> endpoint invoke agent-bricks-my-agent \
561
561
  --path /api/invocations \
562
562
  --json "{\"id\":\"$INVOCATION_ID\",\"input\":[{\"role\":\"user\",\"content\":\"Run the report\"}]}"
563
563
 
564
- ab --profile <profile> endpoint invoke agent-bricks-my-agent \
564
+ agentbricks --profile <profile> endpoint invoke agent-bricks-my-agent \
565
565
  --path /api/invocations \
566
566
  --sse \
567
567
  --json "{\"id\":\"$INVOCATION_ID\",\"input\":[{\"role\":\"user\",\"content\":\"Hello\"}],\"stream\":true}"
@@ -578,33 +578,33 @@ Use the conventional help flag at any command level. Every command's help includ
578
578
  examples:
579
579
 
580
580
  ```sh
581
- ab --help
582
- ab deploy --help
583
- ab sessions items append --help
581
+ agentbricks --help
582
+ agentbricks deploy --help
583
+ agentbricks sessions items append --help
584
584
  ```
585
585
 
586
586
  ## Agent tools
587
587
 
588
- For projects with `[agent].server = "agentbricks"` (the default from `ab init`), `agent.toml` is the
588
+ For projects with `[agent].server = "agentbricks"` (the default from `agentbricks init`), `agent.toml` is the
589
589
  declarative source of truth for Databricks-managed infrastructure: the Runtime Store, sandbox,
590
- managed MCP, Genie and Unity Catalog function bindings, plus memory and session resources. `ab tools
590
+ managed MCP, Genie and Unity Catalog function bindings, plus memory and session resources. `agentbricks tools
591
591
  add` updates only this file; direct TOML edits have the same behavior. Both managed-server framework
592
592
  adapters read the managed bindings at runtime without generating or patching agent source:
593
593
 
594
594
  ```sh
595
- ab tools add sandbox --scope table:samples.nyctaxi.trips
596
- ab tools add mcp system.ai.web_search
597
- ab tools add uc-function catalog.schema.lookup_ticket
598
- ab tools add genie-one
599
- ab tools add genie-agent SPACE_ID
600
- ab tools remove mcp system.ai.web_search
601
- ab tools list
595
+ agentbricks tools add sandbox --scope table:samples.nyctaxi.trips
596
+ agentbricks tools add mcp system.ai.web_search
597
+ agentbricks tools add uc-function catalog.schema.lookup_ticket
598
+ agentbricks tools add genie-one
599
+ agentbricks tools add genie-agent SPACE_ID
600
+ agentbricks tools remove mcp system.ai.web_search
601
+ agentbricks tools list
602
602
  ```
603
603
 
604
604
  ### Managed tool identity and migration
605
605
 
606
- `ab tools add mcp`, `ab tools add sandbox`, `ab tools add genie-one`, and
607
- `ab tools add genie-agent` write explicit `auth = "user"` by default.
606
+ `agentbricks tools add mcp`, `agentbricks tools add sandbox`, `agentbricks tools add genie-one`, and
607
+ `agentbricks tools add genie-agent` write explicit `auth = "user"` by default.
608
608
  Use `--auth app` for the App service principal instead. This field is on the tool entry, not
609
609
  inside `source` or `policy`:
610
610
 
@@ -649,8 +649,8 @@ Deploy derives Apps user scopes from explicit `auth = "user"` bindings:
649
649
  For example, bind Genie tools in a current project with `server = "agentbricks"`:
650
650
 
651
651
  ```sh
652
- ab tools add mcp system.ai.genie_one_mcp --auth user
653
- ab tools add genie-agent SPACE_ID --auth user
652
+ agentbricks tools add mcp system.ai.genie_one_mcp --auth user
653
+ agentbricks tools add genie-agent SPACE_ID --auth user
654
654
  ```
655
655
 
656
656
  Mixed bindings request the union. App-auth and legacy bindings add no user scopes. These are
@@ -663,7 +663,7 @@ initial typed SDK create request before uploading source. An existing App that i
663
663
  scope needs one-time explicit permission:
664
664
 
665
665
  ```sh
666
- ab --profile my-workspace deploy my-agent --allow-user-scope-update
666
+ agentbricks --profile my-workspace deploy my-agent --allow-user-scope-update
667
667
  ```
668
668
 
669
669
  The `system.ai.dbsql` managed MCP additionally requests the Apps `sql` user scope. This is full SQL
@@ -700,32 +700,32 @@ for **all** App-auth tools, or deploy those tools separately. Models, custom MCP
700
700
  Memory/Session Stores, and tracing keep their existing credentials.
701
701
 
702
702
  For MCP services, the remove command accepts the same service name as the add command. You can also
703
- remove any binding by its `id` in `agent.toml`, for example `ab tools remove web_search`.
703
+ remove any binding by its `id` in `agent.toml`, for example `agentbricks tools remove web_search`.
704
704
  Every successful add (including an already-configured no-op) points you to the target project's
705
705
  `agent.toml` to review configured managed tools and MCP bindings. With `--source`, the message
706
706
  points to that project's file. JSON add output includes its path in `manifest`.
707
707
 
708
- `ab tools list` discovers **available integrations to add**, not configured bindings. By default
708
+ `agentbricks tools list` discovers **available integrations to add**, not configured bindings. By default
709
709
  it shows built-in add recipes and caller-visible MCP Services in `system.ai`. A recipe may still
710
710
  need your resources: sandbox scopes, a concrete UC function name, or a Genie Space ID. Genie One
711
711
  needs no additional argument. `system.ai.sandbox` is represented by its scoped recipe rather than
712
712
  a second unscoped add command. The list does not enumerate every workspace schema, individual
713
713
  operations inside MCP services, or custom Python tools.
714
714
 
715
- `ab tools add mcp` looks up the service in the selected workspace before writing `agent.toml`.
716
- Use `ab --profile <profile> tools add mcp <service>` to select a workspace. A missing service or
715
+ `agentbricks tools add mcp` looks up the service in the selected workspace before writing `agent.toml`.
716
+ Use `agentbricks --profile <profile> tools add mcp <service>` to select a workspace. A missing service or
717
717
  failed lookup (including authentication or permission errors) leaves the project unchanged. This
718
718
  checks service metadata access, not whether every tool can be executed at runtime. Removing local
719
719
  bindings does not require workspace access.
720
720
 
721
721
  ```sh
722
- ab tools list
723
- ab tools list --kind mcp
724
- ab tools list --kind mcp --schema main.tools
725
- ab tools list --kind sandbox
726
- ab tools list --kind genie-one
727
- ab tools list --kind genie-agent
728
- ab --output json tools list
722
+ agentbricks tools list
723
+ agentbricks tools list --kind mcp
724
+ agentbricks tools list --kind mcp --schema main.tools
725
+ agentbricks tools list --kind sandbox
726
+ agentbricks tools list --kind genie-one
727
+ agentbricks tools list --kind genie-agent
728
+ agentbricks --output json tools list
729
729
  ```
730
730
 
731
731
  No agent project is required for discovery. MCP discovery uses your Databricks profile; the
@@ -739,9 +739,9 @@ runtime execution permissions.
739
739
  Read `agent.toml` (its `[[tools]]` entries) to inspect configured bindings. Discovery JSON uses
740
740
  `schema_version: 2`, with `available_tools` (`name`, `kind`, `add_command`), `mcp_schema` (null for
741
741
  local-only recipes), `complete`, and `errors`. Replace old scripts that read configured-list JSON
742
- with TOML inspection. Replace `ab mcp list [--schema catalog.schema]` with
743
- `ab tools list --kind mcp [--schema catalog.schema]`; the former command is removed. Use
744
- `ab tools list --help` for the new discovery contract.
742
+ with TOML inspection. Replace `agentbricks mcp list [--schema catalog.schema]` with
743
+ `agentbricks tools list --kind mcp [--schema catalog.schema]`; the former command is removed. Use
744
+ `agentbricks tools list --help` for the new discovery contract.
745
745
 
746
746
  Read-only live discovery can be checked against the installed wheel without creating a project
747
747
  or deploying an agent:
@@ -762,11 +762,11 @@ code in `agent/mcps.py` and are joined with the managed bindings by `mcp_tools(.
762
762
  `mcp_servers(...)`.
763
763
 
764
764
  Projects created with `--server custom` do not auto-discover `agent/tools/` or load managed tool
765
- bindings from `agent.toml`, so `ab tools add` rejects those projects. Wire framework-native Python
765
+ bindings from `agent.toml`, so `agentbricks tools add` rejects those projects. Wire framework-native Python
766
766
  tools and MCP servers directly in `agent/agent.py` instead.
767
767
 
768
768
  If a manifest with `server = "agentbricks"` contains `source = { kind = "python", ... }`, remove that
769
- `[[tools]]` entry; the decorated tool in `agent/tools/` remains active. `ab dev` and `ab deploy`
769
+ `[[tools]]` entry; the decorated tool in `agent/tools/` remains active. `agentbricks dev` and `agentbricks deploy`
770
770
  do not generate or patch Python tool code, and do not alter the manifest's `[[tools]]` bindings.
771
771
 
772
772
  Sandbox scopes default to read-only access. Repeat `--scope` to allow more than one resource, use
@@ -781,12 +781,12 @@ Installing the current `databricks-agentbricks` distribution does not configure
781
781
  capabilities your agent needs:
782
782
 
783
783
  ```sh
784
- ab tools add genie-one --name genie_one --auth user
785
- ab tools add genie-agent SPACE_ID --name genie_agent --auth user
786
- ab tools list --kind genie-one
787
- ab tools list --kind genie-agent
788
- ab tools remove genie_one
789
- ab tools remove genie_agent
784
+ agentbricks tools add genie-one --name genie_one --auth user
785
+ agentbricks tools add genie-agent SPACE_ID --name genie_agent --auth user
786
+ agentbricks tools list --kind genie-one
787
+ agentbricks tools list --kind genie-agent
788
+ agentbricks tools remove genie_one
789
+ agentbricks tools remove genie_agent
790
790
  ```
791
791
 
792
792
  `--name` is optional and defaults to `genie_one` or `genie_agent`, respectively. `--auth` defaults
@@ -795,7 +795,7 @@ without `auth` preserve App/default identity. Both add commands and `remove` acc
795
795
  to select a project instead of the current directory. Discovery needs no project; read that
796
796
  project's `agent.toml` to inspect configured bindings. For scripted output, put the global
797
797
  `-o json` option before `tools`, as in
798
- `ab -o json tools add genie-one --source ./my-agent`. Adding a binding is offline: it updates
798
+ `agentbricks -o json tools add genie-one --source ./my-agent`. Adding a binding is offline: it updates
799
799
  `agent.toml` without contacting Genie or checking permissions. The corresponding sources are:
800
800
 
801
801
  ```toml
@@ -856,11 +856,11 @@ It is included by default for `--framework langgraph`; pass `--disable-chat-app`
856
856
  API-only backend instead.
857
857
 
858
858
  ```sh
859
- ab init --framework langgraph \
859
+ agentbricks init --framework langgraph \
860
860
  --profile <profile> \
861
861
  ./my-agent
862
862
  cd ./my-agent
863
- ab dev
863
+ agentbricks dev
864
864
  ```
865
865
 
866
866
  The chat app includes synchronous, SSE streaming, background polling, Session Store, Memory Store,
@@ -870,14 +870,14 @@ and HITL resume UI. The framework-specific overlay adds `ui/`, `runtime/ui.py`,
870
870
  For the full deployed demo, bind both managed stores, then deploy:
871
871
 
872
872
  ```sh
873
- ab sessions bind agent-bricks-demo-sessions
874
- ab memory bind agent-bricks-demo-memory
875
- ab --profile <profile> deploy agent-bricks-agent-demo --source .
873
+ agentbricks sessions bind agent-bricks-demo-sessions
874
+ agentbricks memory bind agent-bricks-demo-memory
875
+ agentbricks --profile <profile> deploy agent-bricks-agent-demo --source .
876
876
  ```
877
877
 
878
- (`bind` declares the store name in `agent.toml`; `ab deploy` creates any declared-but-missing
878
+ (`bind` declares the store name in `agent.toml`; `agentbricks deploy` creates any declared-but-missing
879
879
  store and grants the app's service principal access to it. The memory store id flows to the runtime
880
- via the `AGENT_MEMORY_STORE` env var that `deploy` injects; `ab dev` runs locally with memory off
880
+ via the `AGENT_MEMORY_STORE` env var that `deploy` injects; `agentbricks dev` runs locally with memory off
881
881
  and does not inject it. The id is not persisted in `agent.toml`.)
882
882
 
883
883
  The chat UI generates a stable application session UUID in browser local storage, places it inside
@@ -892,6 +892,6 @@ sync/streaming/background transport selector is manual.
892
892
 
893
893
  ## Contributing
894
894
 
895
- Developing Agent Bricks CLI (`ab`), AgentKit, the runtime, and templates - plus the local dev loop and how to
896
- test unreleased changes on `ab dev` and `ab deploy`, is covered in
895
+ Developing Agent Bricks CLI (`agentbricks`), AgentKit, the runtime, and templates - plus the local dev loop and how to
896
+ test unreleased changes on `agentbricks dev` and `agentbricks deploy`, is covered in
897
897
  [CONTRIBUTING.md](CONTRIBUTING.md).