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

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

@@ -1494,7 +1494,6 @@ dependencies = [
  "pyo3-build-config",
  "pyo3-ffi",
  "pyo3-macros",
- "serde",
  "unindent",
 ]
 
@@ -1543,6 +1542,16 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "pythonize"
+version = "0.22.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "90fcf491425978bd889015d5430f6473d91bdfa2097262f1e731aadcf6c2113e"
+dependencies = [
+ "pyo3",
+ "serde",
+]
+
 [[package]]
 name = "quanta"
 version = "0.12.6"
@@ -2026,7 +2035,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agent-type"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
  "blake3",
@@ -2041,7 +2050,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cache"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2050,7 +2059,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cli"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "clap",
  "serde",
@@ -2067,9 +2076,10 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-core"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
+ "futures-core",
  "serde",
  "serde_json",
  "simple-agent-type",
@@ -2094,7 +2104,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-ffi"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2105,7 +2115,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-healing"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "criterion",
  "proptest",
@@ -2121,7 +2131,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-macros"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -2133,7 +2143,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-napi"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "napi",
  "napi-derive",
@@ -2145,7 +2155,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-providers"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
  "bytes",
@@ -2173,22 +2183,29 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-py"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
+ "async-trait",
+ "futures-util",
  "pyo3",
+ "pythonize",
  "reqwest",
  "serde_json",
  "simple-agent-type",
+ "simple-agents-cache",
  "simple-agents-core",
+ "simple-agents-healing",
  "simple-agents-providers",
+ "simple-agents-router",
  "tokio",
 ]
 
 [[package]]
 name = "simple-agents-router"
-version = "0.1.10"
+version = "0.1.14"
 dependencies = [
  "async-trait",
+ "futures-core",
  "rand 0.8.5",
  "serde_json",
  "simple-agent-type",

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simple-agents-py
-Version: 0.1.10
+Version: 0.1.14
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Rust
 Classifier: License :: OSI Approved :: MIT License
@@ -60,6 +60,121 @@ response = client.complete("gpt-4", "Hello from Python!", max_tokens=128, temper
 print(response)
 ```
 
+## Feature Guide
+
+### Streaming
+
+```python
+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):
+    if chunk.content:
+        print(chunk.content, end="", flush=True)
+print()
+```
+
+### Structured Output (JSON Mode)
+
+```python
+from simple_agents_py import Client
+import json
+
+client = Client("openai")
+schema = {
+    "type": "object",
+    "properties": {
+        "name": {"type": "string"},
+        "age": {"type": "number"},
+    },
+    "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")
+print(json.loads(json_text))
+```
+
+### Structured Streaming
+
+```python
+from simple_agents_py import Client
+
+client = Client("openai")
+schema = {
+    "type": "object",
+    "properties": {"name": {"type": "string"}, "age": {"type": "number"}},
+    "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):
+    if event.is_partial:
+        print("partial:", event.partial_value)
+    else:
+        print("complete:", event.value)
+```
+
+### Response Healing
+
+```python
+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)
+print(healed.content)
+print(healed.was_healed, healed.confidence)
+```
+
+### Tool Calling
+
+```python
+from simple_agents_py import Client
+
+client = Client("openai")
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "get_weather",
+            "description": "Get the weather for a city",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "city": {"type": "string"},
+                    "unit": {"type": "string", "enum": ["c", "f"]},
+                },
+                "required": ["city"],
+            },
+        },
+    }
+]
+messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
+response = client.complete_with_tools("gpt-4o-mini", messages, tools)
+print(response.tool_calls)
+```
+
+### ClientBuilder (Routing, Cache, Healing, Middleware)
+
+```python
+from simple_agents_py import ClientBuilder
+
+class TimingMiddleware:
+    def before_request(self, request):
+        print("sending", request.model)
+
+builder = (
+    ClientBuilder()
+    .add_provider("openai", api_key="sk-...")
+    .with_routing("direct")
+    .with_cache(ttl_seconds=60)
+    .with_healing_config({"enabled": True, "min_confidence": 0.7})
+    .add_middleware(TimingMiddleware())
+)
+client = builder.build()
+print(client.complete("gpt-4o-mini", "Give me one idea."))
+```
+
 ## Examples
 
 OpenAI with a short prompt:

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

@@ -45,6 +45,121 @@ response = client.complete("gpt-4", "Hello from Python!", max_tokens=128, temper
 print(response)
 ```
 
