wcgw 3.0.6__py3-none-any.whl → 4.0.0__py3-none-any.whl

wcgw/relay/serve.py

@@ -16,23 +16,14 @@ from pydantic import BaseModel
 from ..types_ import (
     BashCommand,
     ContextSave,
-    FileEdit,
+    FileWriteOrEdit,
     Initialize,
     ReadFiles,
-    WriteIfEmpty,
 )
 
 
 class Mdata(BaseModel):
-    data: (
-        BashCommand
-        | WriteIfEmpty
-        | FileEdit
-        | ReadFiles
-        | Initialize
-        | ContextSave
-        | str
-    )
+    data: BashCommand | FileWriteOrEdit | ReadFiles | Initialize | ContextSave | str
     user_id: UUID
 
 
@@ -93,12 +84,12 @@ async def register_websocket(websocket: WebSocket, uuid: UUID) -> None:
         print(f"Client {uuid} disconnected")
 
 
-class WriteIfEmptyWithUUID(WriteIfEmpty):
+class FileWriteOrEdithUUID(FileWriteOrEdit):
     user_id: UUID
 
 
-@app.post("/v1/create_file")
-async def create_file(write_file_data: WriteIfEmptyWithUUID) -> str:
+@app.post("/v1/file_write_or_edit")
+async def file_write_or_edit(write_file_data: FileWriteOrEdithUUID) -> 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."
@@ -122,42 +113,6 @@ async def create_file(write_file_data: WriteIfEmptyWithUUID) -> str:
     raise fastapi.HTTPException(status_code=500, detail="Timeout error")
 
 
-class FileEditWithUUID(FileEdit):
-    user_id: UUID
-
-
-@app.post("/v1/full_file_edit")
-async def file_edit_find_replace(
-    file_edit_find_replace: FileEditWithUUID,
-) -> str:
-    user_id = file_edit_find_replace.user_id
-    if user_id not in clients:
-        return "Failure: id not found, ask the user to check it."
-
-    results: Optional[str] = None
-
-    def put_results(result: str) -> None:
-        nonlocal results
-        results = result
-
-    gpts[user_id] = put_results
-
-    await clients[user_id](
-        Mdata(
-            data=file_edit_find_replace,
-            user_id=user_id,
-        )
-    )
-
-    start_time = time.time()
-    while time.time() - start_time < 30:
-        if results is not None:
-            return results
-        await asyncio.sleep(0.1)
-
-    raise fastapi.HTTPException(status_code=500, detail="Timeout error")
-
-
 class CommandWithUUID(BashCommand):
     user_id: UUID
 
