simple-agents-py 0.1.14__tar.gz → 0.1.15__tar.gz

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/Cargo.lock

@@ -2035,7 +2035,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agent-type"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "blake3",
@@ -2050,7 +2050,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cache"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2059,7 +2059,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cli"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "clap",
  "serde",
@@ -2076,7 +2076,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-core"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "futures-core",
@@ -2104,7 +2104,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-ffi"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2115,7 +2115,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-healing"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "criterion",
  "proptest",
@@ -2131,7 +2131,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-macros"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -2143,7 +2143,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-napi"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "napi",
  "napi-derive",
@@ -2155,7 +2155,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-providers"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "bytes",
@@ -2183,7 +2183,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-py"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "futures-util",
@@ -2202,7 +2202,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-router"
-version = "0.1.14"
+version = "0.1.15"
 dependencies = [
  "async-trait",
  "futures-core",

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/Cargo.toml

@@ -3,7 +3,7 @@ members = ["crates/*"]
 resolver = "2"
 
 [workspace.package]
-version = "0.1.14"
+version = "0.1.15"
 edition = "2021"
 rust-version = "1.75"
 authors = ["SimpleAgents Contributors"]

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simple-agents-py
-Version: 0.1.14
+Version: 0.1.15
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Rust
 Classifier: License :: OSI Approved :: MIT License
@@ -57,7 +57,7 @@ from simple_agents_py import Client
 
 client = Client("openai")
 response = client.complete("gpt-4", "Hello from Python!", max_tokens=128, temperature=0.7)
-print(response)
+print(response.content)
 ```
 
 ## Feature Guide
@@ -95,6 +95,28 @@ json_text = client.complete_json_schema("gpt-4o-mini", messages, schema, "person
 print(json.loads(json_text))
 ```
 
+Pydantic models are accepted too:
+
+```python
+from pydantic import BaseModel
+from simple_agents_py import Client
+
+class Person(BaseModel):
+    name: str
+    age: int
+
+client = Client("openai")
+messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
+json_text = client.complete_json_schema("gpt-4o-mini", messages, Person, "person")
+print(json_text)
+```
+
+**Healing for Structured Outputs**: Healing is enabled by default for structured outputs, automatically fixing malformed JSON, type mismatches, and missing fields. To disable healing:
+
+```python
+client = Client("openai", healing=False)
+```
+
 ### Structured Streaming
 
 ```python
@@ -124,6 +146,7 @@ messages = [{"role": "user", "content": "Return JSON: {\"name\":\"Sam\",\"age\":
 healed = client.complete_json_healed("gpt-4o-mini", messages, max_tokens=64)
 print(healed.content)
 print(healed.was_healed, healed.confidence)
+print(healed.usage.get("total_tokens"))
 ```
 
 ### Tool Calling
@@ -172,7 +195,7 @@ builder = (
     .add_middleware(TimingMiddleware())
 )
 client = builder.build()
-print(client.complete("gpt-4o-mini", "Give me one idea."))
+print(client.complete("gpt-4o-mini", "Give me one idea.").content)
 ```
 
 ## Examples
@@ -183,8 +206,8 @@ OpenAI with a short prompt:
 from simple_agents_py import Client
 
 client = Client("openai")
-text = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
-print(text)
+response = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
+print(response.content)
 ```
 
 Anthropic with custom tokens and temperature:
@@ -193,13 +216,13 @@ Anthropic with custom tokens and temperature:
 from simple_agents_py import Client
 
 client = Client("anthropic")
-text = client.complete(
+response = client.complete(
     "claude-3-5-sonnet-20240620",
     "Write a friendly welcome message for a new user.",
     max_tokens=120,
     temperature=0.5,
 )
-print(text)
+print(response.content)
 ```
 
 OpenRouter with a specific model:
@@ -208,8 +231,8 @@ OpenRouter with a specific model:
 from simple_agents_py import Client
 
 client = Client("openrouter")
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenRouter with explicit API base and key:
@@ -222,8 +245,8 @@ client = Client(
     api_base="https://openrouter.ai/api/v1",
     api_key="sk-your-openrouter-key",
 )
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenAI with a custom gateway:
@@ -236,8 +259,8 @@ client = Client(
     api_base="http://localhost:4000/v1",
     api_key="sk-your-openai-key",
 )
-text = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
-print(text)
+response = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
+print(response.content)
 ```
 
 Basic error handling:
@@ -247,7 +270,7 @@ from simple_agents_py import Client
 
 try:
     client = Client("openai")
-    print(client.complete("gpt-4o-mini", "Hello!"))
+    print(client.complete("gpt-4o-mini", "Hello!").content)
 except RuntimeError as exc:
     print(f"Request failed: {exc}")
 ```

{simple_agents_py-0.1.14/crates/simple-agents-py → simple_agents_py-0.1.15}/README.md

@@ -42,7 +42,7 @@ from simple_agents_py import Client
 
 client = Client("openai")
 response = client.complete("gpt-4", "Hello from Python!", max_tokens=128, temperature=0.7)
-print(response)
+print(response.content)
 ```
 
 ## Feature Guide
@@ -80,6 +80,28 @@ json_text = client.complete_json_schema("gpt-4o-mini", messages, schema, "person
 print(json.loads(json_text))
 ```
 
+Pydantic models are accepted too:
+
+```python
+from pydantic import BaseModel
+from simple_agents_py import Client
+
+class Person(BaseModel):
+    name: str
+    age: int
+
+client = Client("openai")
+messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
+json_text = client.complete_json_schema("gpt-4o-mini", messages, Person, "person")
+print(json_text)
+```
+
+**Healing for Structured Outputs**: Healing is enabled by default for structured outputs, automatically fixing malformed JSON, type mismatches, and missing fields. To disable healing:
+
+```python
+client = Client("openai", healing=False)
+```
+
 ### Structured Streaming
 
 ```python
@@ -109,6 +131,7 @@ messages = [{"role": "user", "content": "Return JSON: {\"name\":\"Sam\",\"age\":
 healed = client.complete_json_healed("gpt-4o-mini", messages, max_tokens=64)
 print(healed.content)
 print(healed.was_healed, healed.confidence)
+print(healed.usage.get("total_tokens"))
 ```
 
 ### Tool Calling
@@ -157,7 +180,7 @@ builder = (
     .add_middleware(TimingMiddleware())
 )
 client = builder.build()
-print(client.complete("gpt-4o-mini", "Give me one idea."))
+print(client.complete("gpt-4o-mini", "Give me one idea.").content)
 ```
 
 ## Examples
@@ -168,8 +191,8 @@ OpenAI with a short prompt:
 from simple_agents_py import Client
 
 client = Client("openai")
-text = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
-print(text)
+response = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
+print(response.content)
 ```
 
 Anthropic with custom tokens and temperature:
@@ -178,13 +201,13 @@ Anthropic with custom tokens and temperature:
 from simple_agents_py import Client
 
 client = Client("anthropic")
-text = client.complete(
+response = client.complete(
     "claude-3-5-sonnet-20240620",
     "Write a friendly welcome message for a new user.",
     max_tokens=120,
     temperature=0.5,
 )
-print(text)
+print(response.content)
 ```
 
 OpenRouter with a specific model:
@@ -193,8 +216,8 @@ OpenRouter with a specific model:
 from simple_agents_py import Client
 
 client = Client("openrouter")
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenRouter with explicit API base and key:
@@ -207,8 +230,8 @@ client = Client(
     api_base="https://openrouter.ai/api/v1",
     api_key="sk-your-openrouter-key",
 )
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenAI with a custom gateway:
@@ -221,8 +244,8 @@ client = Client(
     api_base="http://localhost:4000/v1",
     api_key="sk-your-openai-key",
 )
-text = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
-print(text)
+response = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
+print(response.content)
 ```
 
 Basic error handling:
@@ -232,7 +255,7 @@ from simple_agents_py import Client
 
 try:
     client = Client("openai")
-    print(client.complete("gpt-4o-mini", "Hello!"))
+    print(client.complete("gpt-4o-mini", "Hello!").content)
 except RuntimeError as exc:
     print(f"Request failed: {exc}")
 ```

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/crates/simple-agents-core/Cargo.toml

@@ -18,9 +18,9 @@ serde_json.workspace = true
 tokio.workspace = true
 
 simple-agent-type = { path = "../simple-agent-type", version = "0.1.0" }
-simple-agents-router = { path = "../simple-agents-router", version = "0.1.14" }
-simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.14" }
-simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.14" }
+simple-agents-router = { path = "../simple-agents-router", version = "0.1.15" }
+simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.15" }
+simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.15" }
 futures-core = "0.3"
 
 [dev-dependencies]

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/crates/simple-agents-healing/Cargo.toml

@@ -32,7 +32,7 @@ regex-support = ["regex"]
 tokio = { version = "1.41", features = ["full"] }
 proptest = "1.5"
 criterion = { version = "0.5", features = ["html_reports"] }
-simple-agents-macros = { path = "../simple-agents-macros", version = "0.1.14" }
+simple-agents-macros = { path = "../simple-agents-macros", version = "0.1.15" }
 
 [[test]]
 name = "parser_tests"

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/crates/simple-agents-healing/src/coercion.rs

@@ -163,6 +163,24 @@ impl CoercionEngine {
             (Value::Number(n), Schema::UInt) if n.is_f64() => {
                 self.coerce_float_to_uint(n.as_f64().unwrap(), flags, confidence)
             }
+            // Int to Float coercion (integers are valid numbers in JSON Schema)
+            (Value::Number(n), Schema::Float) if n.is_i64() || n.is_u64() => {
+                // Convert integer to float - this is lossless for most practical values
+                let float_val = if let Some(i) = n.as_i64() {
+                    i as f64
+                } else if let Some(u) = n.as_u64() {
+                    u as f64
+                } else {
+                    unreachable!()
+                };
+                Ok(Value::Number(
+                    serde_json::Number::from_f64(float_val)
+                        .ok_or_else(|| HealingError::ParseError {
+                            input: format!("{}", n),
+                            expected_type: "float".to_string(),
+                        })?,
+                ))
+            }
 
             // Array coercion
             (Value::Array(arr), Schema::Array(elem_schema)) => {
@@ -427,6 +445,13 @@ impl CoercionEngine {
                     });
                     *confidence *= 0.9;
                     result.insert(field.name.clone(), default.clone());
+                } else if flags.contains(&CoercionFlag::TruncatedJson) {
+                    // If response was truncated, allow missing required fields by injecting null
+                    flags.push(CoercionFlag::UsedDefaultValue {
+                        field: field.name.clone(),
+                    });
+                    *confidence *= 0.7; // Lower confidence for missing required field
+                    result.insert(field.name.clone(), Value::Null);
                 } else {
                     return Err(HealingError::MissingField {
                         field: field.name.clone(),

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/crates/simple-agents-providers/Cargo.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 
 [dependencies]
 simple-agent-type = { path = "../simple-agent-type", version = "0.1.0" }
-simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.14" }
+simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.15" }
 reqwest = { version = "0.12", features = ["json", "stream"] }
 tokio = { version = "1.42", features = ["full"] }
 futures = "0.3"
@@ -34,6 +34,6 @@ prometheus = ["dep:metrics-exporter-prometheus"]
  [dev-dependencies]
  tokio-test = "0.4"
  tracing-subscriber = "0.3"
- simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.14" }
- simple-agents-macros = { path = "../simple-agents-macros", version = "0.1.14" }
+ simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.15" }
+ simple-agents-macros = { path = "../simple-agents-macros", version = "0.1.15" }
  dotenv = "0.15"

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15}/crates/simple-agents-py/Cargo.toml

@@ -18,9 +18,9 @@ serde_json = { workspace = true }
 futures-util = "0.3"
 async-trait = "0.1"
 
-simple-agents-core = { path = "../simple-agents-core", version = "0.1.14" }
-simple-agents-providers = { path = "../simple-agents-providers", version = "0.1.14" }
-simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.14" }
-simple-agents-router = { path = "../simple-agents-router", version = "0.1.14" }
+simple-agents-core = { path = "../simple-agents-core", version = "0.1.15" }
+simple-agents-providers = { path = "../simple-agents-providers", version = "0.1.15" }
+simple-agents-cache = { path = "../simple-agents-cache", version = "0.1.15" }
+simple-agents-router = { path = "../simple-agents-router", version = "0.1.15" }
 simple-agent-type = { path = "../simple-agent-type", version = "0.1.0" }
-simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.14" }
+simple-agents-healing = { path = "../simple-agents-healing", version = "0.1.15" }

{simple_agents_py-0.1.14 → simple_agents_py-0.1.15/crates/simple-agents-py}/README.md

@@ -42,7 +42,7 @@ from simple_agents_py import Client
 
 client = Client("openai")
 response = client.complete("gpt-4", "Hello from Python!", max_tokens=128, temperature=0.7)
-print(response)
+print(response.content)
 ```
 
 ## Feature Guide
@@ -80,6 +80,28 @@ json_text = client.complete_json_schema("gpt-4o-mini", messages, schema, "person
 print(json.loads(json_text))
 ```
 
+Pydantic models are accepted too:
+
+```python
+from pydantic import BaseModel
+from simple_agents_py import Client
+
+class Person(BaseModel):
+    name: str
+    age: int
+
+client = Client("openai")
+messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
+json_text = client.complete_json_schema("gpt-4o-mini", messages, Person, "person")
+print(json_text)
+```
+
+**Healing for Structured Outputs**: Healing is enabled by default for structured outputs, automatically fixing malformed JSON, type mismatches, and missing fields. To disable healing:
+
+```python
+client = Client("openai", healing=False)
+```
+
 ### Structured Streaming
 
 ```python
@@ -109,6 +131,7 @@ messages = [{"role": "user", "content": "Return JSON: {\"name\":\"Sam\",\"age\":
 healed = client.complete_json_healed("gpt-4o-mini", messages, max_tokens=64)
 print(healed.content)
 print(healed.was_healed, healed.confidence)
+print(healed.usage.get("total_tokens"))
 ```
 
 ### Tool Calling
@@ -157,7 +180,7 @@ builder = (
     .add_middleware(TimingMiddleware())
 )
 client = builder.build()
-print(client.complete("gpt-4o-mini", "Give me one idea."))
+print(client.complete("gpt-4o-mini", "Give me one idea.").content)
 ```
 
 ## Examples
@@ -168,8 +191,8 @@ OpenAI with a short prompt:
 from simple_agents_py import Client
 
 client = Client("openai")
-text = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
-print(text)
+response = client.complete("gpt-4o-mini", "Summarize this in one sentence.")
+print(response.content)
 ```
 
 Anthropic with custom tokens and temperature:
@@ -178,13 +201,13 @@ Anthropic with custom tokens and temperature:
 from simple_agents_py import Client
 
 client = Client("anthropic")
-text = client.complete(
+response = client.complete(
     "claude-3-5-sonnet-20240620",
     "Write a friendly welcome message for a new user.",
     max_tokens=120,
     temperature=0.5,
 )
-print(text)
+print(response.content)
 ```
 
 OpenRouter with a specific model:
@@ -193,8 +216,8 @@ OpenRouter with a specific model:
 from simple_agents_py import Client
 
 client = Client("openrouter")
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenRouter with explicit API base and key:
@@ -207,8 +230,8 @@ client = Client(
     api_base="https://openrouter.ai/api/v1",
     api_key="sk-your-openrouter-key",
 )
-text = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
-print(text)
+response = client.complete("openai/gpt-4o-mini", "Give me three project ideas.")
+print(response.content)
 ```
 
 OpenAI with a custom gateway:
@@ -221,8 +244,8 @@ client = Client(
     api_base="http://localhost:4000/v1",
     api_key="sk-your-openai-key",
 )
-text = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
-print(text)
+response = client.complete("gpt-4o-mini", "Say hello from a local proxy.")
+print(response.content)
 ```
 
 Basic error handling:
@@ -232,7 +255,7 @@ from simple_agents_py import Client
 
 try:
     client = Client("openai")
-    print(client.complete("gpt-4o-mini", "Hello!"))
+    print(client.complete("gpt-4o-mini", "Hello!").content)
 except RuntimeError as exc:
     print(f"Request failed: {exc}")
 ```