cube-chat-tool 0.1.0__py3-none-any.whl

cube_chat_tool/__init__.py

@@ -0,0 +1,5 @@
+"""cube-chat-tool: chat tool implementation for cube-standard benchmarks."""
+
+from cube_chat_tool.chat_tool import ChatTool, ChatToolConfig
+
+__all__ = ["ChatTool", "ChatToolConfig"]

cube_chat_tool/chat_tool.py

@@ -0,0 +1,126 @@
+"""Chat tool for cube-standard benchmarks.
+
+Provides ``ChatToolConfig`` and ``ChatTool``, a concrete implementation of the
+``Tool`` base class that wraps a ``ChatSession`` and exposes ``send_message``
+as the single agent-facing action.
+"""
+
+import datetime
+
+from cube.resources.chat_session import ChatConfig, ChatRole, ChatSession
+from cube.tool import Tool, ToolConfig, tool_action
+from cube_chat import BasicChatConfig
+from pydantic import Field
+
+
+class ChatToolConfig(ToolConfig):
+    """Configuration for ``ChatTool``.
+
+    Parameters
+    ----------
+    chat : ChatConfig
+        Chat session configuration. Defaults to an in-memory BasicChatConfig.
+    """
+
+    chat: ChatConfig = Field(default_factory=BasicChatConfig)
+
+    def make(self, container=None) -> "ChatTool":
+        """Create a ChatTool from this configuration."""
+        session = self.chat.make()
+        return ChatTool(config=self, session=session)
+
+
+class ChatTool(Tool):
+    """Chat tool that exposes a side-channel message interface to the agent.
+
+    The agent uses ``send_message`` to communicate with the task. The task uses
+    ``add_message`` and ``wait_for_user_message`` to drive the conversation.
+    After every agent action, ``chat_obs()`` is appended to the observation so
+    the agent always sees the up-to-date message history.
+
+    Call ``reset()`` between episodes to get a fresh session. Call ``close()``
+    to release session resources.
+    """
+
+    def __init__(self, config: ChatToolConfig, session: ChatSession) -> None:
+        self.config = config
+        self._session = session
+
+    @property
+    def session(self) -> ChatSession:
+        """Expose the raw ChatSession for advanced use (escape hatch, SR-003)."""
+        return self._session
+
+    @property
+    def messages(self) -> list[dict]:
+        """Delegate to session.messages."""
+        return self._session.messages
+
+    # ------------------------------------------------------------------
+    # Tool lifecycle
+    # ------------------------------------------------------------------
+
+    def reset(self) -> None:
+        """Stop the current session and create a fresh one."""
+        self._session.stop()
+        self._session = self.config.chat.make()
+
+    def close(self) -> None:
+        """Release session resources."""
+        self._session.stop()
+
+    # ------------------------------------------------------------------
+    # Task-internal methods (not agent-facing actions)
+    # ------------------------------------------------------------------
+
+    def add_message(self, role: ChatRole, msg: str) -> None:
+        """Post a message to the chat from the task side.
+
+        Parameters
+        ----------
+        role : ChatRole
+            Role of the message sender.
+        msg : str
+            Message content.
+        """
+        self._session.add_message(role, msg)
+
+    def wait_for_user_message(self) -> str:
+        """Block until the agent sends a message and return it.
+
+        Returns
+        -------
+        str
+            The agent's message text.
+        """
+        return self._session.wait_for_user_message()
+
+    def chat_obs(self) -> str:
+        """Format the full message history as a string.
+
+        Returns
+        -------
+        str
+            Formatted chat history with one line per message.
+        """
+        lines = []
+        for entry in self._session.messages:
+            ts = datetime.datetime.fromtimestamp(entry["timestamp"]).strftime("%H:%M:%S")
+            lines.append(f"[{ts}] {entry['role']}: {entry['message']}")
+        return "\n".join(lines)
+
+    # ------------------------------------------------------------------
+    # Agent-facing actions
+    # ------------------------------------------------------------------
+
+    @tool_action
+    def send_message(self, text: str) -> str:
+        """Send a message to the task.
+
+        Parameters
+        ----------
+        text : str
+            Message content to send.
+        """
+        self._session.send_message(text)
+        return self.chat_obs()

cube_chat_tool-0.1.0.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.3
+Name: cube-chat-tool
+Version: 0.1.0
+Summary: Chat tool implementation for cube-standard benchmarks
+Requires-Dist: cube-standard
+Requires-Dist: cube-chat
+Requires-Dist: pytest>=8.0.0 ; extra == 'dev'
+Requires-Python: >=3.12
+Provides-Extra: dev

cube_chat_tool-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+cube_chat_tool/__init__.py,sha256=7d5e69f3a42d0fae0922f1b94fcfbff365a620c2e5ecfc8577124510cbb44263,182
+cube_chat_tool/chat_tool.py,sha256=187af1dc127306b42081e56c075816b4fde343e2a992e215b84dd80d67110f04,4025
+cube_chat_tool-0.1.0.dist-info/WHEEL,sha256=2e78c4e84e52e71c127f39f0e9b8377baf8d97cfd8882e97be34163cff82eac7,79
+cube_chat_tool-0.1.0.dist-info/METADATA,sha256=b60e7b2e045bc14930fd88e524a2d5a44dbc021602ae5d4fe29a84cadd138193,265
+cube_chat_tool-0.1.0.dist-info/RECORD,,

cube_chat_tool-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.6.17
+Root-Is-Purelib: true
+Tag: py3-none-any