open-data-sci 0.2.0__tar.gz → 0.2.1__tar.gz

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: open-data-sci
-Version: 0.2.0
+Version: 0.2.1
 Summary: AI agent for data science and machine learning
 Project-URL: Homepage, https://github.com/f4roukb/open-data-sci
 Project-URL: Repository, https://github.com/f4roukb/open-data-sci
@@ -219,69 +219,70 @@ Requires-Python: <3.13,>=3.12
 Requires-Dist: catboost<2.0.0,>=1.2.0
 Requires-Dist: category-encoders<3.0.0,>=2.6.0
 Requires-Dist: duckdb<2.0.0,>=1.0.0
-Requires-Dist: duckduckgo-search<7.0.0,>=6.0.0
+Requires-Dist: duckduckgo-search<9.0.0,>=8.0.0
 Requires-Dist: httpx<1.0.0,>=0.27.0
 Requires-Dist: imbalanced-learn<1.0.0,>=0.12.0
-Requires-Dist: langchain-anthropic<1.0.0,>=0.2.0
-Requires-Dist: langchain-core<1.0.0,>=0.3.0
-Requires-Dist: langchain-openai<1.0.0,>=0.2.0
-Requires-Dist: langgraph<1.0.0,>=0.2.0
+Requires-Dist: langchain-anthropic<2.0.0,>=1.0.0
+Requires-Dist: langchain-core<2.0.0,>=1.0.0
+Requires-Dist: langchain-openai<2.0.0,>=1.0.0
+Requires-Dist: langgraph<2.0.0,>=1.0.0
 Requires-Dist: lightgbm<5.0.0,>=4.0.0
 Requires-Dist: lxml<7.0.0,>=6.1.0
 Requires-Dist: matplotlib<4.0.0,>=3.7.0
 Requires-Dist: networkx<4.0.0,>=3.0.0
-Requires-Dist: numpy<2.0.0,>=1.26.0
-Requires-Dist: optuna<4.0.0,>=3.6.0
-Requires-Dist: pandas[excel,feather,parquet]<3.0.0,>=2.0.0
-Requires-Dist: plotly<6.0.0,>=5.15.0
+Requires-Dist: numpy<2.5.0,>=2.0.0
+Requires-Dist: optuna<5.0.0,>=4.0.0
+Requires-Dist: pandas[excel,feather,parquet]<4.0.0,>=3.0.0
+Requires-Dist: plotly<7.0.0,>=6.0.0
 Requires-Dist: polars[calamine]<2.0.0,>=1.0.0
 Requires-Dist: prophet<2.0.0,>=1.1.0
 Requires-Dist: pydantic-settings<3.0.0,>=2.0.0
 Requires-Dist: pydantic<3.0.0,>=2.0.0
-Requires-Dist: pyod<3.0.0,>=2.0.0
+Requires-Dist: pyod<4.0.0,>=3.0.0
 Requires-Dist: python-dotenv<2.0.0,>=1.0.0
 Requires-Dist: pyyaml<7.0,>=6.0
-Requires-Dist: rich<14.0.0,>=13.0.0
+Requires-Dist: rich<16.0.0,>=15.0.0
 Requires-Dist: sandbox-runtime<1.0.0,>=0.2.0
 Requires-Dist: scikit-learn<2.0.0,>=1.5.0
 Requires-Dist: scipy<2.0.0,>=1.14.0
 Requires-Dist: seaborn<1.0.0,>=0.12.0
 Requires-Dist: shap<1.0.0,>=0.46.0
 Requires-Dist: statsmodels<1.0.0,>=0.14.0
-Requires-Dist: textual<0.48.0,>=0.47.0
+Requires-Dist: textual<9.0.0,>=8.0.0
 Requires-Dist: umap-learn<1.0.0,>=0.5.0
