2879a3b3d4
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
26 lines
942 B
Python
26 lines
942 B
Python
import ast
|
|
from pathlib import Path
|
|
|
|
|
|
def test_deep_agent_uses_deep_context_schema():
|
|
from yuxi.agents.buildin.deep_agent.graph import DeepAgent
|
|
from yuxi.agents.buildin.deep_agent.context import DeepContext
|
|
|
|
assert DeepAgent.context_schema is DeepContext
|
|
|
|
|
|
def test_deep_agent_does_not_prepend_context_system_prompt_before_runtime_middleware():
|
|
graph_path = (
|
|
Path(__file__).parents[3] / "package" / "yuxi" / "agents" / "buildin" / "deep_agent" / "graph.py"
|
|
)
|
|
module = ast.parse(graph_path.read_text(encoding="utf-8"))
|
|
create_agent = next(
|
|
node
|
|
for node in ast.walk(module)
|
|
if isinstance(node, ast.Call) and isinstance(node.func, ast.Name) and node.func.id == "create_agent"
|
|
)
|
|
system_prompt = next(keyword.value for keyword in create_agent.keywords if keyword.arg == "system_prompt")
|
|
|
|
assert isinstance(system_prompt, ast.Constant)
|
|
assert system_prompt.value == ""
|