wcgw 2.7.0__tar.gz → 2.7.1__tar.gz

{wcgw-2.7.0 → wcgw-2.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 2.7.0
+Version: 2.7.1
 Summary: Shell and coding agent on claude and chatgpt
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
@@ -33,7 +33,7 @@ Description-Content-Type: text/markdown
 - 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"
+⚠️ Warning: do not allow BashCommand tool without reviewing the command, it may result in data loss.
 
 [![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)
@@ -55,7 +55,18 @@ Description-Content-Type: text/markdown
 - ⚡ **Large file edit**: Supports large file incremental edits to avoid token limit issues. Faster than full file write.
 - ⚡ **Syntax checking on edits**: Reports feedback to the LLM if its edits have any syntax errors, so that it can redo it.
 - ⚡ **Interactive Command Handling**: Supports interactive commands using arrow keys, interrupt, and ansi escape sequences.
-- ⚡ **Full Shell Access**: No restrictions, complete control.
+- ⚡ **File protections**: 
+  - The AI needs to read a file at least once before it's allowed to edit or rewrite it. This avoids accidental overwrites. 
+  - Avoids context filling up while reading very large files. Files get chunked based on token length. 
+  - On initialisation the provided workspace's directory structure is returned after selecting important files (based on .gitignore as well as a statistical approach)
+  - File edit based on search-replace tries to find correct search block if it has multiple matches based on previous search blocks. Fails otherwise (for correctness).
+  - File edit has spacing tolerant matching, with warning on issues like indentation mismatch. If there's no match, the closest match is returned to the AI to fix its mistakes.
+  - Using Aider-like search and replace, which has better performance than tool call based search and replace.
+- ⚡ **Shell optimisations**: 
+  - Only one command is allowed to be run at a time, simplifying management and avoiding rogue processes. There's only single shell instance at any point of time.
+  - Current working directory is always returned after any shell command to prevent AI from getting lost. 
+  - Command polling exits after a quick timeout to avoid slow feedback. However, status checking has wait tolerance based on fresh output streaming from a command. Both of these approach combined provides a good shell interaction experience.
+   
 
 ## Top use cases examples
 

{wcgw-2.7.0 → wcgw-2.7.1}/README.md

@@ -4,7 +4,7 @@
 - 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"
+⚠️ Warning: do not allow BashCommand tool without reviewing the command, it may result in data loss.
 
 [![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)
@@ -26,7 +26,18 @@
 - ⚡ **Large file edit**: Supports large file incremental edits to avoid token limit issues. Faster than full file write.
 - ⚡ **Syntax checking on edits**: Reports feedback to the LLM if its edits have any syntax errors, so that it can redo it.
 - ⚡ **Interactive Command Handling**: Supports interactive commands using arrow keys, interrupt, and ansi escape sequences.
-- ⚡ **Full Shell Access**: No restrictions, complete control.
+- ⚡ **File protections**: 
+  - The AI needs to read a file at least once before it's allowed to edit or rewrite it. This avoids accidental overwrites. 
+  - Avoids context filling up while reading very large files. Files get chunked based on token length. 
+  - On initialisation the provided workspace's directory structure is returned after selecting important files (based on .gitignore as well as a statistical approach)
+  - File edit based on search-replace tries to find correct search block if it has multiple matches based on previous search blocks. Fails otherwise (for correctness).
+  - File edit has spacing tolerant matching, with warning on issues like indentation mismatch. If there's no match, the closest match is returned to the AI to fix its mistakes.
+  - Using Aider-like search and replace, which has better performance than tool call based search and replace.
+- ⚡ **Shell optimisations**: 
+  - Only one command is allowed to be run at a time, simplifying management and avoiding rogue processes. There's only single shell instance at any point of time.
+  - Current working directory is always returned after any shell command to prevent AI from getting lost. 
+  - Command polling exits after a quick timeout to avoid slow feedback. However, status checking has wait tolerance based on fresh output streaming from a command. Both of these approach combined provides a good shell interaction experience.
+   
 
 ## Top use cases examples
 

{wcgw-2.7.0 → wcgw-2.7.1}/gpt_action_json_schema.json

@@ -290,16 +290,16 @@
         }
       }
     },