-Requires-Dist: xgboost<3.0.0,>=2.0.0
+Requires-Dist: xgboost<4.0.0,>=3.0.0
 Requires-Dist: xxhash<4.0.0,>=3.5.0
 Provides-Extra: aws
 Requires-Dist: boto3<2.0.0,>=1.34.0; extra == 'aws'
-Requires-Dist: langchain-aws<1.0.0,>=0.2.0; extra == 'aws'
+Requires-Dist: langchain-aws<2.0.0,>=1.0.0; extra == 'aws'
 Provides-Extra: azure
 Requires-Dist: azure-identity<2.0.0,>=1.15.0; extra == 'azure'
 Provides-Extra: dev
-Requires-Dist: kaggle<2.0.0,>=1.6.0; extra == 'dev'
-Requires-Dist: mypy<2.0.0,>=1.13.0; extra == 'dev'
-Requires-Dist: pandas-stubs<3.0.0,>=2.0.0; extra == 'dev'
-Requires-Dist: pytest-asyncio<1.0.0,>=0.23.0; extra == 'dev'
+Requires-Dist: kaggle<3.0.0,>=2.0.0; extra == 'dev'
+Requires-Dist: mypy<3.0.0,>=2.0.0; extra == 'dev'
+Requires-Dist: pandas-stubs<4.0.0,>=3.0.0; extra == 'dev'
+Requires-Dist: pyright<2.0.0,>=1.1.390; extra == 'dev'
+Requires-Dist: pytest-asyncio<2.0.0,>=0.23.0; extra == 'dev'
 Requires-Dist: pytest-cov<5.0.0,>=4.0.0; extra == 'dev'
 Requires-Dist: pytest-xdist<4.0.0,>=3.0.0; extra == 'dev'
-Requires-Dist: pytest<8.0.0,>=7.0.0; extra == 'dev'
+Requires-Dist: pytest<10.0.0,>=9.0.0; extra == 'dev'
 Requires-Dist: ruff<1.0.0,>=0.8.0; extra == 'dev'
 Requires-Dist: types-openpyxl<4.0.0,>=3.1.0; extra == 'dev'
 Requires-Dist: vulture<3.0.0,>=2.0.0; extra == 'dev'
 Provides-Extra: docs
 Requires-Dist: mkdocs-material<10.0.0,>=9.5.0; extra == 'docs'
 Requires-Dist: mkdocs<2.0.0,>=1.6.0; extra == 'docs'
-Requires-Dist: mkdocstrings[python]<1.0.0,>=0.26.0; extra == 'docs'
+Requires-Dist: mkdocstrings[python]<2.0.0,>=1.0.0; extra == 'docs'
 Provides-Extra: gcp
-Requires-Dist: langchain-google-vertexai<3.0.0,>=2.0.0; extra == 'gcp'
+Requires-Dist: langchain-google-vertexai<4.0.0,>=3.0.0; extra == 'gcp'
 Provides-Extra: gemini
-Requires-Dist: langchain-google-genai<3.0.0,>=2.0.0; extra == 'gemini'
+Requires-Dist: langchain-google-genai<5.0.0,>=4.0.0; extra == 'gemini'
 Provides-Extra: jax
 Requires-Dist: flax<1.0.0,>=0.8.0; extra == 'jax'
 Requires-Dist: jax[cpu]<1.0.0,>=0.4.0; extra == 'jax'
 Requires-Dist: optax<1.0.0,>=0.2.0; extra == 'jax'
 Provides-Extra: ollama
-Requires-Dist: langchain-ollama<1.0.0,>=0.2.0; extra == 'ollama'
+Requires-Dist: langchain-ollama<2.0.0,>=1.0.0; extra == 'ollama'
 Description-Content-Type: text/markdown
 
 # OpenDataSci

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/opendatasci/_tui/adapter.py

@@ -11,11 +11,11 @@ from abc import ABC, abstractmethod
 
 class MessageHandle(ABC):
     @abstractmethod
