wcgw 2.6.2__tar.gz → 2.7.0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wcgw
-Version: 2.6.2
+Version: 2.7.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>

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

@@ -289,6 +289,46 @@
           }
         }
       }
+    },
+    "/v1/knowledge_transfer": {
+      "post": {
+        "x-openai-isConsequential": false,
+        "summary": "Knowledge Transfer",
+        "operationId": "knowledge_transfer_v1_knowledge_transfer_post",
+        "requestBody": {
+          "content": {
+            "application/json": {
+              "schema": {
+                "$ref": "#/components/schemas/KTWithUUID"
+              }
+            }
+          },
+          "required": true
+        },
+        "responses": {
+          "200": {
+            "description": "Successful Response",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "type": "string",
+                  "title": "Response Knowledge Transfer V1 Knowledge Transfer Post"
+                }
+              }
+            }
+          },
+          "422": {
+            "description": "Validation Error",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/HTTPValidationError"
+                }
+              }
+            }
+          }
+        }
+      }
     }
   },
   "components": {
@@ -438,6 +478,10 @@
             "type": "array",
             "title": "Initial Files To Read"
           },
+          "task_id_to_resume": {
+            "type": "string",
+            "title": "Task Id To Resume"
+          },
           "user_id": {
             "type": "string",
             "format": "uuid",
@@ -449,10 +493,69 @@
         "required": [
           "any_workspace_path",
           "initial_files_to_read",
+          "task_id_to_resume",
           "user_id"
         ],
         "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.6.2 → wcgw-2.7.0}/pyproject.toml

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

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

@@ -1,59 +1,51 @@
 import base64
 import json
 import mimetypes
-from pathlib import Path
-import sys
+import os
+import subprocess
+import tempfile
 import traceback
-from typing import Callable, DefaultDict, Optional, cast, Literal
-import anthropic
+import uuid
+from pathlib import Path
+from typing import Literal, Optional, cast
+
+import rich
 from anthropic import Anthropic
 from anthropic.types import (
-    ToolParam,
+    ImageBlockParam,
     MessageParam,
+    TextBlockParam,
+    ToolParam,
     ToolResultBlockParam,
     ToolUseBlockParam,
-    ImageBlockParam,
-    TextBlockParam,
 )
-
-import rich
-import petname  # type: ignore[import-untyped]
+from dotenv import load_dotenv
 from typer import Typer
-import uuid
 
 from ..types_ import (
     BashCommand,
     BashInteraction,
-    WriteIfEmpty,
-    FileEditFindReplace,
     FileEdit,
+    GetScreenInfo,
     Keyboard,
+    KnowledgeTransfer,
     Mouse,
     ReadFiles,
     ReadImage,
     ResetShell,
     ScreenShot,
-    GetScreenInfo,
+    WriteIfEmpty,
+)
+from .common import discard_input
+from .memory import load_memory
+from .tools import (
+    DoneFlag,
+    ImageData,
+    default_enc,
+    get_tool_output,
+    initialize,
+    which_tool_name,
 )
-
-from .common import Models, discard_input
-from .common import CostData
-from .tools import ImageData
-from .computer_use import Computer
-
-from .tools import DoneFlag, get_tool_output, which_tool_name, default_enc
-
-from urllib import parse
-import subprocess
-import os
-import tempfile
-
-import toml
-from pydantic import BaseModel
-
-
-from dotenv import load_dotenv
-
 
 History = list[MessageParam]
 
@@ -135,19 +127,23 @@ def loop(
 
     history: History = []
     waiting_for_assistant = False
+    memory = None
     if resume:
-        if resume == "latest":
-            resume_path = sorted(Path(".wcgw").iterdir(), key=os.path.getmtime)[-1]
-        else:
-            resume_path = Path(resume)
-        if not resume_path.exists():
-            raise FileNotFoundError(f"File {resume} not found")
-        with resume_path.open() as f:
-            history = json.load(f)
-        if len(history) <= 2:
-            raise ValueError("Invalid history file")
-        first_message = ""
-        waiting_for_assistant = history[-1]["role"] != "assistant"
+        try:
+            memory = load_memory(resume)
+        except OSError:
+            if resume == "latest":
+                resume_path = sorted(Path(".wcgw").iterdir(), key=os.path.getmtime)[-1]
+            else:
+                resume_path = Path(resume)
+            if not resume_path.exists():
+                raise FileNotFoundError(f"File {resume} not found")
+            with resume_path.open() as f:
+                history = json.load(f)
+            if len(history) <= 2:
+                raise ValueError("Invalid history file")
+            first_message = ""
+            waiting_for_assistant = history[-1]["role"] != "assistant"
 
     limit = 1
 
@@ -216,6 +212,25 @@ def loop(
 - Use absolute file path only.
 - Use SEARCH/REPLACE blocks to edit the file.
 - If the edit fails due to block not matching, please retry with correct block till it matches. Re-read the file to ensure you've all the lines correct.
+""",
+        ),
+        ToolParam(
+            input_schema=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.
 """,
         ),
     ]
@@ -270,9 +285,10 @@ def loop(
 """,
             ),
         ]
-    uname_sysname = os.uname().sysname
-    uname_machine = os.uname().machine
 
+    initial_info = initialize(
+        os.getcwd(), [], resume if (memory and resume) else "", 8000
+    )
     system = f"""
 You're an expert software engineer with shell and code knowledge.
 
@@ -284,10 +300,7 @@ Instructions:
     - Do not provide code snippets unless asked by the user, instead directly add/edit the code.
     - Do not install new tools/packages before ensuring no such tools/package or an alternative already exists.
 
-System information:
-    - System: {uname_sysname}
-    - Machine: {uname_machine}
-    - Current directory: {os.getcwd()}
+{initial_info}
 """
 
     with open(os.path.join(os.path.dirname(__file__), "diff-instructions.txt")) as f:

{wcgw-2.6.2 → wcgw-2.7.0}/src/wcgw/client/diff-instructions.txt

@@ -47,7 +47,6 @@ Every *SEARCH/REPLACE block* must use this format:
 
 Every "<<<<<<< SEARCH" section must *EXACTLY MATCH* the existing file content, character for character, including all comments, docstrings, whitespaces, etc.
 
-*SEARCH/REPLACE* blocks will *only* replace the first match occurrence.
 Including multiple unique *SEARCH/REPLACE* blocks if needed.
 Include enough lines in each SEARCH section to uniquely match each set of lines that need to change.