Skip to content

Add serialized_space support to Genie tools for export/import/clone#270

Open
CheeYuTan wants to merge 1 commit intodatabricks-solutions:mainfrom
CheeYuTan:feat/genie-serialized-space
Open

Add serialized_space support to Genie tools for export/import/clone#270
CheeYuTan wants to merge 1 commit intodatabricks-solutions:mainfrom
CheeYuTan:feat/genie-serialized-space

Conversation

@CheeYuTan
Copy link
Contributor

Summary

  • Adds serialized_space support to Genie MCP tools, enabling export/import/clone of Genie spaces with their full configuration (settings, instructions, sample questions)
  • get_genie: new include_serialized_space parameter to export full space config
  • create_or_update_genie: new serialized_space parameter to import config during creation/update
  • New clone_genie MCP tool for one-step space cloning across environments
  • Core library (AgentBricksManager) updated to pass serialized_space through the /api/2.0/data-rooms REST API

Fixes #83

Test proof

Tested against live workspace (e2-demo-field-eng):

TEST 1 - Get without include_serialized_space:
  Status: 200
  Has serialized_space: False
  Keys: ['id', 'space_id', 'display_name', 'description', 'warehouse_id', ...]

TEST 2 - Get with include_serialized_space=true:
  Status: 200
  Has serialized_space: True
  serialized_space length: 11894 chars
  serialized_space preview: {"version": 2, "config": {"sample_questions": [...]}}

TEST 3 - Core library genie_get with include_serialized_space:
  Has serialized_space: True
  serialized_space length: 11894 chars
  PASS
Test Result
REST API include_serialized_space=true returns config PASS
REST API without param omits serialized_space PASS
Core library genie_get(include_serialized_space=True) works PASS
genie_create signature accepts serialized_space param PASS
genie_update signature accepts serialized_space param PASS
MCP get_genie passes through to core library PASS
MCP create_or_update_genie passes through to core library PASS
New clone_genie tool composes export + create PASS

- genie_get: add include_serialized_space param to export full config
- genie_create/genie_update: accept serialized_space for import
- New clone_genie MCP tool for one-step space cloning
- Core library (AgentBricksManager) updated to pass serialized_space
  through the /api/2.0/data-rooms REST API

Fixes databricks-solutions#83
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Update of Genie Create/Update/Import tools with serialized_space support

1 participant