+## Feature Guide
+
+### Streaming
+
+```python
+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):
+    if chunk.content:
+        print(chunk.content, end="", flush=True)
+print()
+```
+
+### Structured Output (JSON Mode)
+
+```python
+from simple_agents_py import Client
+import json
+
+client = Client("openai")
+schema = {
+    "type": "object",
+    "properties": {
+        "name": {"type": "string"},
+        "age": {"type": "number"},
+    },
+    "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")
+print(json.loads(json_text))
+```
+
+### Structured Streaming
+
+```python
+from simple_agents_py import Client
+
+client = Client("openai")
+schema = {
+    "type": "object",
+    "properties": {"name": {"type": "string"}, "age": {"type": "number"}},
+    "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):
+    if event.is_partial:
+        print("partial:", event.partial_value)
+    else:
+        print("complete:", event.value)
+```
+
+### Response Healing
+
+```python
+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)
+print(healed.content)
+print(healed.was_healed, healed.confidence)
+```
+
+### Tool Calling
+
+```python
+from simple_agents_py import Client
+
+client = Client("openai")
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "get_weather",
+            "description": "Get the weather for a city",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "city": {"type": "string"},
+                    "unit": {"type": "string", "enum": ["c", "f"]},
+                },
+                "required": ["city"],
+            },
+        },
+    }
+]
+messages = [{"role": "user", "content": "What's the weather in Tokyo?"}]
+response = client.complete_with_tools("gpt-4o-mini", messages, tools)
+print(response.tool_calls)
+```
+
+### ClientBuilder (Routing, Cache, Healing, Middleware)
+
+```python
+from simple_agents_py import ClientBuilder
+
+class TimingMiddleware:
+    def before_request(self, request):
+        print("sending", request.model)
+
+builder = (
+    ClientBuilder()
+    .add_provider("openai", api_key="sk-...")
+    .with_routing("direct")
+    .with_cache(ttl_seconds=60)
+    .with_healing_config({"enabled": True, "min_confidence": 0.7})
+    .add_middleware(TimingMiddleware())
+)
+client = builder.build()
+print(client.complete("gpt-4o-mini", "Give me one idea."))
+```
+
 ## Examples
 
 OpenAI with a short prompt:

{simple_agents_py-0.1.10 → simple_agents_py-0.1.14}/crates/simple-agent-type/src/lib.rs

@@ -58,6 +58,7 @@ pub mod provider;
 pub mod request;
 pub mod response;
 pub mod router;
+pub mod tool;
 pub mod validation;
 
 // Re-export commonly used types at crate root
@@ -101,6 +102,12 @@ pub mod prelude {
     // Coercion
     pub use crate::coercion::{CoercionFlag, CoercionResult};
 
+    // Tool calling
+    pub use crate::tool::{
+        ToolCall, ToolCallFunction, ToolChoice, ToolChoiceFunction, ToolChoiceMode, ToolChoiceTool,
+        ToolDefinition, ToolFunction, ToolType,
+    };
+
     // Traits
     pub use crate::cache::Cache;
     pub use crate::provider::Provider;

{simple_agents_py-0.1.10 → simple_agents_py-0.1.14}/crates/simple-agent-type/src/message.rs

@@ -4,6 +4,8 @@
 
 use serde::{Deserialize, Serialize};
 
+use crate::tool::ToolCall;
+
 /// Role of a message in a conversation.
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
 #[serde(rename_all = "lowercase")]
@@ -32,6 +34,9 @@ pub struct Message {
     /// Tool call ID (for tool role messages)
     #[serde(skip_serializing_if = "Option::is_none")]
     pub tool_call_id: Option<String>,
+    /// Tool calls emitted by the assistant.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tool_calls: Option<Vec<ToolCall>>,
 }
 
 impl Message {
@@ -51,6 +56,7 @@ impl Message {
             content: content.into(),
             name: None,
             tool_call_id: None,
+            tool_calls: None,
         }
     }
 
@@ -69,6 +75,7 @@ impl Message {
             content: content.into(),
             name: None,
             tool_call_id: None,
+            tool_calls: None,
         }
     }
 
@@ -87,6 +94,7 @@ impl Message {
             content: content.into(),
             name: None,
             tool_call_id: None,
+            tool_calls: None,
         }
     }
 
@@ -106,6 +114,7 @@ impl Message {
             content: content.into(),
             name: None,
             tool_call_id: Some(tool_call_id.into()),
+            tool_calls: None,
         }
     }
 
@@ -122,6 +131,12 @@ impl Message {
         self.name = Some(name.into());
         self
     }
+
+    /// Set tool calls for assistant messages.
+    pub fn with_tool_calls(mut self, tool_calls: Vec<ToolCall>) -> Self {
+        self.tool_calls = Some(tool_calls);
+        self
+    }
 }
 
 #[cfg(test)]
@@ -135,6 +150,7 @@ mod tests {
         assert_eq!(msg.content, "test");
         assert_eq!(msg.name, None);
         assert_eq!(msg.tool_call_id, None);
+        assert_eq!(msg.tool_calls, None);
     }
 
     #[test]
@@ -142,6 +158,7 @@ mod tests {
         let msg = Message::assistant("response");
         assert_eq!(msg.role, Role::Assistant);
         assert_eq!(msg.content, "response");
+        assert_eq!(msg.tool_calls, None);
     }
 
     #[test]
@@ -149,6 +166,7 @@ mod tests {
         let msg = Message::system("instruction");
         assert_eq!(msg.role, Role::System);
         assert_eq!(msg.content, "instruction");
+        assert_eq!(msg.tool_calls, None);
     }
 
     #[test]