-    def append(self, chunk: str) -> None: ...
+    async def append(self, chunk: str) -> None: ...
     @abstractmethod
-    def set_content(self, text: str) -> None: ...
+    async def set_content(self, text: str) -> None: ...
     @abstractmethod
-    def finish(self) -> None: ...
+    async def finish(self) -> None: ...
 
 
 class EphemeralHandle(ABC):

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/opendatasci/_tui/app.py

@@ -33,6 +33,8 @@ from .widgets import (
     TurnStatusBar,
 )
 
+logger = logging.getLogger(__name__)
+
 
 def _print_providers() -> None:
     table = Table(title=None, show_header=True, header_style="bold")
@@ -47,9 +49,7 @@ def _get_version() -> str:
     try:
         return importlib.metadata.version("open-data-sci")
     except importlib.metadata.PackageNotFoundError:
-        logging.getLogger(__name__).warning(
-            "open-data-sci package not found; falling back to hardcoded version '0.2.0'"
-        )
+        logger.warning("open-data-sci package not found; falling back to hardcoded version '0.2.0'")
         return "0.2.0"
 
 
@@ -206,8 +206,8 @@ class OpenDataSciApp(App[None]):
             self.exit()
 
     @on(CommandApprovalPrompt.Decision)
-    def on_approval_decision(self, event: CommandApprovalPrompt.Decision) -> None:
-        resume_input = self._controller.resolve_approval(event.approved)
+    async def on_approval_decision(self, event: CommandApprovalPrompt.Decision) -> None:
+        resume_input = await self._controller.resolve_approval(event.approved)
         self.query_one("#user-input", Input).focus()
         self._run_agent(resume_input)
 
@@ -284,7 +284,7 @@ class OpenDataSciApp(App[None]):
         self._controller.hide_completion()
         self._controller.clear_paste_attachment()
         if self._controller.awaiting_choice:
-            resume_input = self._controller.cancel_choice()
+            resume_input = await self._controller.cancel_choice()
             if resume_input is not None:
                 self._run_agent(resume_input)
         elif not had_completion and not had_paste and self._controller.agent_running:

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/opendatasci/_tui/controller.py

@@ -216,23 +216,23 @@ class CLIController:
                 f"Check the path and try again.{hint}"
             )
             msg = ui.add_message("agent", "")
-            msg.set_content(msg_text)
-            msg.finish()
+            await msg.set_content(msg_text)
+            await msg.finish()
         except PermissionError:
             self._boot_failed = True
             msg = ui.add_message("agent", "")
-            msg.set_content(f"❌ Permission denied: `{escape_markup(self._workspace_path)}`")
-            msg.finish()
+            await msg.set_content(f"❌ Permission denied: `{escape_markup(self._workspace_path)}`")
+            await msg.finish()
         except ValueError as exc:
             self._boot_failed = True
             msg = ui.add_message("agent", "")
-            msg.set_content(f"❌ Provider error: {exc}")
-            msg.finish()
+            await msg.set_content(f"❌ Provider error: {exc}")
+            await msg.finish()
         except Exception as exc:
             self._boot_failed = True
             msg = ui.add_message("agent", "")
-            msg.set_content(f"❌ Failed to load: {exc}")
-            msg.finish()
+            await msg.set_content(f"❌ Failed to load: {exc}")
+            await msg.finish()
 
     @staticmethod
     def _did_you_mean(workspace_path: str) -> str:
@@ -317,7 +317,7 @@ class CLIController:
                 self._exit_choice_mode()
                 should_quit = await self._handle_slash(raw)
                 return ("quit" if should_quit else ""), ""
-            answer = self._handle_user_choice(raw)
+            answer = await self._handle_user_choice(raw)
             if answer is not None:
                 return "run", answer
             return "", ""
@@ -339,7 +339,9 @@ class CLIController:
         clean_text, refs = _parse_file_refs(raw)
         valid_refs, missing_refs = _split_existing_file_refs(refs)
         for ref in missing_refs:
