toml update

client.py

@@ -4,12 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""
-Doc Sweeper Environment Client.
-
-This module provides the client for connecting to a Doc Sweeper Environment server
-via WebSocket for persistent sessions.
-"""
 
 from __future__ import annotations
 
@@ -22,30 +16,12 @@ from models import DocAction, DocObservation, DocState
 
 
 class DocSweeperEnv(EnvClient[DocAction, DocObservation, DocState]):
-    """
-    Client for Doc Sweeper Environment.
-
-    This client maintains a persistent WebSocket connection to the environment
-    server, enabling efficient multi-step interactions with lower latency.
-
-    Example:
-        >>> with DocSweeperEnv(base_url="http://localhost:8000") as client:
-        ...     result = client.reset()
-        ...     print(result.observation.terminal_feedback)
-        ...
-        ...     result = client.step(DocAction(tool_name="open", path="/docs/setup.md"))
-        ...     print(result.observation.file_content)
-    """
 
     def _step_payload(self, action: DocAction) -> Dict[str, Any]:
         """
-        Convert DocAction to JSON payload for step request.
 
         Args:
             action: DocAction instance with tool parameters.
-
-        Returns:
-            Dictionary representation suitable for JSON encoding.
         """
         return {
             "tool_name": action.tool_name,
@@ -57,13 +33,9 @@ class DocSweeperEnv(EnvClient[DocAction, DocObservation, DocState]):
 
     def _parse_result(self, payload: Dict[str, Any]) -> StepResult[DocObservation]:
         """
-        Parse server response into StepResult[DocObservation].
-
         Args:
             payload: JSON response from server.
 
-        Returns:
-            StepResult with DocObservation.
         """
         obs_data = payload.get("observation", {})
 
@@ -85,13 +57,9 @@ class DocSweeperEnv(EnvClient[DocAction, DocObservation, DocState]):
 
     def _parse_state(self, payload: Dict[str, Any]) -> DocState:
         """
-        Parse server response into DocState object.
-
         Args:
             payload: JSON response from /state endpoint.
 
-        Returns:
-            DocState object with environment state information.
         """
         return DocState(
             episode_id=payload.get("episode_id", ""),

inference.py

@@ -1,5 +1,3 @@
-"""Inference script for the Doc Sweeper Environment."""
-
 import os
 import json
 import time

main.py

@@ -1,5 +1,5 @@
 def main():
-    print("Hello from cust-env!")
+    print("Hello from DocSweeper!")
 
 
 if __name__ == "__main__":

models.py

@@ -4,13 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""
-Data models for Doc Sweeper Environment.
-
-This module defines the Action, Observation, and State types for the documentation
-maintenance tasks via the OpenEnv interface.
-"""
-
 from __future__ import annotations
 
 from typing import Dict, List, Optional
@@ -21,8 +14,6 @@ from pydantic import Field
 
 class DocAction(Action):
     """
-    Action for Doc Sweeper environment.
-
     Attributes:
         tool_name: The command to run ('open', 'edit', 'grep', 'done').
         path: File path for opening or editing.
@@ -40,7 +31,6 @@ class DocAction(Action):
 
 class DocObservation(Observation):
     """
-    Observation for Doc Sweeper environment.
 
     Attributes:
         active_file: Currently opened file path.
@@ -62,8 +52,6 @@ class DocObservation(Observation):
 
 class DocState(State):
     """
-    State for Doc Sweeper environment.
-
     Attributes:
         episode_id: Unique ID for the current task session.
         step_count: Number of actions taken.

pyproject.toml

@@ -13,3 +13,5 @@ dependencies = [
     "pydantic>=2.12.5",
     "uvicorn>=0.42.0",
 ]
+[project.scripts]
+server = "server.app:main"

server/__init__.py

@@ -4,8 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""Cust Env environment server components."""
-
 from .cust_env_environment import DocSweeperEnvironment
 
 __all__ = ["DocSweeperEnvironment"]

server/app.py

@@ -4,8 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""FastAPI application for the Doc Sweeper Environment."""
-
 from openenv.core.env_server import create_app
 import uvicorn
 
@@ -17,7 +15,6 @@ from .cust_env_environment import DocSweeperEnvironment
 app = create_app(DocSweeperEnvironment, DocAction, DocObservation, env_name="doc_sweeper")
 
 def main():
-    """Entry point for the OpenEnv multi-mode deployment."""
     uvicorn.run("server.app:app", host="0.0.0.0", port=8000)
 
 if __name__ == '__main__':

server/cust_env_environment.py

@@ -4,13 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-"""
-Doc Sweeper Environment for OpenEnv.
-
-This module provides an environment where an agent navigates a virtual file system
-to fix outdated documentation, broken links, and deprecated configurations.
-"""
-
 import uuid
 from typing import Dict, List
 
@@ -20,12 +13,6 @@ from models import DocAction, DocObservation, DocState
 
 
 class DocSweeperEnvironment(Environment):
-    """
-    Doc Sweeper environment implementing the OpenEnv interface.
-
-    Simulates a Virtual File System (VFS) of markdown and config files. The agent
-    must find and replace deprecated patterns without destructive behavior.
-    """
 
     def __init__(
         self,
@@ -33,8 +20,6 @@ class DocSweeperEnvironment(Environment):
         max_steps: int = 30,
     ):
         """
-        Initialize the Doc Sweeper environment.
-
         Args:
             task: Task to run - "version_bump", "config_migration", or "broken_links".
             max_steps: Maximum allowed actions before forced termination.
@@ -48,8 +33,6 @@ class DocSweeperEnvironment(Environment):
 
     def reset(self, **kwargs):
         """
-        Initialize a new task episode.
-
         Returns:
             Initial observation of the virtual file system.
         """
@@ -86,13 +69,8 @@ class DocSweeperEnvironment(Environment):
 
     def step(self, action: DocAction):
         """
-        Execute an action and return the resulting state.
-
         Args:
             action: The tool action to execute (open, edit, grep, done).
-
-        Returns:
-            Observation with reward and done flag.
         """
         if self._state is None:
             raise RuntimeError("Environment not initialized. Call reset() first.")
@@ -159,7 +137,6 @@ class DocSweeperEnvironment(Environment):
             return -0.1
 
     def _evaluate_final_grade(self) -> float:
-        # Simplified deterministic grader for example purposes
         text = "".join(self._state.vfs.values())
         if self._task == "version_bump":
             target_count = text.count("v2.0.0")
@@ -189,5 +166,4 @@ class DocSweeperEnvironment(Environment):
 
     @property
     def state(self):
-        """Return the current episode state."""
         return self._state