wcgw 2.1.2__tar.gz → 2.2.0__tar.gz

{wcgw-2.1.2 → wcgw-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 2.1.2
+Version: 2.2.0
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -9,7 +9,6 @@ Requires-Dist: anthropic>=0.39.0
 Requires-Dist: fastapi>=0.115.0
 Requires-Dist: humanize>=4.11.0
 Requires-Dist: mcp
-Requires-Dist: mypy>=1.11.2
 Requires-Dist: nltk>=3.9.1
 Requires-Dist: openai>=1.46.0
 Requires-Dist: petname>=2.6
@@ -33,12 +32,17 @@ Description-Content-Type: text/markdown
 - Claude - An MCP server on claude desktop for autonomous shell, coding and desktop control agent. (mac only)
 - Chatgpt - Allows custom gpt to talk to your shell via a relay server. (linux or mac)
 
+
+⚠️ Warning: do not use this repo if you aren't scared of "Autonomous shell command execution"
+
 [![Tests](https://github.com/rusiaaman/wcgw/actions/workflows/python-tests.yml/badge.svg?branch=main)](https://github.com/rusiaaman/wcgw/actions/workflows/python-tests.yml)
 [![Mypy strict](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml/badge.svg?branch=main)](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml)
 [![Build](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml/badge.svg)](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml)
 
 ## Updates
 
+- [9 Dec 2024] [Vscode extension to paste context on Claude app](https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw)  
+
 - [01 Dec 2024] Removed author hosted relay server for chatgpt.
 
 - [26 Nov 2024] Introduced claude desktop support through mcp
@@ -143,7 +147,7 @@ The following requirements should be installed and working in the linux docker i
 2. Needs `scrot` to take screenshots.
 3. Needs `convert` from imagemagick to convert images.
 
-## Usage
+### Usage
 
 Wait for a few seconds. You should be able to see this icon if everything goes right.
 
@@ -156,6 +160,13 @@ Then ask claude to execute shell commands, read files, edit files, run your code
 
 If you've run the docker for LLM to access, you can ask it to control the "docker os". If you don't provide the docker container id to it, it'll try to search for available docker using `docker ps` command.
 
+
+### [Optional] Vs code extension 
+https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw
+
+Commands: 
+- Select a text and press `cmd+'` and then enter instructions. This will switch the app to Claude and paste a text containing your instructions, file path, workspace dir, and the selected text.
+
 ## Chatgpt Setup
 
 Read here: https://github.com/rusiaaman/wcgw/blob/main/openai.md

{wcgw-2.1.2 → wcgw-2.2.0}/README.md

@@ -3,12 +3,17 @@
 - Claude - An MCP server on claude desktop for autonomous shell, coding and desktop control agent. (mac only)
 - Chatgpt - Allows custom gpt to talk to your shell via a relay server. (linux or mac)
 
+
+⚠️ Warning: do not use this repo if you aren't scared of "Autonomous shell command execution"
+
 [![Tests](https://github.com/rusiaaman/wcgw/actions/workflows/python-tests.yml/badge.svg?branch=main)](https://github.com/rusiaaman/wcgw/actions/workflows/python-tests.yml)
 [![Mypy strict](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml/badge.svg?branch=main)](https://github.com/rusiaaman/wcgw/actions/workflows/python-types.yml)
 [![Build](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml/badge.svg)](https://github.com/rusiaaman/wcgw/actions/workflows/python-publish.yml)
 
 ## Updates
 
+- [9 Dec 2024] [Vscode extension to paste context on Claude app](https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw)  
+
 - [01 Dec 2024] Removed author hosted relay server for chatgpt.
 
 - [26 Nov 2024] Introduced claude desktop support through mcp
@@ -113,7 +118,7 @@ The following requirements should be installed and working in the linux docker i
 2. Needs `scrot` to take screenshots.
 3. Needs `convert` from imagemagick to convert images.
 
-## Usage
+### Usage
 
 Wait for a few seconds. You should be able to see this icon if everything goes right.
 
@@ -126,6 +131,13 @@ Then ask claude to execute shell commands, read files, edit files, run your code
 
 If you've run the docker for LLM to access, you can ask it to control the "docker os". If you don't provide the docker container id to it, it'll try to search for available docker using `docker ps` command.
 
+
+### [Optional] Vs code extension 
+https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw
+
+Commands: 
+- Select a text and press `cmd+'` and then enter instructions. This will switch the app to Claude and paste a text containing your instructions, file path, workspace dir, and the selected text.
+
 ## Chatgpt Setup
 
 Read here: https://github.com/rusiaaman/wcgw/blob/main/openai.md

{wcgw-2.1.2 → wcgw-2.2.0}/gpt_action_json_schema.json

@@ -19,7 +19,7 @@
           "content": {
             "application/json": {
               "schema": {
-                "$ref": "#/components/schemas/CreateFileNewWithUUID"
+                "$ref": "#/components/schemas/WriteIfEmptyWithUUID"
               }
             }
           },
@@ -384,7 +384,7 @@
         ],
         "title": "CommandWithUUID"
       },
-      "CreateFileNewWithUUID": {
+      "WriteIfEmptyWithUUID": {
         "properties": {
           "file_path": {
             "type": "string",
@@ -406,7 +406,7 @@
           "file_content",
           "user_id"
         ],
-        "title": "CreateFileNewWithUUID"
+        "title": "WriteIfEmptyWithUUID"
       },
       "FileEditWithUUID": {
         "properties": {

{wcgw-2.1.2 → wcgw-2.2.0}/gpt_instructions.txt

@@ -24,8 +24,8 @@ Instructions for `Read File`
 - Read full content of a file.
 - Provide absolute file path only.
 
-Instructions for `Create File New`
-- Write content to a new file. Provide file path and content. Use this instead of BashCommand for writing new files.
+Instructions for `Write if Empty`
+- Write content to an empty or non-existent file. Provide file path and content. Use this instead of BashCommand for writing new files.
 - Provide absolute file path only.
 - For editing existing files, use FileEdit.
 

{wcgw-2.1.2 → wcgw-2.2.0}/pyproject.toml

@@ -1,13 +1,12 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "gapypi@arcfu.com" }]
 name = "wcgw"
-version = "2.1.2"
+version = "2.2.0"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11, <3.13"
 dependencies = [
     "openai>=1.46.0",
-    "mypy>=1.11.2",
     "typer>=0.12.5",
     "rich>=13.8.1",
     "python-dotenv>=1.0.1",

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/client/anthropic_client.py

@@ -24,7 +24,7 @@ import uuid
 from ..types_ import (
     BashCommand,
     BashInteraction,
-    CreateFileNew,
+    WriteIfEmpty,
     FileEditFindReplace,
     FileEdit,
     Keyboard,
@@ -198,10 +198,10 @@ def loop(
 """,
         ),
         ToolParam(
-            input_schema=CreateFileNew.model_json_schema(),
-            name="CreateFileNew",
+            input_schema=WriteIfEmpty.model_json_schema(),
+            name="WriteIfEmpty",
             description="""
-- Write content to a new file. Provide file path and content. Use this instead of BashCommand for writing new files.
+- Write content to an empty or non-existent file. Provide file path and content. Use this instead of BashCommand for writing new files.
 - Provide absolute file path only.
 - For editing existing files, use FileEdit instead of this tool.
 """,
@@ -495,7 +495,12 @@ System information:
                             )
                         else:
                             _histories.append(
-                                {"role": "assistant", "content": full_response}
+                                {
+                                    "role": "assistant",
+                                    "content": full_response
+                                    if full_response.strip()
+                                    else "...",
+                                }  # Fixes anthropic issue of non empty response only
                             )
 
         except KeyboardInterrupt:

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/client/mcp_server/server.py

@@ -17,7 +17,7 @@ from ..tools import DoneFlag, get_tool_output, which_tool_name, default_enc
 from ...types_ import (
     BashCommand,
     BashInteraction,
-    CreateFileNew,
+    WriteIfEmpty,
     FileEdit,
     Keyboard,
     Mouse,
@@ -117,10 +117,10 @@ async def handle_list_tools() -> list[types.Tool]:
 """,
         ),
         ToolParam(
-            inputSchema=CreateFileNew.model_json_schema(),
-            name="CreateFileNew",
+            inputSchema=WriteIfEmpty.model_json_schema(),
+            name="WriteIfEmpty",
             description="""
-- Write content to a new file. Provide file path and content. Use this instead of BashCommand for writing new files.
+- Write content to an empty or non-existent file. Provide file path and content. Use this instead of BashCommand for writing new files.
 - Provide absolute file path only.
 - For editing existing files, use FileEdit instead of this tool.
 """,

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/client/openai_client.py

@@ -23,7 +23,7 @@ import uuid
 from ..types_ import (
     BashCommand,
     BashInteraction,
-    CreateFileNew,
+    WriteIfEmpty,
     FileEdit,
     ReadImage,
     ReadFile,
@@ -195,9 +195,9 @@ def loop(
 """,
         ),
         openai.pydantic_function_tool(
-            CreateFileNew,
+            WriteIfEmpty,
             description="""
-- Write content to a new file. Provide file path and content. Use this instead of BashCommand for writing new files.
+- Write content to an empty or non-existent file. Provide file path and content. Use this instead of BashCommand for writing new files.
 - Provide absolute file path only.
 - For editing existing files, use FileEdit instead of this tool.""",
         ),

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/client/tools.py

@@ -55,7 +55,7 @@ from nltk.metrics.distance import edit_distance  # type: ignore[import-untyped]
 from ..types_ import (
     BashCommand,
     BashInteraction,
-    CreateFileNew,
+    WriteIfEmpty,
     FileEditFindReplace,
     FileEdit,
     Initialize,
@@ -531,7 +531,7 @@ def read_image_from_shell(file_path: str) -> ImageData:
             return ImageData(media_type=image_type, data=image_b64)  # type: ignore
 
 
-def write_file(writefile: CreateFileNew, error_on_exist: bool) -> str:
+def write_file(writefile: WriteIfEmpty, error_on_exist: bool) -> str:
     if not os.path.isabs(writefile.file_path):
         return f"Failure: file_path should be absolute path, current working directory is {BASH_STATE.cwd}"
     else:
@@ -620,7 +620,7 @@ def find_least_edit_distance_substring(
                 + 1
             )
             min_edit_distance_lines = orig_content_lines[
-                orig_start_index:orig_end_index
+                max(0, orig_start_index - 10) : (orig_end_index + 10)
             ]
     return "\n".join(min_edit_distance_lines), min_edit_distance
 
@@ -638,7 +638,9 @@ def edit_content(content: str, find_lines: str, replace_with_lines: str) -> str:
                 f"Exact match not found, found with whitespace removed edit distance: {min_edit_distance}"
             )
         raise Exception(
-            f"Error: no match found for the provided `find_lines` in the file. Closest match:\n---\n{closest_match}\n---\nFile not edited"
+            f"""Error: no match found for the provided search block.
+                Requested search block: \n```\n{find_lines}\n```
+                Possible relevant section in the file:\n---\n```\n{closest_match}\n```\n---\nFile not edited"""
         )
 
     content = content.replace(find_lines, replace_with_lines, 1)
@@ -765,7 +767,7 @@ TOOLS = (
     | BashCommand
     | BashInteraction
     | ResetShell
-    | CreateFileNew
+    | WriteIfEmpty
     | FileEditFindReplace
     | FileEdit
     | AIAssistant
@@ -794,8 +796,8 @@ def which_tool_name(name: str) -> Type[TOOLS]:
         return BashInteraction
     elif name == "ResetShell":
         return ResetShell
-    elif name == "CreateFileNew":
-        return CreateFileNew
+    elif name == "WriteIfEmpty":
+        return WriteIfEmpty
     elif name == "FileEditFindReplace":
         return FileEditFindReplace
     elif name == "FileEdit":
@@ -828,7 +830,7 @@ def get_tool_output(
     | BashCommand
     | BashInteraction
     | ResetShell
-    | CreateFileNew
+    | WriteIfEmpty
     | FileEditFindReplace
     | FileEdit
     | AIAssistant
@@ -852,7 +854,7 @@ def get_tool_output(
             | BashCommand
             | BashInteraction
             | ResetShell
-            | CreateFileNew
+            | WriteIfEmpty
             | FileEditFindReplace
             | FileEdit
             | AIAssistant
@@ -869,7 +871,7 @@ def get_tool_output(
             | BashCommand
             | BashInteraction
             | ResetShell
-            | CreateFileNew
+            | WriteIfEmpty
             | FileEditFindReplace
             | FileEdit
             | AIAssistant
@@ -892,7 +894,7 @@ def get_tool_output(
     elif isinstance(arg, (BashCommand | BashInteraction)):
         console.print("Calling execute bash tool")
         output = execute_bash(enc, arg, max_tokens, None)
-    elif isinstance(arg, CreateFileNew):
+    elif isinstance(arg, WriteIfEmpty):
         console.print("Calling write file tool")
         output = write_file(arg, True), 0
     elif isinstance(arg, FileEdit):
@@ -915,6 +917,8 @@ def get_tool_output(
         output = reset_shell(), 0.0
     elif isinstance(arg, Initialize):
         console.print("Calling initial info tool")
+        # First force reset
+        reset_shell()
         output = initial_info(), 0.0
     elif isinstance(arg, (Mouse, Keyboard, ScreenShot, GetScreenInfo)):
         console.print(f"Calling {type(arg).__name__} tool")
@@ -974,7 +978,7 @@ class Mdata(BaseModel):
     data: (
         BashCommand
         | BashInteraction
-        | CreateFileNew
+        | WriteIfEmpty
         | ResetShell
         | FileEditFindReplace
         | FileEdit

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/relay/serve.py

@@ -17,7 +17,7 @@ from dotenv import load_dotenv
 from ..types_ import (
     BashCommand,
     BashInteraction,
-    CreateFileNew,
+    WriteIfEmpty,
     FileEditFindReplace,
     FileEdit,
     Initialize,
@@ -31,7 +31,7 @@ class Mdata(BaseModel):
     data: (
         BashCommand
         | BashInteraction
-        | CreateFileNew
+        | WriteIfEmpty
         | ResetShell
         | FileEditFindReplace
         | FileEdit
@@ -99,12 +99,12 @@ async def register_websocket(websocket: WebSocket, uuid: UUID) -> None:
         print(f"Client {uuid} disconnected")
 
 
-class CreateFileNewWithUUID(CreateFileNew):
+class WriteIfEmptyWithUUID(WriteIfEmpty):
     user_id: UUID
 
 
 @app.post("/v1/create_file")
-async def create_file(write_file_data: CreateFileNewWithUUID) -> str:
+async def create_file(write_file_data: WriteIfEmptyWithUUID) -> str:
     user_id = write_file_data.user_id
     if user_id not in clients:
         return "Failure: id not found, ask the user to check it."

{wcgw-2.1.2 → wcgw-2.2.0}/src/wcgw/types_.py

@@ -24,7 +24,7 @@ class ReadImage(BaseModel):
     type: Literal["ReadImage"]
 
 
-class CreateFileNew(BaseModel):
+class WriteIfEmpty(BaseModel):
     file_path: str
     file_content: str
 

{wcgw-2.1.2 → wcgw-2.2.0}/tests/test_tools.py

@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import patch
 from src.wcgw.client.tools import render_terminal_output, ask_confirmation, Confirmation
-from src.wcgw.types_ import CreateFileNew
+from src.wcgw.types_ import WriteIfEmpty
 
 
 class TestTools(unittest.TestCase):
@@ -32,7 +32,7 @@ class TestTools(unittest.TestCase):
 
     def test_writefile_model(self):
         # Test the Writefile Pydantic model
-        file = CreateFileNew(file_path="test.txt", file_content="This is a test.")
+        file = WriteIfEmpty(file_path="test.txt", file_content="This is a test.")
         self.assertEqual(file.file_path, "test.txt")
         self.assertEqual(file.file_content, "This is a test.")
 

{wcgw-2.1.2 → wcgw-2.2.0}/uv.lock

@@ -869,14 +869,13 @@ wheels = [
 
 [[package]]
 name = "wcgw"
-version = "2.1.2"
+version = "2.2.0"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },
     { name = "fastapi" },
     { name = "humanize" },
     { name = "mcp" },
-    { name = "mypy" },
     { name = "nltk" },
     { name = "openai" },
     { name = "petname" },
@@ -910,7 +909,6 @@ requires-dist = [
     { name = "fastapi", specifier = ">=0.115.0" },
     { name = "humanize", specifier = ">=4.11.0" },
     { name = "mcp", git = "https://github.com/rusiaaman/python-sdk?rev=53b69f397eae6ac81a51b84b34ff52b3119f11cb" },
-    { name = "mypy", specifier = ">=1.11.2" },
     { name = "nltk", specifier = ">=3.9.1" },
     { name = "openai", specifier = ">=1.46.0" },
     { name = "petname", specifier = ">=2.6" },