mirror of
https://github.com/katanemo/plano.git
synced 2026-05-08 15:22:43 +02:00
pass model name in header when a route is selected when using usage preferences (#531)
This commit is contained in:
parent
2340a45353
commit
f819ee3507
4 changed files with 150 additions and 117 deletions
|
|
@ -16,6 +16,10 @@ pub trait RouterModel: Send + Sync {
|
|||
messages: &[Message],
|
||||
usage_preferences: &Option<Vec<ModelUsagePreference>>,
|
||||
) -> ChatCompletionsRequest;
|
||||
fn parse_response(&self, content: &str) -> Result<Option<String>>;
|
||||
fn parse_response(
|
||||
&self,
|
||||
content: &str,
|
||||
usage_preferences: &Option<Vec<ModelUsagePreference>>,
|
||||
) -> Result<Option<(String, String)>>;
|
||||
fn get_model_name(&self) -> String;
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue