Skip to content

Commit b99d520

Browse files
committed
chore: Update dependencies and refactor main function for async support
Signed-off-by: Eden Reich <eden.reich@gmail.com>
1 parent 3da7e50 commit b99d520

File tree

4 files changed

+105
-77
lines changed

4 files changed

+105
-77
lines changed

Cargo.lock

Lines changed: 13 additions & 12 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,12 @@ keywords = ["ai", "llm", "inference", "sdk", "gateway"]
1212
categories = ["api-bindings", "web-programming::http-client"]
1313

1414
[dependencies]
15-
reqwest = { version = "0.12.12", features = ["blocking", "json"] }
15+
reqwest = { version = "0.12.12", features = ["json"] }
1616
serde = { version = "1.0.217", features = ["derive"] }
1717
serde_json = "1.0.138"
1818
thiserror = "2.0.11"
19+
tokio = { version = "1.43.0", features = ["macros", "rt-multi-thread"] }
1920

2021
[dev-dependencies]
2122
mockito = "1.6.1"
23+
tokio = { version = "1.43.0", features = ["macros", "rt"] }

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,11 +75,12 @@ use inference_gateway_sdk::{
7575
};
7676
use log::info;
7777

78+
#[tokio::main]
7879
fn main() -> Result<(), GatewayError> {
7980
// ...create a client
8081

8182
// List all providers and models
82-
let models = client.list_models()?;
83+
let models = client.list_models().await?;
8384
for provider_models in models {
8485
info!("Provider: {:?}", provider_models.provider);
8586
for model in provider_models.models {
@@ -88,7 +89,7 @@ fn main() -> Result<(), GatewayError> {
8889
}
8990

9091
// List models for a specific provider
91-
let resp = client.list_models_by_provider(Provider::Ollama)?;
92+
let resp = client.list_models_by_provider(Provider::Ollama).await?;
9293
let models = resp.models;
9394
info!("Provider: {:?}", resp.provider);
9495
for model in models {
@@ -106,7 +107,7 @@ To list all available models from a specific provider, use the `list_models_by_p
106107
use log::info;
107108

108109
// ...main function
109-
let resp = client.list_models_by_provider(Provider::Ollama)?;
110+
let resp = client.list_models_by_provider(Provider::Ollama).await?;
110111
let models = resp.models;
111112
info!("Provider: {:?}", resp.provider);
112113
for model in models {
@@ -130,7 +131,7 @@ let response = client.generate_content(
130131
role: MessageRole::User,
131132
content: "Tell me a joke".to_string(),
132133
}],
133-
)?;
134+
).await?;
134135

135136
info!("Provider: {:?}", response.provider);
136137
info!("Response: {:?}", response.response);

0 commit comments

Comments
 (0)