Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions pycodeloop/cli/serve.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,12 @@ def _respond_error(self, request_id, code: int, message: str) -> None:
def _wire_callbacks(self) -> None:
agent = self.flow.agent

def on_request(message_count: int, tool_count: int) -> None:
self._notify(
Comment thread
FernandoCelmer marked this conversation as resolved.
"chat/request",
{"messageCount": message_count, "toolCount": tool_count},
)

def on_text_delta(delta: str) -> None:
self._notify("chat/textDelta", {"delta": delta})

Expand Down Expand Up @@ -135,6 +141,7 @@ def confirm(name: str, preview: str) -> bool | str:
finally:
self._confirm_waiters.pop(request_id, None)

agent.on_request = on_request
agent.on_text_delta = on_text_delta
agent.on_turn_end = on_turn_end
agent.on_tool_call = on_tool_call
Expand Down
Loading