wcgw 5.4.1__py3-none-any.whl → 5.4.3__py3-none-any.whl

wcgw/client/bash_state/bash_state.py

@@ -4,6 +4,7 @@ import os
 import platform
 import random
 import subprocess
+import tempfile
 import threading
 import time
 import traceback
@@ -65,7 +66,7 @@ def is_mac() -> bool:
 def get_tmpdir() -> str:
     current_tmpdir = os.environ.get("TMPDIR", "")
     if current_tmpdir or not is_mac():
-        return current_tmpdir
+        return tempfile.gettempdir()
     try:
         # Fix issue while running ocrmypdf -> tesseract -> leptonica, set TMPDIR
         # https://github.com/tesseract-ocr/tesseract/issues/4333
@@ -78,7 +79,7 @@ def get_tmpdir() -> str:
     except (subprocess.CalledProcessError, FileNotFoundError):
         return "//tmp"
     except Exception:
-        return ""
+        return tempfile.gettempdir()
 
 
 def check_if_screen_command_available() -> bool:

wcgw/client/mcp_server/__init__.py

@@ -1,14 +1,27 @@
 # mypy: disable-error-code="import-untyped"
-from wcgw.client.mcp_server import server
 import asyncio
+import importlib
+
+import typer
 from typer import Typer
 
+from wcgw.client.mcp_server import server
+
 main = Typer()
 
 
 @main.command()
-def app() -> None:
+def app(
+    version: bool = typer.Option(
+        False, "--version", "-v", help="Show version and exit"
+    ),
+) -> None:
     """Main entry point for the package."""
+    if version:
+        version_ = importlib.metadata.version("wcgw")
+        print(f"wcgw version: {version_}")
+        raise typer.Exit()
+
     asyncio.run(server.main())
 
 

wcgw/client/schema_generator.py

