wcgw 1.0.1__py3-none-any.whl → 1.1.0__py3-none-any.whl

wcgw/client/claude.py

@@ -1,384 +0,0 @@
-import base64
-import json
-import mimetypes
-from pathlib import Path
-import sys
-import traceback
-from typing import Callable, DefaultDict, Optional, cast
-import rich
-import petname
-from typer import Typer
-import uuid
-
-from .common import Models, discard_input
-from .common import CostData, Config, text_from_editor
-from .tools import ExecuteBash, ReadImage, ImageData
-
-from .tools import (
-    BASH_CLF_OUTPUT,
-    Confirmation,
-    DoneFlag,
-    Writefile,
-    get_is_waiting_user_input,
-    get_tool_output,
-    SHELL,
-    start_shell,
-    which_tool,
-)
-import tiktoken
-
-from urllib import parse
-import subprocess
-import os
-import tempfile
-
-import toml
-from pydantic import BaseModel
-
-
-from dotenv import load_dotenv
-
-from anthropic.types import MessageParam
-
-History = list[MessageParam]
-
-def save_history(history: History, session_id: str) -> None:
-    myid = str(history[1]["content"]).replace("/", "_").replace(" ", "_").lower()[:60]
-    myid += "_" + session_id
-    myid = myid + ".json"
-
-    mypath = Path(".wcgw") / myid
-    mypath.parent.mkdir(parents=True, exist_ok=True)
-    with open(mypath, "w") as f:
-        json.dump(history, f, indent=3)
-
-
-def parse_user_message_special(msg: str) -> ChatCompletionUserMessageParam:
-    # Search for lines starting with `%` and treat them as special commands
-    parts: list[ChatCompletionContentPartParam] = []
-    for line in msg.split("\n"):
-        if line.startswith("%"):
-            args = line[1:].strip().split(" ")
-            command = args[0]
-            assert command == 'image'
-            image_path = args[1]
-            with open(image_path, 'rb') as f:
-                image_bytes = f.read()
-                image_b64 = base64.b64encode(image_bytes).decode("utf-8")
-                image_type = mimetypes.guess_type(image_path)[0]
-                dataurl=f'data:{image_type};base64,{image_b64}'
-            parts.append({
-                'type': 'image_url',
-                'image_url': {
-                    'url': dataurl,
-                    'detail': 'auto'
-                }
-            })
-        else:
-            if len(parts) > 0 and parts[-1]['type'] == 'text':
-                parts[-1]['text'] += '\n' + line
-            else:
-                parts.append({'type': 'text', 'text': line})
-    return {
-        'role': 'user',
-        'content': parts
-    }
-
-
-app = Typer(pretty_exceptions_show_locals=False)
-
-
-@app.command()
-def loop(
-    first_message: Optional[str] = None,
-    limit: Optional[float] = None,
-    resume: Optional[str] = None,
-) -> tuple[str, float]:
-    load_dotenv()
-
-    session_id = str(uuid.uuid4())[:6]
-
-    history: History = []
-    waiting_for_assistant = False
-    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")
-        if history[1]["role"] != "user":
-            raise ValueError("Invalid history file, second message should be user")
-        first_message = ""
-        waiting_for_assistant = history[-1]['role'] != 'assistant'
-
-    my_dir = os.path.dirname(__file__)
-    config_file = os.path.join(my_dir, "..", "..", "config.toml")
-    with open(config_file) as f:
-        config_json = toml.load(f)
-        config = Config.model_validate(config_json)
-
-    if limit is not None:
-        config.cost_limit = limit
-    limit = config.cost_limit
-
-    enc = tiktoken.encoding_for_model(
-        config.model if not config.model.startswith("o1") else "gpt-4o"
-    )
-    is_waiting_user_input = get_is_waiting_user_input(
-        config.model, config.cost_file[config.model]
-    )
-
-    tools = [
-        openai.pydantic_function_tool(
-            ExecuteBash,
-            description="""
-Execute a bash script. Stateful (beware with subsequent calls).
-Execute commands using `execute_command` attribute.
-Do not use interactive commands like nano. Prefer writing simpler commands.
-Last line will always be `(exit <int code>)` except if
-the last line is `(pending)` if the program is still running or waiting for user inputs. You can then send input using `send_ascii` attributes. You get status by sending `send_ascii: [10]`.
-Optionally the last line is `(won't exit)` in which case you need to kill the process if you want to run a new command.
-Optionally `exit shell has restarted` is the output, in which case environment resets, you can run fresh commands.
-The first line might be `(...truncated)` if the output is too long.""",
-        ),
-        openai.pydantic_function_tool(
-            Writefile,
-            description="Write content to a file. Provide file path and content. Use this instead of ExecuteBash for writing files.",
-        ),
-        openai.pydantic_function_tool(
-            ReadImage, description="Read an image from the shell."
-        ),
-    ]
-    uname_sysname = os.uname().sysname
-    uname_machine = os.uname().machine
-
-    system = f"""
-You're a cli assistant.
-
-Instructions:
-
-    - You should use the provided bash execution tool to run script to complete objective. 
-    - Do not use sudo. Do not use interactive commands. 
-    - Ask user for confirmation before running anything major
-    
-System information:
-    - System: {uname_sysname}
-    - Machine: {uname_machine}
-"""
-
-    if not history:
-        history = [{"role": "system", "content": system}]
-    else:
-        if history[-1]["role"] == "tool":
-            waiting_for_assistant = True
-
-    client = OpenAI()
-
-    cost: float = 0
-    input_toks = 0
-    output_toks = 0
-    system_console = rich.console.Console(style="blue", highlight=False)
-    error_console = rich.console.Console(style="red", highlight=False)
-    user_console = rich.console.Console(style="bright_black", highlight=False)
-    assistant_console = rich.console.Console(style="white bold", highlight=False)
-
-    while True:
-        if cost > limit:
-            system_console.print(
-                f"\nCost limit exceeded. Current cost: {cost}, input tokens: {input_toks}, output tokens: {output_toks}"
-            )
-            break
-
-        if not waiting_for_assistant:
-            if first_message:
-                msg = first_message
-                first_message = ""
-            else:
-                msg = text_from_editor(user_console)
-
-            history.append(parse_user_message_special(msg))
-        else:
-            waiting_for_assistant = False
-
-        cost_, input_toks_ = get_input_cost(
-            config.cost_file[config.model], enc, history
-        )
-        cost += cost_
-        input_toks += input_toks_
-
-        stream = client.chat.completions.create(
-            messages=history,
-            model=config.model,
-            stream=True,
-            tools=tools,
-        )
-
-        system_console.print(
-            "\n---------------------------------------\n# Assistant response",
-            style="bold",
-        )
-        tool_call_args_by_id = DefaultDict[str, DefaultDict[int, str]](
-            lambda: DefaultDict(str)
-        )
-        _histories: History = []
-        item: ChatCompletionMessageParam
-        full_response: str = ""
-        image_histories: History = []
-        try:
-            for chunk in stream:
-                if chunk.choices[0].finish_reason == "tool_calls":
-                    assert tool_call_args_by_id
-                    item = {
-                        "role": "assistant",
-                        "content": full_response,
-                        "tool_calls": [
-                            {
-                                "id": tool_call_id + str(toolindex),
-                                "type": "function",
-                                "function": {
-                                    "arguments": tool_args,
-                                    "name": type(which_tool(tool_args)).__name__,
-                                },
-                            }
-                            for tool_call_id, toolcallargs in tool_call_args_by_id.items()
-                            for toolindex, tool_args in toolcallargs.items()
-                        ],
-                    }
-                    cost_, output_toks_ = get_output_cost(
-                        config.cost_file[config.model], enc, item
-                    )
-                    cost += cost_
-                    system_console.print(
-                        f"\n---------------------------------------\n# Assistant invoked tools: {[which_tool(tool['function']['arguments']) for tool in item['tool_calls']]}"
-                    )
-                    system_console.print(f"\nTotal cost: {config.cost_unit}{cost:.3f}")
-                    output_toks += output_toks_
-                    
-                    _histories.append(item)
-                    for tool_call_id, toolcallargs in tool_call_args_by_id.items():
-                        for toolindex, tool_args in toolcallargs.items():
-                            try:
-                                output_or_done, cost_ = get_tool_output(
-                                    json.loads(tool_args),
-                                    enc,
-                                    limit - cost,
-                                    loop,
-                                    is_waiting_user_input,
-                                )
-                            except Exception as e:
-                                output_or_done = (
-                                    f"GOT EXCEPTION while calling tool. Error: {e}"
-                                )
-                                tb = traceback.format_exc()
-                                error_console.print(output_or_done + "\n" + tb)
-                                cost_ = 0
-                            cost += cost_
-                            system_console.print(
-                                f"\nTotal cost: {config.cost_unit}{cost:.3f}"
-                            )
-
-                            if isinstance(output_or_done, DoneFlag):
-                                system_console.print(
-                                    f"\n# Task marked done, with output {output_or_done.task_output}",
-                                )
-                                system_console.print(
-                                    f"\nTotal cost: {config.cost_unit}{cost:.3f}"
-                                )
-                                return output_or_done.task_output, cost
-                                
-                            output = output_or_done
-
-                            if isinstance(output, ImageData):
-                                randomId = petname.Generate(2, "-")
-                                if not image_histories:
-                                    image_histories.extend([
-                                        {
-                                            'role': 'assistant',
-                                            'content': f'Share images with ids: {randomId}'
-
-                                        },
-                                        {
-                                            'role': 'user',
-                                            'content': [{
-                                                'type': 'image_url',
-                                                'image_url': {
-                                                    'url': output.dataurl,
-                                                    'detail': 'auto'
-                                                }
-                                            }]
-                                        }]
-                                    )
-                                else:
-                                    image_histories[0]['content'] += ', ' + randomId
-                                    image_histories[1]["content"].append({ # type: ignore
-                                        'type': 'image_url',
-                                        'image_url': {
-                                            'url': output.dataurl,
-                                            'detail': 'auto'
-                                        }
-                                    })
-
-                                item = {
-                                    "role": "tool",
-                                    "content": f'Ask user for image id: {randomId}',
-                                    "tool_call_id": tool_call_id + str(toolindex),
-                                }
-                            else:
-                                item = {
-                                    "role": "tool",
-                                    "content": str(output),
-                                    "tool_call_id": tool_call_id + str(toolindex),
-                                }
-                            cost_, output_toks_ = get_output_cost(
-                                config.cost_file[config.model], enc, item
-                            )
-                            cost += cost_
-                            output_toks += output_toks_
-
-                            _histories.append(item)
-                    waiting_for_assistant = True
-                    break
-                elif chunk.choices[0].finish_reason:
-                    assistant_console.print("")
-                    item = {
-                        "role": "assistant",
-                        "content": full_response,
-                    }
-                    cost_, output_toks_ = get_output_cost(
-                        config.cost_file[config.model], enc, item
-                    )
-                    cost += cost_
-                    output_toks += output_toks_
-
-                    system_console.print(f"\nTotal cost: {config.cost_unit}{cost:.3f}")
-                    _histories.append(item)
-                    break
-
-                if chunk.choices[0].delta.tool_calls:
-                    tool_call = chunk.choices[0].delta.tool_calls[0]
-                    if tool_call.function and tool_call.function.arguments:
-                        tool_call_args_by_id[tool_call.id or ""][tool_call.index] += (
-                            tool_call.function.arguments
-                        )
-
-                chunk_str = chunk.choices[0].delta.content or ""
-                assistant_console.print(chunk_str, end="")
-                full_response += chunk_str
-        except KeyboardInterrupt:
-            waiting_for_assistant = False
-            input("Interrupted...enter to redo the current turn")
-        else:
-            history.extend(_histories)
-            history.extend(image_histories)
-            save_history(history, session_id)
-
-    return "Couldn't finish the task", cost
-
-
-if __name__ == "__main__":
-    app()

