cube-chat-tool 0.1.0__tar.gz

cube_chat_tool-0.1.0/PKG-INFO

@@ -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/pyproject.toml

@@ -0,0 +1,39 @@
+[project]
+name = "cube-chat-tool"
+version = "0.1.0"
+description = "Chat tool implementation for cube-standard benchmarks"
+requires-python = ">=3.12"
+dependencies = [
+    "cube-standard",
+    "cube-chat",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0.0",
+]
+
+[build-system]
+requires = ["uv_build>=0.6.0,<0.7.0"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-name = "cube_chat_tool"
+
+[tool.ruff]
+fix = true
+line-length = 120
+indent-width = 4
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+skip-magic-trailing-comma = false
+line-ending = "auto"
+
+[tool.ruff.lint]
+extend-select = ["I"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+addopts = "-rs"

cube_chat_tool-0.1.0/src/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-0.1.0/src/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()