-            self._ui.add_message("agent", f"⚠️ File not found: {escape_markup(ref._path)}").finish()
+            await self._ui.add_message(
+                "agent", f"⚠️ File not found: {escape_markup(ref._path)}"
+            ).finish()
 
         if refs and not clean_text and not valid_refs and attachment is None:
             return "", ""
@@ -375,14 +377,14 @@ class CLIController:
         """
         if self._service is None:
             if self._boot_failed:
-                self._ui.add_message(
+                await self._ui.add_message(
                     "agent",
                     "❌ Startup failed, so queries can't run in this session. "
                     "Fix the problem shown above and restart the app "
                     "(type `/exit` to quit).",
                 ).finish()
             else:
-                self._ui.add_message(
+                await self._ui.add_message(
                     "agent", "⚠️ Still loading — please wait a moment and try again."
                 ).finish()
             return
@@ -413,14 +415,14 @@ class CLIController:
                 if not isinstance(event, BaseAgentStreamEvent):
                     logger.warning("astream() yielded unexpected type %r; skipping", type(event))
                     continue
-                self._dispatch_stream_event(event, presenter)
+                await self._dispatch_stream_event(event, presenter)
                 if isinstance(event, (ResponseEvent, ErrorEvent)):
                     break
         except Exception as exc:
-            presenter.handle_exception(exc)
+            await presenter.handle_exception(exc)
         finally:
             self._agent_running = False
-            presenter.cleanup()
+            await presenter.cleanup()
             if self._active_turn_status is not None:
                 self._active_turn_status.stop()
                 self._active_turn_status = None
@@ -428,18 +430,18 @@ class CLIController:
                 self._ui.set_input_placeholder("Ask a question about your data…")
             self._ui.add_divider()
 
-    def _dispatch_stream_event(
+    async def _dispatch_stream_event(
         self, event: BaseAgentStreamEvent, presenter: _TurnPresenter
     ) -> None:
         """Route a single stream event to the appropriate presenter handler."""
         if isinstance(event, ReasoningEvent):
             presenter.handle_reasoning(event)
         elif isinstance(event, TokenEvent):
-            presenter.handle_token(event)
+            await presenter.handle_token(event)
         elif isinstance(event, ToolCommunicationEvent):
             presenter.handle_tool_communication(event)
         elif isinstance(event, ToolCallEvent):
-            presenter.handle_tool_call(event)
+            await presenter.handle_tool_call(event)
         elif isinstance(event, WorkerDoneEvent):
             presenter.handle_worker_done(event)
         elif isinstance(event, SubagentEvent):
@@ -449,17 +451,17 @@ class CLIController:
         elif isinstance(event, UsageEvent):
             apply_usage_event(event, self._active_turn_status)
         elif isinstance(event, InputRequiredEvent):
-            self._show_choice_prompt(event.content, list(event.choices))
+            await self._show_choice_prompt(event.content, list(event.choices))
         elif isinstance(event, ApprovalRequiredEvent):
             self._show_approval_prompt(event)
         elif isinstance(event, ResponseEvent):
             presenter.handle_response(event)
         elif isinstance(event, ErrorEvent):
-            presenter.handle_error(event)
+            await presenter.handle_error(event)
 
     # ── Choice handling ───────────────────────────────────────────────────────
 
-    def _show_choice_prompt(self, question: str, choices: list[str]) -> None:
+    async def _show_choice_prompt(self, question: str, choices: list[str]) -> None:
         labels = string.ascii_uppercase[: len(choices)]
         other_label = (
             string.ascii_uppercase[len(choices)]
@@ -484,7 +486,7 @@ class CLIController:
         lines.append(
             f"  [dim {theme['text_secondary']}]Press Esc to cancel[/dim {theme['text_secondary']}]"
         )
-        self._ui.add_message("question", "\n".join(lines)).finish()
+        await self._ui.add_message("question", "\n".join(lines)).finish()
         self._pending_choices = list(choices)
         self._other_choice_label = other_label
         self._awaiting_custom_choice_input = False
@@ -512,7 +514,7 @@ class CLIController:
         self._ui.set_input_placeholder("Ask a question about your data…")
         self._ui.remove_input_class("awaiting-choice")
 
-    def cancel_choice(self) -> str | None:
+    async def cancel_choice(self) -> str | None:
         """Exit choice mode and return the resume input to send to the agent.
 
         Returns ``"cancel"`` when a choice was active (caller must pass this
@@ -521,10 +523,10 @@ class CLIController:
         if not self._awaiting_choice:
             return None
         self._exit_choice_mode()
-        self._ui.add_message("agent", "Choice cancelled.").finish()
+        await self._ui.add_message("agent", "Choice cancelled.").finish()
         return "cancel"
 
-    def _handle_user_choice(self, raw: str) -> str | None:
+    async def _handle_user_choice(self, raw: str) -> str | None:
         raw_stripped = raw.strip()
         upper = raw_stripped.upper()
         if (
@@ -534,7 +536,7 @@ class CLIController:
         ):
             self._awaiting_custom_choice_input = True
             self._other_choice_label = None
-            self._ui.add_message("agent", "Type your answer and press Enter.").finish()
+            await self._ui.add_message("agent", "Type your answer and press Enter.").finish()
             self._ui.set_input_placeholder("Type your answer and press Enter…")
             return None
 
@@ -550,7 +552,7 @@ class CLIController:
             else raw_stripped
         )
 
