chore: clean up legacy tool/UI code paths and remove unused events
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import TYPE_CHECKING, Any, cast
|
||||
from typing import Any, cast
|
||||
|
||||
from ag_ui.core import (
|
||||
BaseEvent,
|
||||
@@ -14,9 +14,6 @@ from ag_ui.core import (
|
||||
from core.agentscope.runtime.ui_compiler import compile as compile_ui_hints
|
||||
from schemas.agent.ui_hints import UiHintsPayload
|
||||
|
||||
if TYPE_CHECKING:
|
||||
pass
|
||||
|
||||
_INTERNAL_TO_AGUI: dict[str, EventType] = {
|
||||
"run.started": EventType.RUN_STARTED,
|
||||
"run.finished": EventType.RUN_FINISHED,
|
||||
|
||||
@@ -165,10 +165,6 @@ class SqlAlchemyEventStore:
|
||||
if not isinstance(role_value, str):
|
||||
role_value = "assistant"
|
||||
role = self._resolve_role(role_value)
|
||||
tool_name = self._event_value(event, "tool_name")
|
||||
tool_name_value = (
|
||||
tool_name if isinstance(tool_name, str) and tool_name else None
|
||||
)
|
||||
|
||||
locked_session = await session_repo.lock_session_for_update(
|
||||
session_id=session_id
|
||||
@@ -185,7 +181,6 @@ class SqlAlchemyEventStore:
|
||||
role=role,
|
||||
content=content,
|
||||
model_code=model_code if isinstance(model_code, str) else None,
|
||||
tool_name=tool_name_value,
|
||||
metadata=metadata_model.model_dump(mode="json", exclude_none=True),
|
||||
input_tokens=input_tokens,
|
||||
output_tokens=output_tokens,
|
||||
|
||||
@@ -37,7 +37,6 @@ class PipelineStageEmitter:
|
||||
self._emitted_tool_calls: set[str] = set()
|
||||
self._emitted_tool_results: set[str] = set()
|
||||
self.latest_text_message_id: str | None = None
|
||||
self.latest_text: str = ""
|
||||
|
||||
async def handle_print(self, *, msg: Msg, last: bool) -> None:
|
||||
del last
|
||||
@@ -72,7 +71,6 @@ class PipelineStageEmitter:
|
||||
if not text:
|
||||
return
|
||||
self.latest_text_message_id = str(msg.id)
|
||||
self.latest_text = text
|
||||
|
||||
async def _emit_tool_events_from_msg(self, msg: Msg) -> None:
|
||||
for block in msg.get_content_blocks("tool_use"):
|
||||
|
||||
@@ -71,14 +71,3 @@ def build_toolkit(
|
||||
register_tool_middlewares(toolkit=toolkit)
|
||||
|
||||
return toolkit
|
||||
|
||||
|
||||
def build_stage_toolkit(
|
||||
*,
|
||||
enabled_skill_names: set[str] | None = None,
|
||||
enable_hitl: bool | None = None,
|
||||
) -> Any:
|
||||
return build_toolkit(
|
||||
enabled_skill_names=enabled_skill_names,
|
||||
enable_hitl=enable_hitl,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user