129 lines
3.2 KiB
Python
129 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""ACP session metadata for TUI Coding Mode projects."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from acp import text_block
|
|
|
|
from pineagents.agents.acp.meta import (
|
|
ACP_CODING_PROJECT_META_KEY,
|
|
ACP_EPHEMERAL_META_KEY,
|
|
)
|
|
from pineagents.agents.acp.server import QwenPawACPAgent
|
|
|
|
|
|
class _FakeConn:
|
|
async def session_update(self, session_id, update): # noqa: ANN001
|
|
del session_id, update
|
|
return None
|
|
|
|
|
|
class _FakeWorkspace:
|
|
def __init__(self) -> None:
|
|
self.requests = []
|
|
|
|
async def stream_query(self, request): # noqa: ANN001
|
|
self.requests.append(request)
|
|
for event in ():
|
|
yield event
|
|
|
|
|
|
class _TestACPAgent(QwenPawACPAgent):
|
|
def __init__(self, workspace: _FakeWorkspace) -> None:
|
|
super().__init__(agent_id="default")
|
|
self._fake_workspace = workspace
|
|
|
|
async def _ensure_workspace(self):
|
|
return self._fake_workspace
|
|
|
|
|
|
async def test_acp_project_metadata_flows_to_request_context(tmp_path):
|
|
project_dir = str(tmp_path)
|
|
workspace = _FakeWorkspace()
|
|
agent = _TestACPAgent(workspace)
|
|
agent.on_connect(_FakeConn())
|
|
|
|
response = await agent.new_session(
|
|
cwd=project_dir,
|
|
**{ACP_CODING_PROJECT_META_KEY: project_dir},
|
|
)
|
|
|
|
await agent.prompt(
|
|
prompt=[text_block("hello")],
|
|
session_id=response.session_id,
|
|
)
|
|
|
|
assert workspace.requests
|
|
assert (
|
|
workspace.requests[0].request_context[ACP_CODING_PROJECT_META_KEY]
|
|
== project_dir
|
|
)
|
|
|
|
|
|
async def test_acp_project_metadata_is_stripped(tmp_path):
|
|
project_dir = str(tmp_path)
|
|
workspace = _FakeWorkspace()
|
|
agent = _TestACPAgent(workspace)
|
|
agent.on_connect(_FakeConn())
|
|
|
|
response = await agent.new_session(
|
|
cwd=project_dir,
|
|
**{ACP_CODING_PROJECT_META_KEY: f" {project_dir} "},
|
|
)
|
|
|
|
await agent.prompt(
|
|
prompt=[text_block("hello")],
|
|
session_id=response.session_id,
|
|
)
|
|
|
|
assert (
|
|
workspace.requests[0].request_context[ACP_CODING_PROJECT_META_KEY]
|
|
== project_dir
|
|
)
|
|
|
|
|
|
async def test_acp_resume_project_metadata_is_stripped(tmp_path):
|
|
project_dir = str(tmp_path)
|
|
workspace = _FakeWorkspace()
|
|
agent = _TestACPAgent(workspace)
|
|
agent.on_connect(_FakeConn())
|
|
|
|
response = await agent.new_session(cwd=project_dir)
|
|
await agent.resume_session(
|
|
cwd=project_dir,
|
|
session_id=response.session_id,
|
|
**{ACP_CODING_PROJECT_META_KEY: f" {project_dir} "},
|
|
)
|
|
|
|
await agent.prompt(
|
|
prompt=[text_block("hello")],
|
|
session_id=response.session_id,
|
|
)
|
|
|
|
assert (
|
|
workspace.requests[0].request_context[ACP_CODING_PROJECT_META_KEY]
|
|
== project_dir
|
|
)
|
|
|
|
|
|
async def test_acp_ephemeral_metadata_flows_to_request_context(tmp_path):
|
|
project_dir = str(tmp_path)
|
|
workspace = _FakeWorkspace()
|
|
agent = _TestACPAgent(workspace)
|
|
agent.on_connect(_FakeConn())
|
|
|
|
response = await agent.new_session(
|
|
cwd=project_dir,
|
|
**{ACP_EPHEMERAL_META_KEY: True},
|
|
)
|
|
|
|
await agent.prompt(
|
|
prompt=[text_block("warmup")],
|
|
session_id=response.session_id,
|
|
)
|
|
|
|
assert workspace.requests
|
|
assert (
|
|
workspace.requests[0].request_context[ACP_EPHEMERAL_META_KEY] is True
|
|
)
|