wcgw-1.0.1.dist-info/RECORD

@@ -1,15 +0,0 @@
-wcgw/__init__.py,sha256=VMi3gCAN_4_Ft8v5dY74u6bt5X-H1QhFJqTMZRd4fvk,76
-wcgw/types_.py,sha256=qpMRh1y136GkjIONIFowLy56v5p0OcdNqEH2mTmHPqU,756
-wcgw/client/__main__.py,sha256=ngI_vBcLAv7fJgmS4w4U7tuWtalGB8c7W5qebuT6Z6o,30
-wcgw/client/basic.py,sha256=2rY5pKm9dBBRuku0uIBUeLWjNc3iXjTSK9CmfuTxoS8,16196
-wcgw/client/claude.py,sha256=Bp45-UMBIJd-4tzX618nu-SpRbVtkTb1Es6c_gW6xy0,14861
-wcgw/client/common.py,sha256=grH-yV_4tnTQZ29xExn4YicGLxEq98z-HkEZwH0ReSg,1410
-wcgw/client/openai_adapters.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wcgw/client/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
-wcgw/client/tools.py,sha256=Ggm5QrSQiyv3Bnme-O9eo_8dXFwS1srCAoow-pODDSE,21212
-wcgw/relay/serve.py,sha256=e3oo4trAWmEKJWcLOsdZCICyEXUCqrI-71DaFnPKI70,8641
-wcgw/relay/static/privacy.txt,sha256=s9qBdbx2SexCpC_z33sg16TptmAwDEehMCLz4L50JLc,529
-wcgw-1.0.1.dist-info/METADATA,sha256=O2BPohrX_Eo6RRuWfY18Cz8MX6qTlgSnExa52Nd8Ggg,5222
-wcgw-1.0.1.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-wcgw-1.0.1.dist-info/entry_points.txt,sha256=WlIB825-Vm9ZtNzgENQsbHj4DRMkbpVR7uSkQyBlaPA,93
-wcgw-1.0.1.dist-info/RECORD,,