Skip to content

Commit 80f50ee

Browse files
committed
more openrouter config typesafety
1 parent b49607b commit 80f50ee

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

crates/llm-proxy/src/provider/openrouter.rs

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,7 @@
1-
use hypr_openrouter::{Client as OpenRouterClient, Error as OpenRouterError};
1+
use hypr_openrouter::{
2+
Client as OpenRouterClient, Error as OpenRouterError, ProviderPreferences, ProviderSort,
3+
ProviderSortUnion,
4+
};
25
use reqwest::Client;
36
use serde::Deserialize;
47

@@ -53,12 +56,18 @@ impl Provider for OpenRouterProvider {
5356
let mut body = serde_json::to_value(request)?;
5457
let obj = body.as_object_mut().unwrap();
5558

59+
let provider_prefs = ProviderPreferences {
60+
sort: Some(ProviderSortUnion::Simple(ProviderSort::Latency)),
61+
preferred_min_throughput: None,
62+
..Default::default()
63+
};
64+
5665
obj.remove("model");
5766
obj.insert("models".to_string(), serde_json::to_value(models)?);
5867
obj.insert("stream".to_string(), serde_json::Value::Bool(stream));
5968
obj.insert(
6069
"provider".to_string(),
61-
serde_json::json!({"sort": "latency"}),
70+
serde_json::to_value(provider_prefs)?,
6271
);
6372

6473
Ok(body)

0 commit comments

Comments
 (0)