@@ -75,11 +75,12 @@ use inference_gateway_sdk::{
7575};
7676use log :: info;
7777
78+ #[tokio:: main]
7879fn 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
106107use 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 ? ;
110111let models = resp . models;
111112info! (" Provider: {:?}" , resp . provider);
112113for 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
135136info! (" Provider: {:?}" , response . provider);
136137info! (" Response: {:?}" , response . response);
0 commit comments