@@ -179,7 +134,8 @@ async def bash_command(command: CommandWithUUID) -> str:
     await clients[user_id](
         Mdata(
             data=BashCommand(
-                action_json=command.action_json, wait_for_seconds=command.wait_for_seconds
+                action_json=command.action_json,
+                wait_for_seconds=command.wait_for_seconds,
             ),
             user_id=user_id,
         )
@@ -281,7 +237,7 @@ async def context_save(context_save_data: ContextSaveWithUUID) -> str:
     raise fastapi.HTTPException(status_code=500, detail="Timeout error")
 
 
-app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/static", StaticFiles(directory="static"))
 
 
 def run() -> None:

wcgw/types_.py

@@ -1,7 +1,8 @@
 import os
-from typing import Any, Literal, Optional, Protocol, Sequence, Union
+from typing import Any, List, Literal, Optional, Protocol, Sequence, Union
 
 from pydantic import BaseModel as PydanticBaseModel
+from pydantic import PrivateAttr
 
 
 class NoExtraArgs(PydanticBaseModel):
@@ -56,9 +57,9 @@ class Initialize(BaseModel):
 
     def model_post_init(self, __context: Any) -> None:
         if self.mode_name == "code_writer":
-            assert (
-                self.code_writer_config is not None
-            ), "code_writer_config can't be null when the mode is code_writer"
+            assert self.code_writer_config is not None, (
+                "code_writer_config can't be null when the mode is code_writer"
+            )
         return super().model_post_init(__context)
 
     @property
@@ -67,9 +68,9 @@ class Initialize(BaseModel):
             return "wcgw"
         if self.mode_name == "architect":
             return "architect"
-        assert (
-            self.code_writer_config is not None
-        ), "code_writer_config can't be null when the mode is code_writer"
+        assert self.code_writer_config is not None, (
+            "code_writer_config can't be null when the mode is code_writer"
+        )
         return self.code_writer_config
 
 
@@ -114,6 +115,94 @@ class WriteIfEmpty(BaseModel):
 
 class ReadFiles(BaseModel):
     file_paths: list[str]
+    show_line_numbers_reason: Optional[str] = None
+    _start_line_nums: List[Optional[int]] = PrivateAttr(default_factory=lambda: [])
+    _end_line_nums: List[Optional[int]] = PrivateAttr(default_factory=lambda: [])
+
+    @property
+    def start_line_nums(self) -> List[Optional[int]]:
+        """Get the start line numbers."""
+        return self._start_line_nums
+
+    @property
+    def end_line_nums(self) -> List[Optional[int]]:
+        """Get the end line numbers."""
+        return self._end_line_nums
+
+    def model_post_init(self, __context: Any) -> None:
+        # Parse file paths for line ranges and store them in private attributes
+        self._start_line_nums = []
+        self._end_line_nums = []
+
+        # Create new file_paths list without line ranges
+        clean_file_paths = []
+
+        for file_path in self.file_paths:
+            start_line_num = None
+            end_line_num = None
+            path_part = file_path
+
+            # Check if the path ends with a line range pattern
+            # We're looking for patterns at the very end of the path like:
+            #  - file.py:10      (specific line)
+            #  - file.py:10-20   (line range)
+            #  - file.py:10-     (from line 10 to end)
+            #  - file.py:-20     (from start to line 20)
+
+            # Split by the last colon
+            if ":" in file_path:
+                parts = file_path.rsplit(":", 1)
+                if len(parts) == 2:
+                    potential_path = parts[0]
+                    line_spec = parts[1]
+
+                    # Check if it's a valid line range format
+                    if line_spec.isdigit():
+                        # Format: file.py:10
+                        try:
+                            start_line_num = int(line_spec)
+                            path_part = potential_path
+                        except ValueError:
+                            # Keep the original path if conversion fails
+                            pass
+
+                    elif "-" in line_spec:
+                        # Could be file.py:10-20, file.py:10-, or file.py:-20
+                        line_parts = line_spec.split("-", 1)
+
+                        if not line_parts[0] and line_parts[1].isdigit():
+                            # Format: file.py:-20
+                            try:
+                                end_line_num = int(line_parts[1])
+                                path_part = potential_path
+                            except ValueError:
+                                # Keep original path
+                                pass
+
+                        elif line_parts[0].isdigit():
+                            # Format: file.py:10-20 or file.py:10-
+                            try:
+                                start_line_num = int(line_parts[0])
+
+                                if line_parts[1].isdigit():
+                                    # file.py:10-20
+                                    end_line_num = int(line_parts[1])
+
+                                # In both cases, update the path
+                                path_part = potential_path
+                            except ValueError:
+                                # Keep original path
+                                pass
+
+            # Add clean path and corresponding line numbers
+            clean_file_paths.append(path_part)
+            self._start_line_nums.append(start_line_num)
+            self._end_line_nums.append(end_line_num)
+
+        # Update file_paths with clean paths
+        self.file_paths = clean_file_paths
+
+        return super().model_post_init(__context)
 
 
 class FileEdit(BaseModel):
@@ -121,6 +210,12 @@ class FileEdit(BaseModel):
     file_edit_using_search_replace_blocks: str
 
 
+class FileWriteOrEdit(BaseModel):
+    file_path: str
+    percentage_to_change: int  # 0.0 to 100.0
+    file_content_or_search_replace_blocks: str
+
+
 class ContextSave(BaseModel):
     id: str
     project_root_path: str
@@ -135,12 +230,4 @@ class Console(Protocol):
 
 
 class Mdata(PydanticBaseModel):
-    data: (
-        BashCommand
-        | WriteIfEmpty
-        | FileEdit
-        | str
-        | ReadFiles
-        | Initialize
-        | ContextSave
-    )
+    data: BashCommand | FileWriteOrEdit | str | ReadFiles | Initialize | ContextSave

{wcgw-3.0.6.dist-info → wcgw-4.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 3.0.6
+Version: 4.0.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>
@@ -17,7 +17,6 @@ Requires-Dist: pyte>=0.8.2
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: rich>=13.8.1
 Requires-Dist: semantic-version>=2.10.0
-Requires-Dist: shell>=1.0.1
 Requires-Dist: syntax-checker>=0.3.0
 Requires-Dist: tokenizers>=0.21.0
 Requires-Dist: toml>=0.10.2
@@ -30,8 +29,8 @@ Description-Content-Type: text/markdown
 
 Empowering chat applications to code, build and run on your local machine.
 
-- Claude - An MCP server on claude desktop for autonomous shell and coding agent. (mac only)
-- Chatgpt - Allows custom gpt to talk to your shell via a relay server. (linux or mac)
+- Claude - An MCP server on claude desktop for autonomous shell and coding agent. (mac, linux, windows on wsl)
+- Chatgpt - Allows custom gpt to talk to your shell via a relay server. (linux, mac, windows on wsl)
 
 ⚠️ Warning: do not allow BashCommand tool without reviewing the command, it may result in data loss.
 
@@ -39,7 +38,11 @@ Empowering chat applications to code, build and run on your local machine.
 [![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)
 [![codecov](https://codecov.io/gh/rusiaaman/wcgw/graph/badge.svg)](https://codecov.io/gh/rusiaaman/wcgw)
-[![smithery badge](https://smithery.ai/badge/wcgw)](https://smithery.ai/server/wcgw)
+[![Reddit](https://img.shields.io/badge/Reddit-r%2Fwcgw_mcp-red)](https://www.reddit.com/r/wcgw_mcp/)
+
+## Demo
+
+![Workflow Demo](static/workflow-demo.gif)
 
 ## Updates
 
@@ -53,7 +56,6 @@ Empowering chat applications to code, build and run on your local machine.
 
 - [9 Dec 2024] [Vscode extension to paste context on Claude app](https://marketplace.visualstudio.com/items?itemName=AmanRusia.wcgw)
 
-
 ## 🚀 Highlights
 
 - ⚡ **Create, Execute, Iterate**: Ask claude to keep running compiler checks till all errors are fixed, or ask it to keep checking for the status of a long running command till it's done.
@@ -96,6 +98,8 @@ Empowering chat applications to code, build and run on your local machine.
 
 ## Claude setup (using mcp)
 
+### Mac and linux
+
 First install `uv` using homebrew `brew install uv`
 
 (**Important:** use homebrew to install uv. Otherwise make sure `uv` is present in a global location like /usr/bin/)
@@ -131,18 +135,33 @@ _If there's an error in setting up_
 - Try using `uv` version `0.6.0` for which this tool was tested.
 - Debug the mcp server using `npx @modelcontextprotocol/inspector@0.1.7 uv tool run --from wcgw@latest --python 3.12 wcgw_mcp`
 
-### Alternative configuration using smithery (npx required)
+### Windows on wsl
 
-You need to first install uv using homebrew. `brew install uv`
+This mcp server works only on wsl on windows.
 
-Then to configure wcgw for Claude Desktop automatically via [Smithery](https://smithery.ai/server/wcgw):
+To set it up, [install uv](https://docs.astral.sh/uv/getting-started/installation/)
 
-```bash
-npx -y @smithery/cli install wcgw --client claude
-```
+Then add or update the claude config file `%APPDATA%\Claude\claude_desktop_config.json` with the following
 
-_If there's an error in setting up_
-- Try removing ~/.cache/uv folder
+```json
+{
+  "mcpServers": {
+    "wcgw": {
+      "command": "wsl.exe",
+      "args": [
+        "uv",
+        "tool",
+        "run",
+        "--from",
+        "wcgw@latest",
+        "--python",
+        "3.12",
+        "wcgw_mcp"
+      ]
+    }
+  }
+}
+```
 
 ### Usage
 
@@ -174,6 +193,7 @@ There are three built-in modes. You may ask Claude to run in one of the modes, l
 Note: in code-writer mode either all commands are allowed or none are allowed for now. If you give a list of allowed commands, Claude is instructed to run only those commands, but no actual check happens. (WIP)
 
 #### Attach to the working terminal to investigate
+
 If you've `screen` command installed, wcgw runs on a screen instance automatically. If you've started wcgw mcp server, you can list the screen sessions:
 
 `screen -ls`
@@ -184,7 +204,7 @@ You can then attach to the session using `screen -x 93358.wcgw.235521`
 
 You may interrupt any running command safely.
 
-You can interact with the terminal but beware that the AI might be running in parallel and it may conflict with what you're doing. It's recommended to keep your interactions to minimum. 
+You can interact with the terminal but beware that the AI might be running in parallel and it may conflict with what you're doing. It's recommended to keep your interactions to minimum.
 
 You shouldn't exit the session using `exit `or Ctrl-d, instead you should use `ctrl+a+d` to safely detach without destroying the screen session.
 
@@ -204,6 +224,30 @@ Read here: https://github.com/rusiaaman/wcgw/blob/main/openai.md
 
 ![example](https://github.com/rusiaaman/wcgw/blob/main/static/example.jpg?raw=true)
 
+## Using mcp server over docker
+
+First build the docker image `docker build -t wcgw https://github.com/rusiaaman/wcgw.git`
+
+Then you can update `/Users/username/Library/Application Support/Claude/claude_desktop_config.json` to have
+
+```
+{
+  "mcpServers": {
+    "filesystem": {
+      "command": "docker",
+      "args": [
+        "run",
+        "-i",
+        "--rm",
+        "--mount",
+        "type=bind,src=/Users/username/Desktop,dst=/workspace/Desktop",
+        "wcgw",
+      ]
+    }
+  }
+}
+```
+
 ## [Optional] Local shell access with openai API key or anthropic API key
 
 ### Openai

{wcgw-3.0.6.dist-info → wcgw-4.0.0.dist-info}/RECORD

@@ -1,33 +1,34 @@
 wcgw/__init__.py,sha256=qUofQOAXCGcWr2u_B8U-MIMhhYaBUpUwNDcscvRmYfo,90
 wcgw/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wcgw/types_.py,sha256=xd3TP9_57iFVoKxkBsl3G7mwIW5uIog68xJ_Khh9E9s,3639
+wcgw/types_.py,sha256=mwr2x20yLF85OoHy3pKxtmG834Gnoqrkx3pA8skGsVk,7360
 wcgw/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wcgw/client/common.py,sha256=OCH7Tx64jojz3M3iONUrGMadE07W21DiZs5sOxWX1Qc,1456
-wcgw/client/diff-instructions.txt,sha256=tmJ9Fu9XdO_72lYXQQNY9RZyx91bjxrXJf9d_KBz57k,1611
-wcgw/client/memory.py,sha256=8LdYsOhvCOoC1kfvDr85kNy07WnhPMvE6B2FRM2w85Y,2902
-wcgw/client/modes.py,sha256=FjIQOjT5oI7dk9VG0oRemN1A6EHBvHnuSQGKuRbguJI,10352
-wcgw/client/tool_prompts.py,sha256=H36Sr3yH2YuHZTc08Y6rTUWlYWjKMFtIGP6vbAVNwZo,3988
-wcgw/client/tools.py,sha256=l0L7kuylDdZ5WA2_P1CfN4Ds0xD9IPqEZ0ot0pruhKs,26512
-wcgw/client/bash_state/bash_state.py,sha256=aY_hW9y7TyPWHR5799v6rxzuWb1VzgjiGjA4K0cpGjE,27287
+wcgw/client/diff-instructions.txt,sha256=HXYfGvhlDMxmiIX9AbB05wJcptJF_gSIobYhYSqWRJo,1685
+wcgw/client/memory.py,sha256=M0plOGE5WXTEAs7nVLg4eCpVhmSW94ckpg5D0ycWX5I,2927
+wcgw/client/modes.py,sha256=Opmy5nKrCA4YITn79G1Q8l55xeBmhwtSv6MasSGkp5U,10338
+wcgw/client/tool_prompts.py,sha256=H2jzhnRDv99R-VaYo38FtGh4DxAVJniqG5Wuj35JfHs,4287
+wcgw/client/tools.py,sha256=rsoEUVVqZkYCTK-qlsik4CuxsmANvNOoCXJgDHAAX1Y,43243
+wcgw/client/bash_state/bash_state.py,sha256=9lBqdnHStgKEOU_t1ACeHed0cwbO2npaLAiEerFj1Es,33874
 wcgw/client/encoder/__init__.py,sha256=Y-8f43I6gMssUCWpX5rLYiAFv3D-JPRs4uNEejPlke8,1514
-wcgw/client/file_ops/diff_edit.py,sha256=sIwXSSkWYff_Dp3oHfefqtSuFqLrxbhKvzkCoLuLqDE,18679
-wcgw/client/file_ops/search_replace.py,sha256=Napa7IWaYPGMNdttunKyRDkb90elZE7r23B_o_htRxo,5585
+wcgw/client/file_ops/diff_edit.py,sha256=vKELPknWu3FGgPos74yN9wox6BCVcaocZ7dpBAT8Blk,18723
+wcgw/client/file_ops/search_replace.py,sha256=2biF6b_l9JRWXJRkeRIo_ltUVBKgsQgPuDd_sqkPMX8,5811
 wcgw/client/mcp_server/Readme.md,sha256=2Z88jj1mf9daYGW1CWaldcJ0moy8owDumhR2glBY3A8,109
 wcgw/client/mcp_server/__init__.py,sha256=mm7xhBIPwJpRT3u-Qsj4cKVMpVyucJoKRlbMP_gRRB0,343
 wcgw/client/mcp_server/server.py,sha256=ayK6qbzCveoQW7RO80m10cAIS3m-hvxzd15XhjiyxmE,5055
-wcgw/client/repo_ops/display_tree.py,sha256=E5q9mrGBb57NyvudSmRIG-fj4FUqupbzjmARpX8X0XY,4166
+wcgw/client/repo_ops/display_tree.py,sha256=uOGX2IbXTKXwtXT2wdDszuH4ODmSYsHm0toU55e1vYI,4021
+wcgw/client/repo_ops/file_stats.py,sha256=AUA0Br7zFRpylWFYZPGMeGPJy3nWp9e2haKi34JptHE,4887
 wcgw/client/repo_ops/path_prob.py,sha256=SWf0CDn37rtlsYRQ51ufSxay-heaQoVIhr1alB9tZ4M,2144
 wcgw/client/repo_ops/paths_model.vocab,sha256=M1pXycYDQehMXtpp-qAgU7rtzeBbCOiJo4qcYFY0kqk,315087
 wcgw/client/repo_ops/paths_tokens.model,sha256=jiwwE4ae8ADKuTZISutXuM5Wfyc_FBmN5rxTjoNnCos,1569052
-wcgw/client/repo_ops/repo_context.py,sha256=5NqRxBY0K-SBFXJ0Ybt7llzYOBD8pRkTpruMMJHWxv4,4336
+wcgw/client/repo_ops/repo_context.py,sha256=vB3ET5RLA7p9OpL3XNPr3nycQkoeu6gC-RbyA9Ml-xA,8871
 wcgw/relay/client.py,sha256=BUeEKUsWts8RpYxXwXcyFyjBJhOCS-CxThAlL_-VCOI,3618
-wcgw/relay/serve.py,sha256=xFAFiRENPae723_kwaaCgBmqL5sffYHKPu3esrMu57M,7896
+wcgw/relay/serve.py,sha256=vaHxSm4DkWUKLMOnz2cO6ClR2udnaXCWAGl0O_bXvrs,6984
 wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
 wcgw_cli/__init__.py,sha256=TNxXsTPgb52OhakIda9wTRh91cqoBqgQRx5TxjzQQFU,21
 wcgw_cli/__main__.py,sha256=wcCrL4PjG51r5wVKqJhcoJPTLfHW0wNbD31DrUN0MWI,28
-wcgw_cli/anthropic_client.py,sha256=-rfnw_tIx1VjhzOsPeTZkoF0VAXhRC9zBW2oLMFJcxg,19624
+wcgw_cli/anthropic_client.py,sha256=R95ht8ct3TgV8PMNzr5yJdAXAmZpozPpyZZ01J9bQHs,19627
 wcgw_cli/cli.py,sha256=-7FBe_lahKyUOhf65iurTA1M1gXXXAiT0OVKQVcZKKo,948
-wcgw_cli/openai_client.py,sha256=4NFv-d7u6B7vMD_w3Qqr-564ZcdvfBUGowUmQ6idglE,15758
+wcgw_cli/openai_client.py,sha256=MnJ93E7vSIRx1SkkanvCTcfc4ZsndlPUs8c-LXdxbOk,15761
 wcgw_cli/openai_utils.py,sha256=xGOb3W5ALrIozV7oszfGYztpj0FnXdD7jAxm5lEIVKY,2439
 mcp_wcgw/__init__.py,sha256=fKCgOdN7cn7gR3YGFaGyV5Goe8A2sEyllLcsRkN0i-g,2601
 mcp_wcgw/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -51,8 +52,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-3.0.6.dist-info/METADATA,sha256=tEFOApXxIWcCkeryPchiGvd0j_zhWFP1_JX2Aqj3CuU,14049
-wcgw-3.0.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-wcgw-3.0.6.dist-info/entry_points.txt,sha256=vd3tj1_Kzfp55LscJ8-6WFMM5hm9cWTfNGFCrWBnH3Q,124
-wcgw-3.0.6.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
-wcgw-3.0.6.dist-info/RECORD,,
+wcgw-4.0.0.dist-info/METADATA,sha256=lep9tKuEznEoFQBQ7VpRAOrhUlFnGcPu8Q4KXIE2QQo,14785
+wcgw-4.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+wcgw-4.0.0.dist-info/entry_points.txt,sha256=vd3tj1_Kzfp55LscJ8-6WFMM5hm9cWTfNGFCrWBnH3Q,124
+wcgw-4.0.0.dist-info/licenses/LICENSE,sha256=BvY8xqjOfc3X2qZpGpX3MZEmF-4Dp0LqgKBbT6L_8oI,11142
+wcgw-4.0.0.dist-info/RECORD,,

wcgw_cli/anthropic_client.py

@@ -217,7 +217,7 @@ def loop(
         system_console, os.getcwd(), None, None, None, None, True, None
     ) as bash_state:
         context = Context(bash_state, system_console)
-        system, context = initialize(
+        system, context, _ = initialize(
             "first_call",
             context,
             os.getcwd(),

wcgw_cli/openai_client.py

@@ -181,7 +181,7 @@ def loop(
         system_console, os.getcwd(), None, None, None, None, True, None
     ) as bash_state:
         context = Context(bash_state, system_console)
-        system, context = initialize(
+        system, context, _ = initialize(
             "first_call",
             context,
             os.getcwd(),