Merge pull request #153 from askui/fix/continue-streaming-run

feat(chat): continuing to stream events of run

Author: adi-wan-askui

pdm.lock

@@ -30,6 +30,7 @@ dependencies = [
     "markitdown[xls,xlsx,docx]>=0.1.2",
     "asyncer==0.0.8",
     "bson>=0.5.10",
+    "aiofiles>=24.1.0",
 ]
 requires-python = ">=3.10"
 readme = "README.md"
@@ -45,6 +46,7 @@ path = "src/askui/__init__.py"
 
 
 
+
 [tool.pdm]
 distribution = true
 
@@ -73,7 +75,7 @@ typecheck = "mypy"
 "json:gen" = "datamodel-codegen --output-model-type pydantic_v2.BaseModel --input src/askui/tools/askui/askui_ui_controller_grpc/json_schema/ --input-file-type jsonschema --output src/askui/tools/askui/askui_ui_controller_grpc/generated/"
 
 [dependency-groups]
-test = [
+dev = [
     "pytest>=8.3.4",
     "ruff>=0.9.5",
     "pytest-mock>=3.14.0",
@@ -90,10 +92,9 @@ test = [
     "types-pynput>=1.8.1.20250318",
     "playwright>=1.41.0",
     "pytest-asyncio>=1.1.0",
-]
-dev = [
     "datamodel-code-generator>=0.31.2",
     "grpcio-tools>=1.73.1",
+    "types-aiofiles>=24.1.0.20250822",
 ]
 
 

pyproject.toml

@@ -9,9 +9,12 @@
 from askui.chat.api.mcp_clients.manager import McpClientManagerManager
 from askui.chat.api.messages.chat_history_manager import ChatHistoryManager
 from askui.chat.api.messages.dependencies import ChatHistoryManagerDep
+from askui.chat.api.runs.models import RunListQuery
 
 from .service import RunService
 
+RunListQueryDep = Depends(RunListQuery)
+
 
 def get_runs_service(
     workspace_dir: Path = WorkspaceDirDep,

src/askui/chat/api/runs/dependencies.py

@@ -0,0 +1,15 @@
+from askui.chat.api.runs.events.done_events import DoneEvent
+from askui.chat.api.runs.events.error_events import ErrorEvent
+from askui.chat.api.runs.events.event_base import EventBase
+from askui.chat.api.runs.events.events import Event
+from askui.chat.api.runs.events.message_events import MessageEvent
+from askui.chat.api.runs.events.run_events import RunEvent
+
+__all__ = [
+    "DoneEvent",
+    "ErrorEvent",
+    "EventBase",
+    "Event",
+    "MessageEvent",
+    "RunEvent",
+]