simple-agents-py 0.1.2__tar.gz → 0.1.3__tar.gz

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/Cargo.lock

@@ -2025,7 +2025,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agent-type"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "async-trait",
  "blake3",
@@ -2040,7 +2040,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-cache"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "async-trait",
  "simple-agent-type",
@@ -2066,7 +2066,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-core"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "async-trait",
  "serde",
@@ -2172,9 +2172,10 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-py"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "pyo3",
+ "reqwest",
  "simple-agent-type",
  "simple-agents-core",
  "simple-agents-providers",
@@ -2183,7 +2184,7 @@ dependencies = [
 
 [[package]]
 name = "simple-agents-router"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "async-trait",
  "rand 0.8.5",

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/Cargo.toml

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

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simple-agents-py
-Version: 0.1.2
+Version: 0.1.3
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Rust
 Classifier: License :: OSI Approved :: MIT License

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/crates/simple-agents-py/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "simple-agents-py"
-version = "0.1.2"
+version = "0.1.3"
 edition = "2021"
 license = "MIT OR Apache-2.0"
 readme = "README.md"
@@ -12,6 +12,7 @@ crate-type = ["cdylib"]
 [dependencies]
 pyo3 = { version = "0.22", features = ["extension-module", "abi3-py312"] }
 tokio = { workspace = true }
+reqwest = { workspace = true }
 
 simple-agents-core = { path = "../simple-agents-core" }
 simple-agents-providers = { path = "../simple-agents-providers" }

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/crates/simple-agents-py/src/lib.rs

@@ -4,6 +4,7 @@
 
 use pyo3::exceptions::PyRuntimeError;
 use pyo3::prelude::*;
+use reqwest::Client as HttpClient;
 use simple_agents_core::{SimpleAgentsClient, SimpleAgentsClientBuilder};
 use simple_agents_providers::anthropic::AnthropicProvider;
 use simple_agents_providers::openai::OpenAIProvider;
@@ -11,6 +12,7 @@ use simple_agents_providers::openrouter::OpenRouterProvider;
 use simple_agent_type::message::Message;
 use simple_agent_type::prelude::{ApiKey, CompletionRequest, Provider, Result, SimpleAgentsError};
 use std::sync::{Arc, Mutex};
+use std::time::Duration;
 
 type Runtime = tokio::runtime::Runtime;
 
@@ -34,7 +36,25 @@ fn provider_from_params(
         "openai" => {
             let provider = match api_key {
                 Some(api_key) => match api_base {
-                    Some(api_base) => OpenAIProvider::with_base_url(api_key, api_base.to_string())?,
+                    Some(api_base) => {
+                        if is_local_base(api_base) {
+                            let client = HttpClient::builder()
+                                .timeout(Duration::from_secs(30))
+                                .pool_max_idle_per_host(10)
+                                .pool_idle_timeout(Duration::from_secs(90))
+                                .no_proxy()
+                                .build()
+                                .map_err(|e| {
+                                    SimpleAgentsError::Config(format!(
+                                        "Failed to create HTTP client: {}",
+                                        e
+                                    ))
+                                })?;
+                            OpenAIProvider::with_client(api_key, api_base.to_string(), client)?
+                        } else {
+                            OpenAIProvider::with_base_url(api_key, api_base.to_string())?
+                        }
+                    }
                     None => OpenAIProvider::new(api_key)?,
                 },
                 None => OpenAIProvider::from_env()?,
@@ -71,6 +91,10 @@ fn provider_from_params(
     }
 }
 
+fn is_local_base(api_base: &str) -> bool {
+    api_base.contains("localhost") || api_base.contains("127.0.0.1")
+}
+
 fn build_request(
     model: &str,
     prompt: &str,

{simple_agents_py-0.1.2 → simple_agents_py-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "simple-agents-py"
-version = "0.1.2"
+version = "0.1.3"
 description = "Python bindings for SimpleAgents"
 readme = "README.md"
 requires-python = ">=3.9"