-        self._ui.add_message("user", escape_markup(raw)).finish()
+        await self._ui.add_message("user", escape_markup(raw)).finish()
         return answer
 
     # ── Approval handling ─────────────────────────────────────────────────────
@@ -564,7 +566,7 @@ class CLIController:
         self._awaiting_approval = True
         self._ui.set_input_placeholder("↑/↓ to select Yes or No, Enter to confirm, Esc to decline…")
 
-    def resolve_approval(self, approved: bool) -> str:
+    async def resolve_approval(self, approved: bool) -> str:
         """Record the user's approval decision and return the resume input.
 
         The caller must pass the returned value to ``run_agent`` so the paused
@@ -572,7 +574,7 @@ class CLIController:
         """
         self._awaiting_approval = False
         self._ui.set_input_placeholder("Ask a question about your data…")
-        self._ui.add_message("user", "Yes" if approved else "No").finish()
+        await self._ui.add_message("user", "Yes" if approved else "No").finish()
         return "yes" if approved else "no"
 
     # ── Slash command dispatch ────────────────────────────────────────────────
@@ -593,26 +595,26 @@ class CLIController:
         elif cmd == "/compact":
             await self.compact()
         elif cmd == "/ls-workspace":
-            self.ls_workspace()
+            await self.ls_workspace()
         elif cmd == "/models":
-            self.show_models()
+            await self.show_models()
         elif cmd == "/stop":
             await self.stop_agent()
         elif cmd == "/cancel-all-messages":
-            self.cancel_pending_messages()
+            await self.cancel_pending_messages()
         elif cmd == "/cancel-message":
-            self.cancel_last_pending_message()
+            await self.cancel_last_pending_message()
         elif cmd == "/help":
-            self.show_help()
+            await self.show_help()
         elif cmd == "/themes":
-            self.show_themes()
+            await self.show_themes()
         elif cmd == "/vars":
