Pinned python version. Bumped up version. Minor print change

Author: Aman Rusia

.python-version

@@ -0,0 +1 @@
+3.12

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 authors = [{ name = "Aman Rusia", email = "[email protected]" }]
 name = "wcgw"
-version = "0.0.6"
+version = "0.0.7"
 description = "What could go wrong giving full shell access to chatgpt?"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -21,6 +21,7 @@ dependencies = [
     "fastapi>=0.115.0",
     "uvicorn>=0.31.0",
     "websockets>=13.1",
+    "anthropic>=0.36.2",
 ]
 
 [project.urls]

src/wcgw/basic.py

@@ -20,8 +20,9 @@
 from typer import Typer
 import uuid
 
-from .common import Models, discard_input
-from .common import CostData, History
+from wcgw.common import Config, text_from_editor
+
+from .common import Models
 from .openai_utils import get_input_cost, get_output_cost
 from .tools import ExecuteBash, ReadImage, ImageData
 
@@ -39,40 +40,14 @@
 import tiktoken
 
 from urllib import parse
-import subprocess
 import os
-import tempfile
 
 import toml
-from pydantic import BaseModel
 
 
 from dotenv import load_dotenv
 
-
-class Config(BaseModel):
-    model: Models
-    secondary_model: Models
-    cost_limit: float
-    cost_file: dict[Models, CostData]
-    cost_unit: str = "$"
-
-
-def text_from_editor(console: rich.console.Console) -> str:
-    # First consume all the input till now
-    discard_input()
-    console.print("\n---------------------------------------\n# User message")
-    data = input()
-    if data:
-        return data
-    editor = os.environ.get("EDITOR", "vim")
-    with tempfile.NamedTemporaryFile(suffix=".tmp") as tf:
-        subprocess.run([editor, tf.name], check=True)
-        with open(tf.name, "r") as f:
-            data = f.read()
-            console.print(data)
-            return data
-
+History = list[ChatCompletionMessageParam]
 
 def save_history(history: History, session_id: str) -> None:
     myid = str(history[1]["content"]).replace("/", "_").replace(" ", "_").lower()[:60]

src/wcgw/common.py

@@ -1,24 +1,20 @@
+import os
 import select
+import subprocess
 import sys
+import tempfile
 import termios
 import tty
 from typing import Literal
 from pydantic import BaseModel
+import rich
 
 
 class CostData(BaseModel):
     cost_per_1m_input_tokens: float
     cost_per_1m_output_tokens: float
 
 
-from openai.types.chat import (
-    ChatCompletionMessageParam,
-    ChatCompletionAssistantMessageParam,
-    ChatCompletionMessage,
-    ParsedChatCompletionMessage,
-)
-
-History = list[ChatCompletionMessageParam]
 Models = Literal["gpt-4o-2024-08-06", "gpt-4o-mini"]
 
 
@@ -45,3 +41,27 @@ def discard_input() -> None:
     finally:
         # Restore old terminal settings
         termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+
+
+class Config(BaseModel):
+    model: Models
+    secondary_model: Models
+    cost_limit: float
+    cost_file: dict[Models, CostData]
+    cost_unit: str = "$"
+
+
+def text_from_editor(console: rich.console.Console) -> str:
+    # First consume all the input till now
+    discard_input()
+    console.print("\n---------------------------------------\n# User message")
+    data = input()
+    if data:
+        return data
+    editor = os.environ.get("EDITOR", "vim")
+    with tempfile.NamedTemporaryFile(suffix=".tmp") as tf:
+        subprocess.run([editor, tf.name], check=True)
+        with open(tf.name, "r") as f:
+            data = f.read()
+            console.print(data)
+            return data

src/wcgw/openai_adapters.py

@@ -467,7 +467,7 @@ async def register_client(server_url: str, client_uuid: str = "") -> None:
 
     # Create the WebSocket connection
     async with websockets.connect(f"{server_url}/{client_uuid}") as websocket:
-        print(f"Connected. Share this user id with the chatbot: {client_uuid}")
+        print(f"Connected. Share this user id with the chatbot: {client_uuid} \nLink: https://chatgpt.com/g/g-Us0AAXkRh-wcgw-giving-shell-access")
         try:
             while True:
                 # Wait to receive data from the server