@@ -0,0 +1,63 @@
+"""
+Custom JSON schema generator to remove title fields from Pydantic models.
+
+This module provides utilities to remove auto-generated title fields from JSON schemas,
+making them more suitable for tool schemas where titles are not needed.
+"""
+
+import copy
+from typing import Any, Dict
+
+
+def recursive_purge_dict_key(d: Dict[str, Any], k: str) -> None:
+    """
+    Remove a key from a dictionary recursively, but only from JSON schema metadata.
+    
+    This function removes the specified key from dictionaries that appear to be
+    JSON schema objects (have "type" or "$ref" or are property definitions).
+    This prevents removing legitimate data fields that happen to have the same name.
+    
+    Args:
+        d: The dictionary to clean
+        k: The key to remove (typically "title")
+    """
+    if isinstance(d, dict):
+        # Only remove the key if this looks like a JSON schema object
+        # This includes objects with "type", "$ref", or if we're in a "properties" context
+        is_schema_object = (
+            "type" in d or 
+            "$ref" in d or 
+            any(schema_key in d for schema_key in ["properties", "items", "additionalProperties", "enum", "const", "anyOf", "allOf", "oneOf"])
+        )
+        
+        if is_schema_object and k in d:
+            del d[k]
+        
+        # Recursively process all values, regardless of key names
+        # This ensures we catch all nested structures
+        for key, value in d.items():
+            if isinstance(value, dict):
+                recursive_purge_dict_key(value, k)
+            elif isinstance(value, list):
+                for item in value:
+                    if isinstance(item, dict):
+                        recursive_purge_dict_key(item, k)
+
+
+def remove_titles_from_schema(schema: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Remove all 'title' keys from a JSON schema dictionary.
+
+    This function creates a copy of the schema and removes all title keys
+    recursively, making it suitable for use with APIs that don't need titles.
+
+    Args:
+        schema: The JSON schema dictionary to clean
+
+    Returns:
+        A new dictionary with all title keys removed
+    """
+
+    schema_copy = copy.deepcopy(schema)
+    recursive_purge_dict_key(schema_copy, "title")
+    return schema_copy

wcgw/client/tool_prompts.py

@@ -10,6 +10,7 @@ from ..types_ import (
     ReadFiles,
     ReadImage,
 )
+from .schema_generator import remove_titles_from_schema
 
 with open(os.path.join(os.path.dirname(__file__), "diff-instructions.txt")) as f:
     diffinstructions = f.read()
@@ -17,7 +18,7 @@ with open(os.path.join(os.path.dirname(__file__), "diff-instructions.txt")) as f
 
 TOOL_PROMPTS = [
     Tool(
-        inputSchema=Initialize.model_json_schema(),
+        inputSchema=remove_titles_from_schema(Initialize.model_json_schema()),
         name="Initialize",
         description="""
 - Always call this at the start of the conversation before using any of the shell tools from wcgw.
@@ -34,7 +35,7 @@ TOOL_PROMPTS = [
         annotations=ToolAnnotations(readOnlyHint=True, openWorldHint=False),
     ),
     Tool(
-        inputSchema=BashCommand.model_json_schema(),
+        inputSchema=remove_titles_from_schema(BashCommand.model_json_schema()),
         name="BashCommand",
         description="""
 - Execute a bash command. This is stateful (beware with subsequent calls).
@@ -51,7 +52,7 @@ TOOL_PROMPTS = [
         annotations=ToolAnnotations(destructiveHint=True, openWorldHint=True),
     ),
     Tool(
-        inputSchema=ReadFiles.model_json_schema(),
+        inputSchema=remove_titles_from_schema(ReadFiles.model_json_schema()),
         name="ReadFiles",
         description="""
 - Read full file content of one or more files.
@@ -62,13 +63,13 @@ TOOL_PROMPTS = [
         annotations=ToolAnnotations(readOnlyHint=True, openWorldHint=False),
     ),
     Tool(
-        inputSchema=ReadImage.model_json_schema(),
+        inputSchema=remove_titles_from_schema(ReadImage.model_json_schema()),
         name="ReadImage",
         description="Read an image from the shell.",
         annotations=ToolAnnotations(readOnlyHint=True, openWorldHint=False),
     ),
     Tool(
-        inputSchema=FileWriteOrEdit.model_json_schema(),
+        inputSchema=remove_titles_from_schema(FileWriteOrEdit.model_json_schema()),
         name="FileWriteOrEdit",
         description="""
 - Writes or edits a file based on the percentage of changes.
@@ -84,7 +85,7 @@ TOOL_PROMPTS = [
         ),
     ),
     Tool(
-        inputSchema=ContextSave.model_json_schema(),
+        inputSchema=remove_titles_from_schema(ContextSave.model_json_schema()),
         name="ContextSave",
         description="""
 Saves provided description and file contents of all the relevant file paths or globs in a single text file.

wcgw/client/tools.py

@@ -6,6 +6,7 @@ import mimetypes
 import os
 import subprocess
 import traceback
+import uuid
 from dataclasses import dataclass
 from hashlib import sha256
 from os.path import expanduser
@@ -34,6 +35,7 @@ from ..client.bash_state.bash_state import (
     execute_bash,
     generate_thread_id,
     get_status,
+    get_tmpdir,
 )
 from ..client.repo_ops.file_stats import (
     FileStats,
@@ -153,6 +155,12 @@ def initialize(
         )
 
     folder_to_start = None
+    if type == "first_call" and not any_workspace_path:
+        tmp_dir = get_tmpdir()
+        any_workspace_path = os.path.join(
+            tmp_dir, "claude-playground-" + uuid.uuid4().hex[:4]
+        )
+
     if any_workspace_path:
         if os.path.exists(any_workspace_path):
             if os.path.isfile(any_workspace_path):
@@ -271,29 +279,38 @@ def initialize(
     except Exception:
         pass
 
-    # First check for global CLAUDE.md in ~/.wcgw/CLAUDE.md
-    global_alignment_file_path = os.path.join(expanduser("~"), ".wcgw", "CLAUDE.md")
-    if os.path.exists(global_alignment_file_path):
-        try:
-            with open(global_alignment_file_path, "r") as f:
-                global_alignment_content = f.read()
-            alignment_context += f"---\n# Important guidelines from the user\n```\n{global_alignment_content}\n```\n---\n\n"
-        except Exception:
-            # Handle any errors when reading the global file
-            pass
+    # Check for global alignment doc in ~/.wcgw: prefer CLAUDE.md, else AGENTS.md
+    try:
+        global_dir = os.path.join(expanduser("~"), ".wcgw")
+        for fname in ("CLAUDE.md", "AGENTS.md"):
+            global_alignment_file_path = os.path.join(global_dir, fname)
+            if os.path.exists(global_alignment_file_path):
+                with open(global_alignment_file_path, "r") as f:
+                    global_alignment_content = f.read()
+                alignment_context += f"---\n# Important guidelines from the user\n```\n{global_alignment_content}\n```\n---\n\n"
+                break
+    except Exception as e:
+        # Log any errors when reading the global file
+        context.console.log(f"Error reading global alignment file: {e}")
 
-    # Then check for workspace-specific CLAUDE.md
+    # Then check for workspace-specific alignment doc: prefer CLAUDE.md, else AGENTS.md
     if folder_to_start:
-        alignment_file_path = os.path.join(folder_to_start, "CLAUDE.md")
-        if os.path.exists(alignment_file_path):
-            try:
-                # Read the CLAUDE.md file content
-                with open(alignment_file_path, "r") as f:
-                    alignment_content = f.read()
-                alignment_context += f"---\n# CLAUDE.md - user shared project guidelines to follow\n```\n{alignment_content}\n```\n---\n\n"
-            except Exception:
-                # Handle any errors when reading the file
-                pass
+        try:
+            base_dir = str(folder_to_start)
+            selected_name = ""
+            alignment_content = ""
+            for fname in ("CLAUDE.md", "AGENTS.md"):
+                alignment_file_path = os.path.join(base_dir, fname)
+                if os.path.exists(alignment_file_path):
+                    with open(alignment_file_path, "r") as f:
+                        alignment_content = f.read()
+                    selected_name = fname
+                    break
+            if alignment_content:
+                alignment_context += f"---\n# {selected_name} - user shared project guidelines to follow\n```\n{alignment_content}\n```\n---\n\n"
+        except Exception as e:
+            # Log any errors when reading the workspace file
+            context.console.log(f"Error reading workspace alignment file: {e}")
 
     uname_sysname = os.uname().sysname
     uname_machine = os.uname().machine

wcgw/types_.py

@@ -49,7 +49,9 @@ class Initialize(BaseModel):
         "reset_shell",
         "user_asked_change_workspace",
     ]
-    any_workspace_path: str
+    any_workspace_path: str = Field(
+        description="Workspce to initialise in. Don't use ~ by default, instead use empty string"
+    )
     initial_files_to_read: list[str]
     task_id_to_resume: str
     mode_name: Literal["wcgw", "architect", "code_writer"]

{wcgw-5.4.1.dist-info → wcgw-5.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 5.4.1
+Version: 5.4.3
 Summary: Shell and coding agent for Claude and other mcp clients
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>

{wcgw-5.4.1.dist-info → wcgw-5.4.3.dist-info}/RECORD

@@ -1,14 +1,15 @@
 wcgw/__init__.py,sha256=JgAY25VsA208v8E7QTIU0E50nsk-TCJ4FWTEHmnssYU,127
 wcgw/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wcgw/types_.py,sha256=zsQVdY1TwDgsUBiKEjn0tbH7_UKSCJ68PD6-OGN9ln4,9494
+wcgw/types_.py,sha256=dvFpZPywqczpX7Sv6-e-r0WZxOZ0Eu10DlpSLtl9zJs,9607
 wcgw/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wcgw/client/common.py,sha256=OCH7Tx64jojz3M3iONUrGMadE07W21DiZs5sOxWX1Qc,1456
 wcgw/client/diff-instructions.txt,sha256=wCmB-Vj8HNSTvxjZIH0fX_J7v9jbsJOJeNj46Un1AJ0,1858
 wcgw/client/memory.py,sha256=U2Nw2si3Zg7n_RhNAuaYcmrrDtZ_Mooi-kfAOKflT-I,3079
 wcgw/client/modes.py,sha256=roH6SPBokJMr5IzAlccdI-vJyvyS5vqSMMyth7TE86A,10315
-wcgw/client/tool_prompts.py,sha256=7tq9ijSo4CznVlMFcWpauELfCT_QAqwi8Chsl3SRfBY,4539
-wcgw/client/tools.py,sha256=WAVnSqisV4hw4mPhbCzDDZy2gFkxXwTfzAajLPciCSw,49107
-wcgw/client/bash_state/bash_state.py,sha256=U-h9uqEcJbZGIVle6vZKYmlLHNO91zvqaUe6zlvLMl4,41876
+wcgw/client/schema_generator.py,sha256=mEIy6BgHlfJeAjJtwY_VwoIDmu-Fax2H9bVtj7IMuEo,2282
+wcgw/client/tool_prompts.py,sha256=1EFQZeXlebOvrDb9t4g63FyzRWCnTwDzwrqwPHg-7sE,4757
+wcgw/client/tools.py,sha256=GrJ9bXAkayd4bgKFdRXJ1wtdUmeE5dD7l2ef-dMOjZQ,49877
+wcgw/client/bash_state/bash_state.py,sha256=bd5RtLbaRzCtrmeDTl3JKZwzmIR-8iAMQpl7Fqyt56M,41918
 wcgw/client/bash_state/parser/__init__.py,sha256=AnlNSmoQTSoqqlLOLX4P1uXfzc5VGeCGJsGgtisq2zE,207
 wcgw/client/bash_state/parser/bash_statement_parser.py,sha256=9a8vPO1r3_tXmaAcubTQ5UY-NseWlalgm8LZA17LXuY,6058
 wcgw/client/encoder/__init__.py,sha256=Y-8f43I6gMssUCWpX5rLYiAFv3D-JPRs4uNEejPlke8,1514
@@ -16,7 +17,7 @@ wcgw/client/file_ops/diff_edit.py,sha256=AwLq6-pY7czv1y-JA5O2Q4rgbvn82YmSL9jD8XB
 wcgw/client/file_ops/extensions.py,sha256=CmfD7ON6SY24Prh2tRZdV9KbhuOrWqqk8qL1VtshzB8,3608
 wcgw/client/file_ops/search_replace.py,sha256=5LFg-_U_ijnNrkYei4SWCPGKPGgDzJs49EDsIBzLmuY,6822
 wcgw/client/mcp_server/Readme.md,sha256=2Z88jj1mf9daYGW1CWaldcJ0moy8owDumhR2glBY3A8,109
-wcgw/client/mcp_server/__init__.py,sha256=mm7xhBIPwJpRT3u-Qsj4cKVMpVyucJoKRlbMP_gRRB0,343
+wcgw/client/mcp_server/__init__.py,sha256=rSNET0SVjUTDn2HAfREisTXTBs89TVsWegDfYFMvy5w,621
 wcgw/client/mcp_server/server.py,sha256=RgsMDmNuYyg4VT1KorcLzh1Xfv49QASi0-FTLz_tlIo,5525
 wcgw/client/repo_ops/display_tree.py,sha256=g282qCKLCwo8O9NHUBnkG_NkIusroVzz3NZi8VIcmAI,4066
 wcgw/client/repo_ops/file_stats.py,sha256=AUA0Br7zFRpylWFYZPGMeGPJy3nWp9e2haKi34JptHE,4887
@@ -30,8 +31,8 @@ wcgw_cli/anthropic_client.py,sha256=8bjDY59-aioyTJgpB-NBHZNhZaq6rqcTJcOf81kzCyA,
 wcgw_cli/cli.py,sha256=-7FBe_lahKyUOhf65iurTA1M1gXXXAiT0OVKQVcZKKo,948
 wcgw_cli/openai_client.py,sha256=GOqoSFazTV-cFjpdZGPM0DIwec8Up2TEcKUbsN40AGY,15990
 wcgw_cli/openai_utils.py,sha256=xGOb3W5ALrIozV7oszfGYztpj0FnXdD7jAxm5lEIVKY,2439
-wcgw-5.4.1.dist-info/METADATA,sha256=kBQDxqLeVpBkBg-QoL34gM1mpdGC6zzXHh4UHgJ5mOc,15679
-wcgw-5.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wcgw-5.4.1.dist-info/entry_points.txt,sha256=UnjK-MAH4Qssh0tGJDMeij1oi-oRKokItkknP_BwShE,94
-wcgw-5.4.1.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
-wcgw-5.4.1.dist-info/RECORD,,
+wcgw-5.4.3.dist-info/METADATA,sha256=fNaQqNspbdo8tWK0-dVsYeaFr6i3kPdNffHoHn-JYss,15679
+wcgw-5.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-5.4.3.dist-info/entry_points.txt,sha256=UnjK-MAH4Qssh0tGJDMeij1oi-oRKokItkknP_BwShE,94
+wcgw-5.4.3.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
+wcgw-5.4.3.dist-info/RECORD,,