-            self._ui.add_message(
+            await self._ui.add_message(
                 "agent",
                 "⚠️ `/vars` has been removed. Use `/help` to see available commands.",
             ).finish()
         else:
-            self._ui.add_message(
+            await self._ui.add_message(
                 "agent",
                 f"⚠️ Unknown command: `{cmd}`\n\nType `/help` to see all available commands.",
             ).finish()
@@ -628,11 +630,11 @@ class CLIController:
         if self._service is not None:
             try:
                 await self._service.reset_session()
-                self._ui.add_message("agent", "✓ Session reset.").finish()
+                await self._ui.add_message("agent", "✓ Session reset.").finish()
             except Exception as exc:
-                self._ui.add_message("agent", f"❌ Reset failed: {exc}").finish()
+                await self._ui.add_message("agent", f"❌ Reset failed: {exc}").finish()
         else:
-            self._ui.add_message("agent", "Not loaded yet.").finish()
+            await self._ui.add_message("agent", "Not loaded yet.").finish()
 
     async def clear_conv(self) -> None:
         """Clear all conversation context (preserves session variables)."""
@@ -648,40 +650,40 @@ class CLIController:
                 await self._service.clear_context()
             except Exception:
                 logger.exception("Failed to clear service context")
-        self._ui.add_message("agent", "✓ Context cleared.").finish()
+        await self._ui.add_message("agent", "✓ Context cleared.").finish()
 
     async def compact(self) -> None:
         """Summarize the conversation and replace it with a compact context preamble."""
         if self._service is None:
-            self._ui.add_message("agent", "Not loaded yet.").finish()
+            await self._ui.add_message("agent", "Not loaded yet.").finish()
             return
         status = self._ui.add_message("agent", "Compacting conversation…")
-        status.set_content("Compacting conversation…")
+        await status.set_content("Compacting conversation…")
         compact_timer: TurnStatusHandle | None = self._ui.add_turn_status_bar()
         try:
             await self._service.compact_chat_history()
         except Exception as exc:
-            status.set_content(f"❌ Compact failed: {exc}")
+            await status.set_content(f"❌ Compact failed: {exc}")
             if compact_timer is not None:
                 compact_timer.stop()
-            status.finish()
+            await status.finish()
             return
         try:
             self._ui.clear_messages()
             compact_timer = None  # removed from DOM by clear_messages()
-            self._ui.add_message(
+            await self._ui.add_message(
                 "agent",
                 "**✓ Compaction done.** You may continue the conversation.",
             ).finish()
         finally:
-            status.finish()
+            await status.finish()
             if compact_timer is not None:
                 compact_timer.stop()
 
-    def show_models(self) -> None:
+    async def show_models(self) -> None:
         """Display the primary and secondary model in use."""
         cfg = self._cfg or self._base_config
-        self._ui.add_message(
+        await self._ui.add_message(
             "agent",
             format_models_message(
                 cfg.provider,
@@ -691,13 +693,13 @@ class CLIController:
             ),
         ).finish()
 
-    def show_help(self) -> None:
+    async def show_help(self) -> None:
         """Display all available slash commands with descriptions."""
-        self._ui.add_message("agent", format_help_message()).finish()
+        await self._ui.add_message("agent", format_help_message()).finish()
 
-    def show_themes(self) -> None:
+    async def show_themes(self) -> None:
         """Display the list of available colour themes and mark the active one."""
-        self._ui.add_message(
+        await self._ui.add_message(
             "agent",
             format_themes_message(_theme.active_name, _theme.THEME_DESCRIPTIONS),
         ).finish()
@@ -705,34 +707,34 @@ class CLIController:
     async def stop_agent(self) -> None:
         """Stop the currently running agent turn."""
         if not self._agent_running:
-            self._ui.add_message("agent", "No agent is currently running.").finish()
+            await self._ui.add_message("agent", "No agent is currently running.").finish()
             return
         self._ui.stop_agent()
         if self._service is not None:
             await self._service.rewind_turn()
-        self._ui.add_message("agent", "⏹ Agent stopped. You can continue from here.").finish()
+        await self._ui.add_message("agent", "⏹ Agent stopped. You can continue from here.").finish()
 
-    def cancel_pending_messages(self) -> None:
+    async def cancel_pending_messages(self) -> None:
         """Discard every message currently queued behind a running agent turn."""
         removed = self._pending_queue.cancel_all()
         for message in removed:
             self._discard_pending_handle(message.id)
         if removed:
             count = len(removed)
-            self._ui.add_message(
+            await self._ui.add_message(
                 "agent", f"✓ Cancelled {count} pending message{'s' if count != 1 else ''}."
             ).finish()
         else:
-            self._ui.add_message("agent", "No pending messages to cancel.").finish()
+            await self._ui.add_message("agent", "No pending messages to cancel.").finish()
 
-    def cancel_last_pending_message(self) -> None:
+    async def cancel_last_pending_message(self) -> None:
         """Discard only the most recently queued message."""
         message = self._pending_queue.cancel_last()
         if message is None:
-            self._ui.add_message("agent", "No pending messages to cancel.").finish()
+            await self._ui.add_message("agent", "No pending messages to cancel.").finish()
             return
         self._discard_pending_handle(message.id)
-        self._ui.add_message("agent", "✓ Cancelled last pending message.").finish()
+        await self._ui.add_message("agent", "✓ Cancelled last pending message.").finish()
 
     def _discard_pending_handle(self, message_id: int) -> None:
         handle = self._pending_handles.pop(message_id, None)
@@ -744,13 +746,13 @@ class CLIController:
         for message in self._pending_queue.cancel_all():
             self._discard_pending_handle(message.id)
 
-    def ls_workspace(self) -> None:
+    async def ls_workspace(self) -> None:
         if self._service is None:
-            self._ui.add_message("agent", "_Not loaded yet._").finish()
+            await self._ui.add_message("agent", "_Not loaded yet._").finish()
             return
         try:
             files = self._service.get_workspace_files()
         except Exception as exc:
-            self._ui.add_message("agent", f"❌ {exc}").finish()
+            await self._ui.add_message("agent", f"❌ {exc}").finish()
             return
         self._ui.show_workspace_panel(files)

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/opendatasci/_tui/presenter.py

@@ -53,8 +53,11 @@ def apply_usage_event(event: UsageEvent, turn_status: TurnStatusHandle | None) -
 class _TurnPresenter:
     """Manages ephemeral UI state (bubbles, tool blocks, thinking) for one turn.
 
-    All methods are synchronous; awaiting happens inside ``MessageBubble``
-    and ``ToolCallBlock`` via Textual's own async machinery.
+    Handlers that touch a ``MessageHandle`` (``handle_token``,
+    ``handle_tool_call``, ``handle_error``, ``handle_exception``,
+    ``cleanup``) are async, since ``MessageHandle.append/set_content/finish``
+    are async (they await Textual's ``MarkdownStream``). The rest stay
+    synchronous.
     """
 
     def __init__(self, ui: UIAdapter) -> None:
@@ -118,11 +121,11 @@ class _TurnPresenter:
             self._thinking_start = time.monotonic()
             self._had_reasoning = True
 
-    def handle_token(self, event: TokenEvent) -> None:
+    async def handle_token(self, event: TokenEvent) -> None:
         self._finish_thinking()
         if self._agent_msg is None:
             self._agent_msg = self._ui.add_message("agent", "")
-        self._agent_msg.append(event.content)
+        await self._agent_msg.append(event.content)
 
     def handle_tool_communication(self, event: ToolCommunicationEvent) -> None:
         tool_display = REGISTRY.get(event.tool_name) if event.tool_name else None
@@ -146,7 +149,7 @@ class _TurnPresenter:
             self._ephemerals.append(block)
             self._ephemerals_by_id[tc_id] = block
 
-    def handle_tool_call(self, event: ToolCallEvent) -> None:
+    async def handle_tool_call(self, event: ToolCallEvent) -> None:
         tool_call_id = event.tool_call_id or ""
         existing = self._pending_ephemerals.pop(tool_call_id, None) if tool_call_id else None
         tool_display = REGISTRY.get(str(event.tool))
@@ -173,7 +176,7 @@ class _TurnPresenter:
         self._finish_thinking()
         has_narration = self._agent_msg is not None
         if self._agent_msg is not None:
-            self._agent_msg.finish()
+            await self._agent_msg.finish()
             self._agent_msg = None
 
         buffered_comm = self._comm_buffers.pop(tool_call_id, "")
@@ -255,24 +258,24 @@ class _TurnPresenter:
         if self._agent_msg is None and event.content:
             self._agent_msg = self._ui.add_message("agent", event.content)
 
-    def handle_error(self, event: ErrorEvent) -> None:
+    async def handle_error(self, event: ErrorEvent) -> None:
         self._dismiss_thinking_block()
         if self._agent_msg is None:
             self._agent_msg = self._ui.add_message("agent", "")
-        self._agent_msg.append(f"\n\n❌ {event.content}")
+        await self._agent_msg.append(f"\n\n❌ {event.content}")
 
-    def handle_exception(self, exc: Exception) -> None:
+    async def handle_exception(self, exc: Exception) -> None:
         self._dismiss_thinking_block()
         if self._agent_msg is None:
             self._agent_msg = self._ui.add_message("agent", "")
-        self._agent_msg.set_content(f"❌ **Error:** {exc}")
+        await self._agent_msg.set_content(f"❌ **Error:** {exc}")
 
     # ── Cleanup ───────────────────────────────────────────────────────────────
 
-    def cleanup(self) -> None:
+    async def cleanup(self) -> None:
         """Finalise all open UI elements (called from the run_agent finally block)."""
         self._finish_thinking()
         for e in self._ephemerals:
             e.set_done()
         if self._agent_msg is not None:
-            self._agent_msg.finish()
+            await self._agent_msg.finish()

{open_data_sci-0.2.0 → open_data_sci-0.2.1}/opendatasci/_tui/tools_display.py

@@ -58,7 +58,7 @@ def register(tool_name: str, display: ToolDisplay) -> None:
 # from opendatasci.tools.factory (which creates a circular dependency).
 
 register("execute_python_code", ToolDisplay(label="Code", icon="🐍", summary_arg="summary"))
-register("execute_cli_command", ToolDisplay(label="Command", icon="⌨️", summary_arg="summary"))
+register("execute_cli_command", ToolDisplay(label="Command", icon="💻", summary_arg="summary"))
 register(
     "list_python_libs",
     ToolDisplay(label="Checking available libraries", icon="📦", summary_arg="summary"),
@@ -67,13 +67,13 @@ register(
     "enter_plan_mode",
     ToolDisplay(label="Planning the next steps", icon="🎯", summary_arg="summary"),
 )
-register("exit_plan_mode", ToolDisplay(label="Planning complete", icon="✅", summary_arg="summary"))
+register("exit_plan_mode", ToolDisplay(label="Planning complete", icon="🏁", summary_arg="summary"))
 register("load_skill", ToolDisplay(label="Loading skill", icon="🧠", summary_arg="summary"))
 register(
     "list_skills",
     ToolDisplay(label="Checking available skills", icon="🧭", summary_arg="summary"),
 )
-register("spawn_workers", ToolDisplay(label="Spawning workers", icon="⚙️", summary_arg="summary"))
+register("spawn_workers", ToolDisplay(label="Spawning workers", icon="👥", summary_arg="summary"))
 register(
     "read_dataset_info",
     ToolDisplay(
@@ -116,5 +116,5 @@ register(
 )
 register(
     "verify_python_code",
-    ToolDisplay(label="Reviewing code", icon="🔎", summary_arg="summary"),
+    ToolDisplay(label="Reviewing code", icon="🧪", summary_arg="summary"),
 )