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

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

@@ -2,6 +2,12 @@
 # It is not intended for manual editing.
 version = 3
 
+[[package]]
+name = "adler2"
+version = "2.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "320119579fcad9c21884f5c4861d16174d0e06250625266f50fe6898340abefa"
+
 [[package]]
 name = "ahash"
 version = "0.8.12"
@@ -23,6 +29,21 @@ dependencies = [
  "memchr",
 ]
 
+[[package]]
+name = "alloc-no-stdlib"
+version = "2.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc7bb162ec39d46ab1ca8c77bf72e890535becd1751bb45f64c597edb4c8c6b3"
+
+[[package]]
+name = "alloc-stdlib"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94fb8275041c72129eb51b7d0322c29b8387a0386127718b096429201a5d6ece"
+dependencies = [
+ "alloc-no-stdlib",
+]
+
 [[package]]
 name = "anes"
 version = "0.1.6"
@@ -91,6 +112,18 @@ version = "0.7.6"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "7c02d123df017efcdfbd739ef81735b36c5ba83ec3c59c80a9d7ecc718f92e50"
 
+[[package]]
+name = "async-compression"
+version = "0.4.37"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d10e4f991a553474232bc0a31799f6d24b034a84c0971d80d2e2f78b2e576e40"
+dependencies = [
+ "compression-codecs",
+ "compression-core",
+ "pin-project-lite",
+ "tokio",
+]
+
 [[package]]
 name = "async-trait"
 version = "0.1.89"
@@ -177,6 +210,27 @@ dependencies = [
  "cpufeatures",
 ]
 
+[[package]]
+name = "brotli"
+version = "8.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4bd8b9603c7aa97359dbd97ecf258968c95f3adddd6db2f7e7a5bef101c84560"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+ "brotli-decompressor",
+]
+
+[[package]]
+name = "brotli-decompressor"
+version = "5.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "874bb8112abecc98cbd6d81ea4fa7e94fb9449648c93cc89aa40c81c24d7de03"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+]
+
 [[package]]
 name = "bumpalo"
 version = "3.19.1"
@@ -295,6 +349,24 @@ version = "1.0.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "b05b61dc5112cbb17e4b6cd61790d9845d13888356391624cbe7e41efeac1e75"
 
+[[package]]
+name = "compression-codecs"
+version = "0.4.36"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "00828ba6fd27b45a448e57dbfe84f1029d4c9f26b368157e9a448a5f49a2ec2a"
+dependencies = [
+ "brotli",
+ "compression-core",
+ "flate2",
+ "memchr",
+]
+
+[[package]]
+name = "compression-core"
+version = "0.4.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "75984efb6ed102a0d42db99afb6c1948f0380d1d91808d5529916e6c08b49d8d"
+
 [[package]]
 name = "constant_time_eq"
 version = "0.4.2"
@@ -345,6 +417,15 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "crc32fast"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9481c1c90cbf2ac953f07c8d4a58aa3945c425b7185c9154d67a65e4230da511"
+dependencies = [
+ "cfg-if",
+]
+
 [[package]]
 name = "criterion"
 version = "0.5.1"
@@ -501,6 +582,16 @@ version = "0.1.8"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "8591b0bcc8a98a64310a2fae1bb3e9b8564dd10e381e6e28010fde8e8e8568db"
 
+[[package]]
+name = "flate2"
+version = "1.1.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b375d6465b98090a5f25b1c7703f3859783755aa9a80433b36e0379a3ec2f369"
+dependencies = [
+ "crc32fast",
+ "miniz_oxide",
+]
+
 [[package]]
 name = "fnv"
 version = "1.0.7"
@@ -1165,6 +1256,16 @@ version = "0.3.17"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "6877bb514081ee2a7ff5ef9de3281f14a4dd4bceac4c09388074a6b5df8a139a"
 
+[[package]]
+name = "miniz_oxide"
+version = "0.8.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fa76a2c86f704bdb222d66965fb3d63269ce38518b83cb0575fca855ebb6316"
+dependencies = [
+ "adler2",
+ "simd-adler32",
+]
+
 [[package]]
 name = "mio"
 version = "1.1.1"
