wcgw 2.6.1__py3-none-any.whl → 2.6.3__py3-none-any.whl

wcgw/client/tools.py

@@ -1127,7 +1127,7 @@ TOOLS = (
 
 
 def which_tool(args: str) -> TOOLS:
-    adapter = TypeAdapter[TOOLS](TOOLS)
+    adapter = TypeAdapter[TOOLS](TOOLS, config={"extra": "forbid"})
     return adapter.validate_python(json.loads(args))
 
 
@@ -1180,7 +1180,7 @@ def get_tool_output(
 ) -> tuple[list[str | ImageData | DoneFlag], float]:
     global IS_IN_DOCKER, TOOL_CALLS
     if isinstance(args, dict):
-        adapter = TypeAdapter[TOOLS](TOOLS)
+        adapter = TypeAdapter[TOOLS](TOOLS, config={"extra": "forbid"})
         arg = adapter.validate_python(args)
     else:
         arg = args
@@ -1242,9 +1242,7 @@ def get_tool_output(
                     # At this point we should go into the docker env
                     res, _ = execute_bash(
                         enc,
-                        BashInteraction(
-                            send_text=f"export PS1={PROMPT}", type="BashInteraction"
-                        ),
+                        BashInteraction(send_text=f"export PS1={PROMPT}"),
                         None,
                         0.2,
                     )

wcgw/types_.py

@@ -1,6 +1,14 @@
 from typing import Literal, Optional, Sequence
 
-from pydantic import BaseModel
+from pydantic import BaseModel as PydanticBaseModel
+
+
+class NoExtraArgs(PydanticBaseModel):
+    class Config:
+        extra = "forbid"
+
+
+BaseModel = NoExtraArgs
 
 
 class BashCommand(BaseModel):
@@ -14,7 +22,6 @@ Specials = Literal[
 
 
 class BashInteraction(BaseModel):
-    type: Literal["BashInteraction"]
     send_text: Optional[str] = None
     send_specials: Optional[Sequence[Specials]] = None
     send_ascii: Optional[Sequence[int]] = None
@@ -23,7 +30,6 @@ class BashInteraction(BaseModel):
 
 class ReadImage(BaseModel):
     file_path: str
-    type: Literal["ReadImage"]
 
 
 class WriteIfEmpty(BaseModel):
@@ -33,7 +39,6 @@ class WriteIfEmpty(BaseModel):
 
 class ReadFiles(BaseModel):
     file_paths: list[str]
-    type: Literal["ReadFiles"]
 
 
 class FileEditFindReplace(BaseModel):
@@ -52,18 +57,15 @@ class FileEdit(BaseModel):
 
 
 class Initialize(BaseModel):
-    type: Literal["Initialize"]
     any_workspace_path: str
     initial_files_to_read: list[str]
 
 
 class GetScreenInfo(BaseModel):
-    type: Literal["GetScreenInfo"]
     docker_image_id: str
 
 
 class ScreenShot(BaseModel):
-    type: Literal["ScreenShot"]
     take_after_delay_seconds: int
 
 
@@ -71,7 +73,6 @@ class MouseMove(BaseModel):
     x: int
     y: int
     do_left_click_on_move: bool
-    type: Literal["MouseMove"]
 
 
 class LeftClickDrag(BaseModel):

{wcgw-2.6.1.dist-info → wcgw-2.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 2.6.1
+Version: 2.6.3
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -41,6 +41,7 @@ Description-Content-Type: text/markdown
 [![codecov](https://codecov.io/gh/rusiaaman/wcgw/graph/badge.svg)](https://codecov.io/gh/rusiaaman/wcgw)
 
 ## Updates
+- [29 Dec 2024] Syntax checking on file writing and edits is now stable. Made `initialize` tool call useful; sending smart repo structure to claude if any repo is referenced. Large file handling is also now improved.
 
 - [9 Dec 2024] [Vscode extension to paste context on Claude app](https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw)  
 

{wcgw-2.6.1.dist-info → wcgw-2.6.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 wcgw/__init__.py,sha256=9K2QW7QuSLhMTVbKbBYd9UUp-ZyrfBrxcjuD_xk458k,118
-wcgw/types_.py,sha256=PJpnzF_TJ_L-AhruPlfD-zkGYj0_YEUAXTbvOtb_Zpo,1891
+wcgw/types_.py,sha256=pi976GyRLBKeCcpqNmVm5tI0iXO4okkkUGDa41ITr50,1792
 wcgw/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wcgw/client/__main__.py,sha256=wcCrL4PjG51r5wVKqJhcoJPTLfHW0wNbD31DrUN0MWI,28
 wcgw/client/anthropic_client.py,sha256=kmS93LEVwXCppfbgCsfWe_SVmTAtsI3x8PrP8DJDYMY,21041
@@ -10,7 +10,7 @@ wcgw/client/diff-instructions.txt,sha256=s5AJKG23JsjwRYhFZFQVvwDpF67vElawrmdXwvu
 wcgw/client/openai_client.py,sha256=AP0B-fKNNlYCAEL0MrT3UxeCcvca7jJZp9kYkinUqSM,17706
 wcgw/client/openai_utils.py,sha256=KfMB1-p2zDiA7pPWwAVarochf7-qeL1UMgtlDV9DtKA,2662
 wcgw/client/sys_utils.py,sha256=GajPntKhaTUMn6EOmopENWZNR2G_BJyuVbuot0x6veI,1376
-wcgw/client/tools.py,sha256=oJ9Zg03CEbzMh8NXQc1lfZQv8C46jQ24PRXGYMiHJ9k,47987
+wcgw/client/tools.py,sha256=2Us5nypzTbwpNpXNkCvL94L3ouEmD5dPEKG7BH0lgts,47965
 wcgw/client/mcp_server/Readme.md,sha256=I8N4dHkTUVGNQ63BQkBMBhCCBTgqGOSF_pUR6iOEiUk,2495
 wcgw/client/mcp_server/__init__.py,sha256=hyPPwO9cabAJsOMWhKyat9yl7OlSmIobaoAZKHu3DMc,381
 wcgw/client/mcp_server/server.py,sha256=bhswcJQt2jWIwVcQRrEaa9E6LkuE_vK7fNG51bsV4hw,12414
@@ -43,8 +43,8 @@ mcp_wcgw/shared/memory.py,sha256=dBsOghxHz8-tycdSVo9kSujbsC8xb_tYsGmuJobuZnw,281
 mcp_wcgw/shared/progress.py,sha256=ymxOsb8XO5Mhlop7fRfdbmvPodANj7oq6O4dD0iUcnw,1048
 mcp_wcgw/shared/session.py,sha256=e44a0LQOW8gwdLs9_DE9oDsxqW2U8mXG3d5KT95bn5o,10393
 mcp_wcgw/shared/version.py,sha256=d2LZii-mgsPIxpshjkXnOTUmk98i0DT4ff8VpA_kAvE,111
-wcgw-2.6.1.dist-info/METADATA,sha256=AY-QOt4XUDIE6Pd7axFCXgSj25XkI9k9gsJCz3LHxjM,6762
-wcgw-2.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wcgw-2.6.1.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
-wcgw-2.6.1.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
-wcgw-2.6.1.dist-info/RECORD,,
+wcgw-2.6.3.dist-info/METADATA,sha256=97Rxv30HNmK8jUnc--eRxN3IkPlzn_OuMn27VgpxsQ4,6979
+wcgw-2.6.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-2.6.3.dist-info/entry_points.txt,sha256=eKo1omwbAggWlQ0l7GKoR7uV1-j16nk9tK0BhC2Oz_E,120
+wcgw-2.6.3.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
+wcgw-2.6.3.dist-info/RECORD,,