version 0.1.2; cargo fmt; fix models pricing

Author: alexylon

README.md

@@ -2,13 +2,13 @@
 
 ![](https://github.com/alexylon/sofos-code/actions/workflows/rust.yml/badge.svg)
 
-A blazingly fast, interactive AI coding assistant powered by Claude or OpenAI, implemented in pure Rust, that can generate code, edit files, and search the web - all from your terminal.
+A blazingly fast, interactive AI coding assistant powered by Claude or GPT, implemented in pure Rust, that can generate code, edit files, and search the web - all from your terminal.
 
 <div align="center"><img src="/assets/sofos_code.png" style="width: 800px;" alt="Sofos Code"></div>
 
 ## Features
 
-- **Interactive REPL** - Multi-turn conversations with Claude or OpenAI
+- **Interactive REPL** - Multi-turn conversations with Claude or GPT
 - **Session History** - Automatic session saving and resume previous conversations
 - **Custom Instructions** - Project and personal instruction files for context-aware assistance
 - **File Operations** - Read, write, list, and create files/directories (sandboxed to current directory)
@@ -17,7 +17,7 @@ A blazingly fast, interactive AI coding assistant powered by Claude or OpenAI, i
 - **Web Search** - Real-time web information via Claude's native search tool
 - **Bash Execution** - Run tests and build commands safely (read-only, sandboxed)
 - **Visual Diff Display** - See exactly what changed with colored diffs (red for deletions, blue for additions)
-- **Iterative Tool Execution** - Claude can use up to 200 tools per request for complex multi-file operations
+- **Iterative Tool Execution** - Sofos can use up to 200 tools per request for complex multi-file operations
 - **Session Usage** – After exiting Sofos, a session usage is displayed, including the input and output tokens used and the estimated cost.
 - **Secure** - All operations restricted to workspace, prevents directory traversal
 
@@ -103,7 +103,7 @@ Estimated cost:     $0.1304
 **Cost Calculation:**
 - Costs are calculated based on official model pricing
 - Claude models use official Anthropic pricing (e.g., Sonnet 4.5: $3/$15 per million input/output tokens)
-- OpenAI models use estimated pricing ($5/$15 per million tokens for gpt-5.1-codex models)
+- OpenAI models use official OpenAI pricing ($1.25/$10 per million tokens for gpt-5.1-codex and gpt-5.1-codex-max models)
 - Accurate for standard API usage
 
 ### Options
@@ -249,7 +249,7 @@ MIT License
 
 ## Acknowledgments
 
-Built with Rust and powered by Anthropic's Claude. Morph Apply integration for fast edits. Inspired by Aider and similar tools.
+Built with Rust and powered by Anthropic's Claude or OpenAI's GPT. Morph Apply integration for fast edits. Inspired by Aider and similar tools.
 
 ---
 

src/api/openai.rs

@@ -5,7 +5,7 @@ use serde::Deserialize;
 use serde_json::json;
 use std::time::Duration;
 
-// OpenAI responses API (for gpt-5.1-codex) and chat completions (for other chat models)
+// OpenAI responses API (for gpt-5.1-codex-max) and chat completions (for other chat models)
 const OPENAI_API_BASE: &str = "https://api.openai.com/v1";
 const REQUEST_TIMEOUT: Duration = super::anthropic::REQUEST_TIMEOUT;
 
@@ -37,7 +37,7 @@ impl OpenAIClient {
         &self,
         request: CreateMessageRequest,
     ) -> Result<CreateMessageResponse> {
-        // gpt-5.1-codex uses the /responses endpoint with `input`
+        // gpt-5.1-codex-max uses the /responses endpoint with `input`
         if request.model.contains("gpt-5.1-codex") {
             return self.call_responses(request).await;
         }

src/main.rs

@@ -48,7 +48,7 @@ fn main() -> Result<()> {
                 eprintln!("Please set your OpenAI API key:");
                 eprintln!("  export OPENAI_API_KEY='your-api-key'");
                 eprintln!("Or use the --openai-api-key flag:");
-                eprintln!("  sofos --openai-api-key 'your-api-key' --model gpt-5.1-codex");
+                eprintln!("  sofos --openai-api-key 'your-api-key' --model gpt-5.1-codex-max");
                 std::process::exit(1);
             }
         }

src/ui.rs

@@ -304,18 +304,10 @@ impl UI {
     fn calculate_cost(model: &str, input_tokens: u32, output_tokens: u32) -> f64 {
         // Prices per million tokens in USD
         let (input_price, output_price) = match model {
-            "claude-sonnet-4-5" | "claude-sonnet-4.5" => (3.0, 15.0),
-            "claude-haiku-4-5" | "claude-haiku-4.5" => (0.8, 4.0),
-            "claude-sonnet-4" | "claude-4-sonnet-20250514" => (3.0, 15.0),
-            "claude-opus-4" | "claude-4-opus-20250514" => (15.0, 75.0),
-            "claude-opus-4.1" | "claude-opus-4-1" => (6.0, 30.0),
-            "claude-sonnet-3-7" | "claude-sonnet-3.7" => (3.0, 15.0),
-            "claude-sonnet-3-5" | "claude-sonnet-3.5" => (3.0, 15.0),
-            "claude-haiku-3-5" | "claude-haiku-3.5" => (0.8, 4.0),
-            "claude-opus-3" | "claude-3-opus-20240229" => (15.0, 75.0),
-            "claude-sonnet-3" | "claude-3-sonnet-20240229" => (3.0, 15.0),
-            "claude-haiku-3" | "claude-3-haiku-20240307" => (0.25, 1.25),
-            "gpt-5.1-codex" => (5.0, 15.0),
+            "claude-sonnet-4-5" => (3.0, 15.0),
+            "claude-opus-4-5" => (5.0, 25.0),
+            "claude-haiku-4-5" => (1.0, 5.0),
+            "gpt-5.1-codex-max" | "gpt-5.1-codex" | "gpt-5-codex" => (1.25, 10.0),
             // Default fallback (use Sonnet 4.5 pricing)
             _ => (3.0, 15.0),
         };