Skip to content

Conversation

@eosho
Copy link
Owner

@eosho eosho commented Dec 30, 2025

This pull request refactors and simplifies the configuration and documentation for the code execution and visualization features in the data agent project. The main change is the removal of the explicit code_interpreter configuration from YAML and schema files, making visualization always enabled and controlled solely by environment variables. Documentation is updated accordingly, and related code and schema are cleaned up. Additional improvements include a new script for generating diagrams and a new dependency for table formatting.

Configuration and Feature Simplification:

  • Removed the code_interpreter section from agent YAML configuration, making visualization always enabled and executor selection automatic based on the AZURE_SESSIONS_POOL_ENDPOINT environment variable. (docs/CONFIGURATION.md, docs/VISUALIZATION.md, src/data_agent/config.py, src/data_agent/config_loader.py, src/data_agent/config/amex.yaml, src/data_agent/config/schema/agent_config.schema.json) [1] [2] [3] [4] [5] [6] [7] [8] [9] [10] [11] [12] [13]

  • Introduced VisualizationSettings in config.py to encapsulate visualization-related environment configuration, replacing the previous CodeInterpreterConfig dataclass. (src/data_agent/config.py)

Documentation and Usability Improvements:

  • Updated documentation to clarify that visualization is always enabled, executor selection is automatic, and no YAML configuration is required. The Azure setup guide is now more concise and refers to the official tutorial. (docs/CONFIGURATION.md, docs/VISUALIZATION.md, README.md) [1] [2] [3] [4] [5] [6] [7] [8] [9]

  • Replaced the ASCII diagram in the visualization documentation with a Mermaid sequence diagram for improved clarity. (docs/VISUALIZATION.md)

Developer Experience and Tooling:

  • Added a script (scripts/generate_diagrams.py) to generate flow diagrams for documentation using LangGraph visualization. (scripts/generate_diagrams.py)

  • Added the tabulate library as a dependency for improved table formatting. (pyproject.toml)

Codebase Cleanup:

  • Refactored the executors module and removed outdated comments and references to the old configuration mechanism. (src/data_agent/executors/__init__.py)

These changes streamline the setup and usage of the visualization feature, reduce configuration complexity, and improve the clarity and maintainability of both the codebase and documentation.

@eosho eosho merged commit 6d0de36 into main Dec 30, 2025
1 check passed
@eosho eosho deleted the feature/cleanup branch December 30, 2025 18:28
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.

2 participants