feat(executor): Integrate VFS helper for in-memory artifact tracking
This commit is contained in:
36
backend/tests/test_executor_vfs.py
Normal file
36
backend/tests/test_executor_vfs.py
Normal file
@@ -0,0 +1,36 @@
|
||||
from ea_chatbot.graph.nodes.executor import executor_node
|
||||
from ea_chatbot.graph.state import AgentState
|
||||
|
||||
def test_executor_with_vfs():
|
||||
"""Verify that the executor node provides VFS access to the code."""
|
||||
state = AgentState(
|
||||
messages=[],
|
||||
question="test",
|
||||
analysis={},
|
||||
next_action="test",
|
||||
iterations=0,
|
||||
checklist=[],
|
||||
current_step=0,
|
||||
vfs={},
|
||||
plots=[],
|
||||
dfs={}
|
||||
)
|
||||
|
||||
# Code that uses the 'vfs' helper
|
||||
code = """
|
||||
vfs.write("output.txt", "Execution Result", metadata={"type": "text"})
|
||||
print("VFS Write Complete")
|
||||
"""
|
||||
state["code"] = code
|
||||
|
||||
result = executor_node(state)
|
||||
|
||||
# Check if the execution was successful
|
||||
assert result["error"] is None
|
||||
assert "VFS Write Complete" in result["code_output"]
|
||||
|
||||
# Verify that the VFS state was updated
|
||||
# Note: executor_node returns a dict of updates, which should include the updated 'vfs'
|
||||
assert "vfs" in result
|
||||
assert "output.txt" in result["vfs"]
|
||||
assert result["vfs"]["output.txt"]["content"] == "Execution Result"
|
||||
Reference in New Issue
Block a user