@@ -2033,9 +2134,15 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "simd-adler32"
+version = "0.3.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e320a6c5ad31d271ad523dcf3ad13e2767ad8b1cb8f047f75a8aeaf8da139da2"
+
 [[package]]
 name = "simple-agent-type"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "blake3",
@@ -2050,7 +2157,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cache"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2059,7 +2166,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cli"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "clap",
  "serde",
@@ -2076,7 +2183,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-core"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "futures-core",
@@ -2091,7 +2198,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-examples"
-version = "0.1.0"
+version = "0.2.0"
 dependencies = [
  "dotenv",
  "futures-util",
@@ -2104,7 +2211,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-ffi"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2115,7 +2222,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-healing"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "criterion",
  "proptest",
@@ -2131,7 +2238,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-macros"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -2143,7 +2250,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-napi"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "napi",
  "napi-derive",
@@ -2155,7 +2262,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-providers"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "bytes",
@@ -2183,7 +2290,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-py"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "futures-util",
@@ -2202,7 +2309,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-router"
-version = "0.1.14"
+version = "0.2.0"
 dependencies = [
  "async-trait",
  "futures-core",
@@ -2552,13 +2659,18 @@ version = "0.6.8"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "d4e6559d53cc268e5031cd8429d05415bc4cb4aefc4aa5d6cc35fbf5b924a1f8"
 dependencies = [
+ "async-compression",
  "bitflags",
  "bytes",
+ "futures-core",
  "futures-util",
  "http",
  "http-body",
+ "http-body-util",
  "iri-string",
  "pin-project-lite",
+ "tokio",
+ "tokio-util",
  "tower",
  "tower-layer",
  "tower-service",

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

@@ -3,7 +3,7 @@ members = ["crates/*"]
 resolver = "2"
 
 [workspace.package]
-version = "0.1.14"
+version = "0.2.0"
 edition = "2021"
 rust-version = "1.75"
 authors = ["SimpleAgents Contributors"]
@@ -28,3 +28,6 @@ futures-util = "0.3"
 metrics = "0.23"
 metrics-exporter-prometheus = "0.15"
 governor = "0.6"
+simple-agent-type = { version = "0.2.0", path = "crates/simple-agent-type" }
+simple-agents-healing = { version = "0.2.0", path = "crates/simple-agents-healing" }
+simple-agents-providers = { version = "0.2.0", path = "crates/simple-agents-providers" }

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simple-agents-py
-Version: 0.1.14
+Version: 0.2.0
 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
@@ -69,7 +69,7 @@ from simple_agents_py import Client
 
 client = Client("openai")
 messages = [{"role": "user", "content": "Say hello in one sentence."}]
-for chunk in client.stream("gpt-4o-mini", messages, max_tokens=64):
+for chunk in client.complete("gpt-4o-mini", messages, max_tokens=64, stream=True):
     if chunk.content:
         print(chunk.content, end="", flush=True)
 print()
@@ -91,10 +91,42 @@ schema = {
     "required": ["name", "age"],
 }
 messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
-json_text = client.complete_json_schema("gpt-4o-mini", messages, schema, "person")
+json_text = client.complete(
+    "gpt-4o-mini",
+    messages,
+    schema=schema,
+    schema_name="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(
+    "gpt-4o-mini",
+    messages,
+    schema=Person,
+    schema_name="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
@@ -107,7 +139,13 @@ schema = {
     "required": ["name", "age"],
 }
 messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
-for event in client.stream_structured("gpt-4o-mini", messages, schema, max_tokens=64):
+for event in client.complete(
+    "gpt-4o-mini",
+    messages,
+    schema=schema,
+    max_tokens=64,
+    stream=True,
+):
     if event.is_partial:
         print("partial:", event.partial_value)
     else:
@@ -121,9 +159,16 @@ from simple_agents_py import Client
 
 client = Client("openai")
 messages = [{"role": "user", "content": "Return JSON: {\"name\":\"Sam\",\"age\":30}"}]
-healed = client.complete_json_healed("gpt-4o-mini", messages, max_tokens=64)
+healed = client.complete(
+    "gpt-4o-mini",
+    messages,
+    max_tokens=64,
+    response_format="json",
+    heal=True,
+)
 print(healed.content)
 print(healed.was_healed, healed.confidence)
+print(healed.usage.get("total_tokens"))
 ```
 
 ### Tool Calling
@@ -150,7 +195,7 @@ tools = [
     }
 ]
 messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
-response = client.complete_with_tools("gpt-4o-mini", messages, tools)
+response = client.complete("gpt-4o-mini", messages, tools=tools)
 print(response.tool_calls)
 ```
 
@@ -172,7 +217,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 +228,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 +238,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 +253,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 +267,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 +281,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 +292,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.2.0}/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
@@ -54,7 +54,7 @@ from simple_agents_py import Client
 
 client = Client("openai")
 messages = [{"role": "user", "content": "Say hello in one sentence."}]
-for chunk in client.stream("gpt-4o-mini", messages, max_tokens=64):
+for chunk in client.complete("gpt-4o-mini", messages, max_tokens=64, stream=True):
     if chunk.content:
         print(chunk.content, end="", flush=True)
 print()
@@ -76,10 +76,42 @@ schema = {
     "required": ["name", "age"],
 }
 messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
-json_text = client.complete_json_schema("gpt-4o-mini", messages, schema, "person")
+json_text = client.complete(
+    "gpt-4o-mini",
+    messages,
+    schema=schema,
+    schema_name="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(
+    "gpt-4o-mini",
+    messages,
+    schema=Person,
+    schema_name="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
@@ -92,7 +124,13 @@ schema = {
     "required": ["name", "age"],
 }
 messages = [{"role": "user", "content": "Extract name and age: Alice is 28."}]
-for event in client.stream_structured("gpt-4o-mini", messages, schema, max_tokens=64):
+for event in client.complete(
+    "gpt-4o-mini",
+    messages,
+    schema=schema,
+    max_tokens=64,
+    stream=True,
+):
     if event.is_partial:
         print("partial:", event.partial_value)
     else:
@@ -106,9 +144,16 @@ from simple_agents_py import Client
 
 client = Client("openai")
 messages = [{"role": "user", "content": "Return JSON: {\"name\":\"Sam\",\"age\":30}"}]
-healed = client.complete_json_healed("gpt-4o-mini", messages, max_tokens=64)
+healed = client.complete(
+    "gpt-4o-mini",
+    messages,
+    max_tokens=64,
+    response_format="json",
+    heal=True,
+)
 print(healed.content)
 print(healed.was_healed, healed.confidence)
+print(healed.usage.get("total_tokens"))
 ```
 
 ### Tool Calling
@@ -135,7 +180,7 @@ tools = [
     }
 ]
 messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
-response = client.complete_with_tools("gpt-4o-mini", messages, tools)
+response = client.complete("gpt-4o-mini", messages, tools=tools)
 print(response.tool_calls)
 ```
 
@@ -157,7 +202,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 +213,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 +223,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 +238,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 +252,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 +266,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 +277,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.2.0}/crates/simple-agents-cache/Cargo.toml

@@ -11,7 +11,7 @@ documentation.workspace = true
 description = "Caching utilities for SimpleAgents"
 
 [dependencies]
-simple-agent-type = { path = "../simple-agent-type", version = "0.1.0" }
+simple-agent-type = { workspace = true }
 tokio = { workspace = true, features = ["sync", "time"] }
 async-trait.workspace = true
 

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

@@ -17,10 +17,10 @@ serde.workspace = true
 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-agent-type = { workspace = true }
+simple-agents-router = { path = "../simple-agents-router", version = "0.2.0" }
+simple-agents-cache = { path = "../simple-agents-cache", version = "0.2.0" }
+simple-agents-healing = { path = "../simple-agents-healing", version = "0.2.0" }
 futures-core = "0.3"
 
 [dev-dependencies]