@@ -157,6 +175,7 @@ mod tests {
         assert_eq!(msg.role, Role::Tool);
         assert_eq!(msg.content, "result");
         assert_eq!(msg.tool_call_id, Some("call_123".to_string()));
+        assert_eq!(msg.tool_calls, None);
     }
 
     #[test]
@@ -194,6 +213,7 @@ mod tests {
         let json = serde_json::to_value(&msg).unwrap();
         assert!(json.get("name").is_none());
         assert!(json.get("tool_call_id").is_none());
+        assert!(json.get("tool_calls").is_none());
     }
 
     #[test]

{simple_agents_py-0.1.10 → simple_agents_py-0.1.14}/crates/simple-agent-type/src/provider.rs

@@ -3,11 +3,12 @@
 //! Defines the interface for LLM providers with transformation hooks.
 
 use crate::config::{Capabilities, RetryConfig};
-use crate::error::{ProviderError, Result, SimpleAgentsError};
+use crate::error::Result;
 use crate::request::CompletionRequest;
 use crate::response::{CompletionChunk, CompletionResponse};
 use async_trait::async_trait;
 use serde::{Deserialize, Serialize};
+use serde_json::Value;
 use std::borrow::Cow;
 use std::time::Duration;
 
@@ -236,11 +237,55 @@ pub trait Provider: Send + Sync {
     /// ```
     async fn execute_stream(
         &self,
-        _req: ProviderRequest,
+        mut req: ProviderRequest,
     ) -> Result<Box<dyn futures_core::Stream<Item = Result<CompletionChunk>> + Send + Unpin>> {
-        Err(SimpleAgentsError::Provider(
-            ProviderError::UnsupportedFeature("streaming".to_string()),
-        ))
+        if let Value::Object(map) = &mut req.body {
+            if let Some(stream_value) = map.get_mut("stream") {
+                *stream_value = Value::Bool(false);
+            }
+        }
+
+        let provider_response = self.execute(req).await?;
+        let response = self.transform_response(provider_response)?;
+
+        struct SingleChunkStream {
+            chunk: Option<Result<CompletionChunk>>,
+        }
+
+        impl futures_core::Stream for SingleChunkStream {
+            type Item = Result<CompletionChunk>;
+
+            fn poll_next(
+                mut self: std::pin::Pin<&mut Self>,
+                _cx: &mut std::task::Context<'_>,
+            ) -> std::task::Poll<Option<Self::Item>> {
+                std::task::Poll::Ready(self.chunk.take())
+            }
+        }
+
+        let choices = response
+            .choices
+            .into_iter()
+            .map(|choice| crate::response::ChoiceDelta {
+                index: choice.index,
+                delta: crate::response::MessageDelta {
+                    role: Some(choice.message.role),
+                    content: Some(choice.message.content),
+                },
+                finish_reason: Some(choice.finish_reason),
+            })
+            .collect();
+
+        let chunk = CompletionChunk {
+            id: response.id,
+            model: response.model,
+            choices,
+            created: response.created,
+        };
+
+        Ok(Box::new(SingleChunkStream {
+            chunk: Some(Ok(chunk)),
+        }))
     }
 }
 

{simple_agents_py-0.1.10 → simple_agents_py-0.1.14}/crates/simple-agent-type/src/request.rs

@@ -4,6 +4,7 @@
 
 use crate::error::{Result, ValidationError};
 use crate::message::Message;
+use crate::tool::{ToolChoice, ToolDefinition};
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
 
@@ -72,6 +73,12 @@ pub struct CompletionRequest {
     /// Response format for structured outputs (OpenAI only)
     #[serde(skip_serializing_if = "Option::is_none")]
     pub response_format: Option<ResponseFormat>,
+    /// Tool definitions for tool calling.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tools: Option<Vec<ToolDefinition>>,
+    /// Tool choice configuration.
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tool_choice: Option<ToolChoice>,
 }
 
 impl CompletionRequest {
@@ -237,6 +244,8 @@ pub struct CompletionRequestBuilder {
     frequency_penalty: Option<f32>,
     user: Option<String>,
     response_format: Option<ResponseFormat>,
+    tools: Option<Vec<ToolDefinition>>,
+    tool_choice: Option<ToolChoice>,
 }
 
 impl CompletionRequestBuilder {
@@ -318,6 +327,18 @@ impl CompletionRequestBuilder {
         self
     }
 
+    /// Set tool definitions for tool calling.
+    pub fn tools(mut self, tools: Vec<ToolDefinition>) -> Self {
+        self.tools = Some(tools);
+        self
+    }
+
+    /// Set tool choice configuration.
+    pub fn tool_choice(mut self, tool_choice: ToolChoice) -> Self {
+        self.tool_choice = Some(tool_choice);
+        self
+    }
+
     /// Enable JSON object mode (no schema validation).
     pub fn json_mode(mut self) -> Self {
         self.response_format = Some(ResponseFormat::JsonObject);
@@ -355,6 +376,8 @@ impl CompletionRequestBuilder {
             frequency_penalty: self.frequency_penalty,
             user: self.user,
             response_format: self.response_format,
+            tools: self.tools,
+            tool_choice: self.tool_choice,
         };
 
         request.validate()?;