wcgw 0.0.8__py3-none-any.whl → 0.0.9__py3-none-any.whl

wcgw/basic.py

@@ -20,9 +20,8 @@ import petname
 from typer import Typer
 import uuid
 
-from wcgw.common import Config, text_from_editor
-
-from .common import Models
+from .common import Models, discard_input
+from .common import CostData, History
 from .openai_utils import get_input_cost, get_output_cost
 from .tools import ExecuteBash, ReadImage, ImageData
 
@@ -40,14 +39,40 @@ from .tools import (
 import tiktoken
 
 from urllib import parse
+import subprocess
 import os
+import tempfile
 
 import toml
+from pydantic import BaseModel
 
 
 from dotenv import load_dotenv
 
-History = list[ChatCompletionMessageParam]
+
+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
+
 
 def save_history(history: History, session_id: str) -> None:
     myid = str(history[1]["content"]).replace("/", "_").replace(" ", "_").lower()[:60]

wcgw/common.py

@@ -1,13 +1,9 @@
-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):
@@ -15,6 +11,14 @@ class CostData(BaseModel):
     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"]
 
 
@@ -41,27 +45,3 @@ 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

{wcgw-0.0.8.dist-info → wcgw-0.0.9.dist-info}/METADATA

@@ -1,11 +1,10 @@
 Metadata-Version: 2.3
 Name: wcgw
-Version: 0.0.8
+Version: 0.0.9
 Summary: What could go wrong giving full shell access to chatgpt?
 Project-URL: Homepage, https://github.com/rusiaaman/wcgw
 Author-email: Aman Rusia <gapypi@arcfu.com>
 Requires-Python: <3.13,>=3.8
-Requires-Dist: anthropic>=0.36.2
 Requires-Dist: fastapi>=0.115.0
 Requires-Dist: mypy>=1.11.2
 Requires-Dist: openai>=1.46.0

wcgw-0.0.9.dist-info/RECORD

@@ -0,0 +1,11 @@
+wcgw/__init__.py,sha256=okSsOWpTKDjEQzgOin3Kdpx4Mc3MFX1RunjopHQSIWE,62
+wcgw/__main__.py,sha256=MjJnFwfYzA1rW47xuSP1EVsi53DTHeEGqESkQwsELFQ,34
+wcgw/basic.py,sha256=BosCItWh4SpRNygEvCtf59Gm9xto3jof6SkPecjExHE,15804
+wcgw/common.py,sha256=jn39zTpaFUO1PWof_z7qBmklaZH5G1blzjlBvez0cg4,1225
+wcgw/openai_adapters.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wcgw/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
+wcgw/tools.py,sha256=ozJtcuOlMrdGoDmSv4UpolnStdY7Xz8Z7JyrRllKA7s,17088
+wcgw-0.0.9.dist-info/METADATA,sha256=meAGr9uOuxltYth3XyRHe2RNOU--BIEQcE_26So-K5Q,1995
+wcgw-0.0.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+wcgw-0.0.9.dist-info/entry_points.txt,sha256=T-IH7w6Vc650hr8xksC8kJfbJR4uwN8HDudejwDwrNM,59
+wcgw-0.0.9.dist-info/RECORD,,

wcgw-0.0.8.dist-info/RECORD

@@ -1,10 +0,0 @@
-wcgw/__init__.py,sha256=okSsOWpTKDjEQzgOin3Kdpx4Mc3MFX1RunjopHQSIWE,62
-wcgw/__main__.py,sha256=MjJnFwfYzA1rW47xuSP1EVsi53DTHeEGqESkQwsELFQ,34
-wcgw/basic.py,sha256=o_iyg3Rmqz08LTWzgO7JIA0u_5l6GGaXyJe0zw73v8w,15085
-wcgw/common.py,sha256=gHiP1RVHkvp10jPc1Xzg5DtUqhGbgQ7pTJK7OvUXfZQ,1764
-wcgw/openai_utils.py,sha256=YNwCsA-Wqq7jWrxP0rfQmBTb1dI0s7dWXzQqyTzOZT4,2629
-wcgw/tools.py,sha256=ozJtcuOlMrdGoDmSv4UpolnStdY7Xz8Z7JyrRllKA7s,17088
-wcgw-0.0.8.dist-info/METADATA,sha256=XebadIrzit30OJD3F1e1llsu2z_0B_9x8lI6T-R3Q8I,2028
-wcgw-0.0.8.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-wcgw-0.0.8.dist-info/entry_points.txt,sha256=T-IH7w6Vc650hr8xksC8kJfbJR4uwN8HDudejwDwrNM,59
-wcgw-0.0.8.dist-info/RECORD,,