-    "/v1/knowledge_transfer": {
+    "/v1/context_save": {
       "post": {
         "x-openai-isConsequential": false,
-        "summary": "Knowledge Transfer",
-        "operationId": "knowledge_transfer_v1_knowledge_transfer_post",
+        "summary": "Context Save",
+        "operationId": "context_save_v1_context_save_post",
         "requestBody": {
           "content": {
             "application/json": {
               "schema": {
-                "$ref": "#/components/schemas/KTWithUUID"
+                "$ref": "#/components/schemas/ContextSaveWithUUID"
               }
             }
           },
@@ -312,7 +312,7 @@
               "application/json": {
                 "schema": {
                   "type": "string",
-                  "title": "Response Knowledge Transfer V1 Knowledge Transfer Post"
+                  "title": "Response Context Save V1 Context Save Post"
                 }
               }
             }
@@ -427,6 +427,44 @@
         ],
         "title": "CommandWithUUID"
       },
+      "ContextSaveWithUUID": {
+        "properties": {
+          "id": {
+            "type": "string",
+            "title": "Id"
+          },
+          "project_root_path": {
+            "type": "string",
+            "title": "Project Root Path"
+          },
+          "description": {
+            "type": "string",
+            "title": "Description"
+          },
+          "relevant_file_globs": {
+            "items": {
+              "type": "string"
+            },
+            "type": "array",
+            "title": "Relevant File Globs"
+          },
+          "user_id": {
+            "type": "string",
+            "format": "uuid",
+            "title": "User Id"
+          }
+        },
+        "additionalProperties": false,
+        "type": "object",
+        "required": [
+          "id",
+          "project_root_path",
+          "description",
+          "relevant_file_globs",
+          "user_id"
+        ],
+        "title": "ContextSaveWithUUID"
+      },
       "FileEditWithUUID": {
         "properties": {
           "file_path": {
@@ -498,64 +536,6 @@
         ],
         "title": "InitializeWithUUID"
       },
-      "KTWithUUID": {
-        "properties": {
-          "id": {
-            "type": "string",
-            "title": "Id"
-          },
-          "project_root_path": {
-            "type": "string",
-            "title": "Project Root Path"
-          },
-          "objective": {
-            "type": "string",
-            "title": "Objective"
-          },
-          "all_user_instructions": {
-            "type": "string",
-            "title": "All User Instructions"
-          },
-          "current_status_of_the_task": {
-            "type": "string",
-            "title": "Current Status Of The Task"
-          },
-          "all_issues_snippets": {
-            "type": "string",
-            "title": "All Issues Snippets"
-          },
-          "relevant_file_paths": {
-            "items": {
-              "type": "string"
-            },
-            "type": "array",
-            "title": "Relevant File Paths"
-          },
-          "build_and_development_instructions": {
-            "type": "string",
-            "title": "Build And Development Instructions"
-          },
-          "user_id": {
-            "type": "string",
-            "format": "uuid",
-            "title": "User Id"
-          }
-        },
-        "additionalProperties": false,
-        "type": "object",
-        "required": [
-          "id",
-          "project_root_path",
-          "objective",
-          "all_user_instructions",
-          "current_status_of_the_task",
-          "all_issues_snippets",
-          "relevant_file_paths",
-          "build_and_development_instructions",
-          "user_id"
-        ],
-        "title": "KTWithUUID"
-      },
       "ReadFileWithUUID": {
         "properties": {
           "file_paths": {

{wcgw-2.7.0 → wcgw-2.7.1}/gpt_instructions.txt

@@ -44,6 +44,11 @@ Instructions for `BashInteraction`
 Instructions for `ResetShell`
 - Resets the shell. Use only if all interrupts and prompt reset attempts have failed repeatedly.
 
+Instructions for `ContextSave`
+- Saves provided description and file contents of all the relevant file paths or globs in a single text file.
+  - Provide random unqiue id or whatever user provided.
+  - Leave project path as empty string if no project path
+
 Instructions for `FileEdit`:
     - Use absolute file path only.
     - Use SEARCH/REPLACE blocks to edit the file.

{wcgw-2.7.0 → wcgw-2.7.1}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "gapypi@arcfu.com" }]
 name = "wcgw"
-version = "2.7.0"
+version = "2.7.1"
 description = "Shell and coding agent on claude and chatgpt"
 readme = "README.md"
 requires-python = ">=3.11, <3.13"

{wcgw-2.7.0 → wcgw-2.7.1}/src/wcgw/client/anthropic_client.py

@@ -25,10 +25,10 @@ from typer import Typer
 from ..types_ import (
     BashCommand,
     BashInteraction,
+    ContextSave,
     FileEdit,
     GetScreenInfo,
     Keyboard,
-    KnowledgeTransfer,
     Mouse,
     ReadFiles,
     ReadImage,
@@ -130,7 +130,12 @@ def loop(
     memory = None
     if resume:
         try:
-            memory = load_memory(resume)
+            _, memory = load_memory(
+                resume,
+                8000,
+                lambda x: default_enc.encode(x).ids,
+                lambda x: default_enc.decode(x),
+            )
         except OSError:
             if resume == "latest":
                 resume_path = sorted(Path(".wcgw").iterdir(), key=os.path.getmtime)[-1]
@@ -215,22 +220,12 @@ def loop(
 """,
         ),
         ToolParam(
-            input_schema=KnowledgeTransfer.model_json_schema(),
-            name="KnowledgeTransfer",
+            input_schema=ContextSave.model_json_schema(),
+            name="ContextSave",
             description="""
-Write detailed description in order to do a KT, if the user asks for it.
-Save all information necessary for a person to understand the task and the problems.
-
-- `all_user_instructions` should contain all instructions user shared in the conversation.
-- `current_status_of_the_task` should contain only what is already achieved, not what's remaining.
-- `all_issues_snippets` should only contain snippets of error, traceback, file snippets, commands, etc., no comments or solutions (important!).
-- Be very verbose in `all_issues_snippets` providing as much error context as possible.
-- Provide an id if the user hasn't provided one. 
-- This tool will return a text file path where the information is saved.
-- After the tool completes succesfully, tell the user the task id and the generate file path. (important!)
-- Leave arguments as empty string if they aren't relevant.
-- This tool marks end of your conversation, do not run any further tools after calling this.
-- Provide absolute file paths only in `relevant_file_paths` containing all relevant files.
+Saves provided description and file contents of all the relevant file paths or globs in a single text file.
+- Provide random unqiue id or whatever user provided.
+- Leave project path as empty string if no project path
 """,
         ),
     ]

{wcgw-2.7.0 → wcgw-2.7.1}/src/wcgw/client/mcp_server/server.py

@@ -14,6 +14,7 @@ from mcp_wcgw.types import Tool as ToolParam
 from ...types_ import (
     BashCommand,
     BashInteraction,
+    ContextSave,
     FileEdit,
     GetScreenInfo,
     Initialize,
@@ -46,14 +47,47 @@ async def handle_read_resource(uri: AnyUrl) -> str:
 
 @server.list_prompts()  # type: ignore
 async def handle_list_prompts() -> list[types.Prompt]:
-    return []
+    return [
+        types.Prompt(
+            name="KnowledgeTransfer",
+            description="Prompt for invoking ContextSave tool in order to do a comprehensive knowledge transfer of a coding task. Prompts to save detailed error log and instructions.",
+        )
+    ]
 
 
 @server.get_prompt()  # type: ignore
 async def handle_get_prompt(
     name: str, arguments: dict[str, str] | None
 ) -> types.GetPromptResult:
-    return types.GetPromptResult(messages=[])
+    messages = []
+    if name == "KnowledgeTransfer":
+        messages = [
+            types.PromptMessage(
+                role="user",
+                content=types.TextContent(
+                    type="text",
+                    text="""Use `ContextSave` tool to do a knowledge transfer of the task in hand.
+Write detailed description in order to do a KT.
+Save all information necessary for a person to understand the task and the problems.
+
+Format the `description` field using Markdown with the following sections.
+- "# Objective" section containing project and task objective.
+- "# All user instructions" section should be provided containing all instructions user shared in the conversation.
+- "# Current status of the task" should be provided containing only what is already achieved, not what's remaining.
+- "# All issues with snippets" section containing snippets of error, traceback, file snippets, commands, etc. But no comments or solutions.
+- Be very verbose in the all issues with snippets section providing as much error context as possible.
+- "# Build and development instructions" section containing instructions to build or run project or run tests, or envrionment related information. Only include what's known. Leave empty if unknown.
+- After the tool completes succesfully, tell me the task id and the file path the tool generated (important!)
+- This tool marks end of your conversation, do not run any further tools after calling this.
+
+Provide all relevant file paths in order to understand and solve the the task. Err towards providing more file paths than fewer.
+
+(Note to self: this conversation can then be resumed later asking "Resume `<generated id>`" which should call Initialize tool)
+""",
+                ),
+            )
+        ]
+    return types.GetPromptResult(messages=messages)
 
 
 @server.list_tools()  # type: ignore
@@ -153,24 +187,14 @@ async def handle_list_tools() -> list[types.Tool]:
 """
             + diffinstructions,
         ),
-        #         ToolParam(
-        #             inputSchema=KnowledgeTransfer.model_json_schema(),
-        #             name="KnowledgeTransfer",
-        #             description="""
-        # Write detailed description in order to do a KT, if the user asks for it.
-        # Save all information necessary for a person to understand the task and the problems.
-        # - `all_user_instructions` should contain all instructions user shared in the conversation.
-        # - `current_status_of_the_task` should contain only what is already achieved, not what's remaining.
-        # - `all_issues_snippets` should only contain snippets of error, traceback, file snippets, commands, etc., no comments or solutions (important!).
-        # - Be very verbose in `all_issues_snippets` providing as much error context as possible.
-        # - Provide an id if the user hasn't provided one.
-        # - This tool will return a text file path where the information is saved.
-        # - After the tool completes succesfully, tell the user the task id and the generate file path. (important!)
-        # - Leave arguments as empty string if they aren't relevant.
-        # - This tool marks end of your conversation, do not run any further tools after calling this.
-        # - Provide absolute file paths only in `relevant_file_paths` containing all relevant files.
-        # """,
-        #         ),
+        ToolParam(
+            inputSchema=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.
+- Provide random unqiue id or whatever user provided.
+- Leave project path as empty string if no project path""",
+        ),
     ]
     if COMPUTER_USE_ON_DOCKER_ENABLED:
         tools += [

wcgw-2.7.1/src/wcgw/client/memory.py

@@ -0,0 +1,78 @@
+import os
+import re
+import shlex
+from typing import Callable, Optional
+
+from ..types_ import ContextSave
+
+
+def get_app_dir_xdg() -> str:
+    xdg_data_dir = os.environ.get("XDG_DATA_HOME", os.path.expanduser("~/.local/share"))
+    return os.path.join(xdg_data_dir, "wcgw")
+
+
+def format_memory(task_memory: ContextSave, relevant_files: str) -> str:
+    memory_data = ""
+    if task_memory.project_root_path:
+        memory_data += (
+            f"# PROJECT ROOT = {shlex.quote(task_memory.project_root_path)}\n"
+        )
+    memory_data += task_memory.description
+
+    memory_data += (
+        "\n\n"
+        + "# Relevant file paths\n"
+        + ", ".join(map(shlex.quote, task_memory.relevant_file_globs))
+    )
+
+    memory_data += "\n\n# Relevant Files:\n" + relevant_files
+
+    return memory_data
+
+
+def save_memory(task_memory: ContextSave, relevant_files: str) -> str:
+    app_dir = get_app_dir_xdg()
+    memory_dir = os.path.join(app_dir, "memory")
+    os.makedirs(memory_dir, exist_ok=True)
+
+    task_id = task_memory.id
+    if not task_id:
+        raise Exception("Task id can not be empty")
+    memory_data = format_memory(task_memory, relevant_files)
+
+    memory_file_full = os.path.join(memory_dir, f"{task_id}.txt")
+
+    with open(memory_file_full, "w") as f:
+        f.write(memory_data)
+
+    return memory_file_full
+
+
+def load_memory[T](
+    task_id: str,
+    max_tokens: Optional[int],
+    encoder: Callable[[str], list[T]],
+    decoder: Callable[[list[T]], str],
+) -> tuple[str, str]:
+    app_dir = get_app_dir_xdg()
+    memory_dir = os.path.join(app_dir, "memory")
+    memory_file = os.path.join(memory_dir, f"{task_id}.txt")
+
+    with open(memory_file, "r") as f:
+        data = f.read()
+
+    if max_tokens:
+        toks = encoder(data)
+        if len(toks) > max_tokens:
+            toks = toks[: max(0, max_tokens - 10)]
+            data = decoder(toks)
+            data += "\n(... truncated)"
+
+    project_root_match = re.search(r"# PROJECT ROOT = \s*(.*?)\s*$", data, re.MULTILINE)
+    project_root_path = ""
+    if project_root_match:
+        matched_path = project_root_match.group(1)
+        parsed_ = shlex.split(matched_path)
+        if parsed_ and len(parsed_) == 1:
+            project_root_path = parsed_[0]
+    return project_root_path, data

{wcgw-2.7.0 → wcgw-2.7.1}/src/wcgw/client/openai_client.py

@@ -26,8 +26,8 @@ from typer import Typer
 from ..types_ import (
     BashCommand,
     BashInteraction,
+    ContextSave,
     FileEdit,
-    KnowledgeTransfer,
     ReadFiles,
     ReadImage,
     ResetShell,
@@ -39,6 +39,7 @@ from .openai_utils import get_input_cost, get_output_cost
 from .tools import (
     DoneFlag,
     ImageData,
+    default_enc,
     get_tool_output,
     initialize,
     which_tool,
@@ -124,7 +125,12 @@ def loop(
     memory = None
     if resume:
         try:
-            memory = load_memory(resume)
+            _, memory = load_memory(
+                resume,
+                8000,
+                lambda x: default_enc.encode(x).ids,
+                lambda x: default_enc.decode(x),
+            )
         except OSError:
             if resume == "latest":
                 resume_path = sorted(Path(".wcgw").iterdir(), key=os.path.getmtime)[-1]
@@ -211,22 +217,12 @@ def loop(
             description="Resets the shell. Use only if all interrupts and prompt reset attempts have failed repeatedly.",
         ),
         openai.pydantic_function_tool(
-            KnowledgeTransfer,
+            ContextSave,
             description="""
-Write detailed description in order to do a KT, if the user asks for it.
-Save all information necessary for a person to understand the task and the problems.
-
-- `all_user_instructions` should contain all instructions user shared in the conversation.
-- `current_status_of_the_task` should contain only what is already achieved, not what's remaining.
-- `all_issues_snippets` should only contain snippets of error, traceback, file snippets, commands, etc., no comments or solutions (important!).
-- Be very verbose in `all_issues_snippets` providing as much error context as possible.
-- Provide an id if the user hasn't provided one. 
-- This tool will return a text file path where the information is saved.
-- After the tool completes succesfully, tell the user the task id and the generate file path. (important!)
-- Leave arguments as empty string if they aren't relevant.
-- This tool marks end of your conversation, do not run any further tools after calling this.
-- Provide absolute file paths only in `relevant_file_paths` containing all relevant files.
-""",
+
+Saves provided description and file contents of all the relevant file paths or globs in a single text file.
+- Provide random unqiue id or whatever user provided.
+- Leave project path as empty string if no project path""",
         ),
     ]
 

{wcgw-2.7.0 → wcgw-2.7.1}/src/wcgw/client/tools.py

@@ -1,5 +1,6 @@
 import base64
 import datetime
+import glob
 import importlib.metadata
 import json
 import mimetypes
@@ -38,12 +39,12 @@ from websockets.sync.client import connect as syncconnect
 from ..types_ import (
     BashCommand,
     BashInteraction,
+    ContextSave,
     FileEdit,
     FileEditFindReplace,
     GetScreenInfo,
     Initialize,
     Keyboard,
-    KnowledgeTransfer,
     Mouse,
     ReadFiles,
     ReadImage,
@@ -53,7 +54,7 @@ from ..types_ import (
 )
 from .computer_use import run_computer_tool
 from .file_ops.search_replace import search_replace_edit
-from .memory import format_memory, load_memory, save_memory
+from .memory import load_memory, save_memory
 from .repo_ops.repo_context import get_repo_context
 from .sys_utils import command_run
 
@@ -307,16 +308,17 @@ def initialize(
     memory = ""
     if task_id_to_resume:
         try:
-            task_mem = load_memory(task_id_to_resume)
-            mem_files = task_mem.relevant_file_paths
-            mem_files_read = read_files(mem_files, max_tokens)
-            memory = "Following is the retrieved task:\n" + format_memory(
-                task_mem, mem_files_read
+            project_root_path, task_mem = load_memory(
+                task_id_to_resume,
+                max_tokens,
+                lambda x: default_enc.encode(x).ids,
+                lambda x: default_enc.decode(x),
             )
+            memory = "Following is the retrieved task:\n" + task_mem
             if (
                 not any_workspace_path or not os.path.exists(any_workspace_path)
-            ) and os.path.exists(task_mem.project_root_path):
-                any_workspace_path = task_mem.project_root_path
+            ) and os.path.exists(project_root_path):
+                any_workspace_path = project_root_path
         except Exception:
             memory = f'Error: Unable to load task with ID "{task_id_to_resume}" '
 
@@ -1003,7 +1005,7 @@ TOOLS = (
     | Keyboard
     | ScreenShot
     | GetScreenInfo
-    | KnowledgeTransfer
+    | ContextSave
 )
 
 
@@ -1045,8 +1047,8 @@ def which_tool_name(name: str) -> Type[TOOLS]:
         return ScreenShot
     elif name == "GetScreenInfo":
         return GetScreenInfo
-    elif name == "KnowledgeTransfer":
-        return KnowledgeTransfer
+    elif name == "ContextSave":
+        return ContextSave
     else:
         raise ValueError(f"Unknown tool name: {name}")
 
@@ -1142,20 +1144,26 @@ def get_tool_output(
                     )
                 BASH_STATE.set_in_docker(arg.docker_image_id)
         return outputs, outputs_cost[1]
-    elif isinstance(arg, KnowledgeTransfer):
+    elif isinstance(arg, ContextSave):
         console.print("Calling task memory tool")
-        relevant_files = arg.relevant_file_paths
-        for i, fpath in enumerate(relevant_files):
-            if not os.path.isabs(fpath):
-                relpath = os.path.join(arg.project_root_path, fpath)
-                if os.path.exists(relpath):
-                    relevant_files[i] = relpath
-                else:
-                    raise Exception(f"The file path {fpath} does not exist")
-            elif not os.path.exists(fpath):
-                raise Exception(f"The file path {fpath} does not exist")
-        relevant_files_data = read_files(relevant_files, None)
-        output = save_memory(arg, relevant_files_data), 0.0
+        assert not BASH_STATE.is_in_docker, "KT not supported in docker"
+        relevant_files = []
+        warnings = ""
+        for fglob in arg.relevant_file_globs:
+            fglob = expand_user(fglob, None)
+            if not os.path.isabs(fglob) and arg.project_root_path:
+                fglob = os.path.join(arg.project_root_path, fglob)
+            globs = glob.glob(fglob)
+            relevant_files.extend(globs[:1000])
+            if not globs:
+                warnings += f"Warning: No files found for the glob: {fglob}\n"
+        relevant_files_data = read_files(relevant_files[:10_000], None)
+        output_ = save_memory(arg, relevant_files_data)
+        if not relevant_files and arg.relevant_file_globs:
+            output_ = f'Error: No files found for the given globs. Context file successfully saved at "{output_}", but please fix the error.'
+        elif warnings:
+            output_ = warnings + "\nContext file successfully saved at " + output_
+        output = output_, 0.0
     else:
         raise ValueError(f"Unknown tool: {arg}")
     if isinstance(output[0], str):
@@ -1184,7 +1192,7 @@ class Mdata(BaseModel):
         | str
         | ReadFiles
         | Initialize
-        | KnowledgeTransfer
+        | ContextSave
     )