Enable runtime access to provider name (#5399)
This commit is contained in:
@@ -23,6 +23,8 @@ pub struct LiteLLMProvider {
|
||||
api_client: ApiClient,
|
||||
base_path: String,
|
||||
model: ModelConfig,
|
||||
#[serde(skip)]
|
||||
name: String,
|
||||
}
|
||||
|
||||
impl LiteLLMProvider {
|
||||
@@ -67,6 +69,7 @@ impl LiteLLMProvider {
|
||||
api_client,
|
||||
base_path,
|
||||
model,
|
||||
name: Self::metadata().name,
|
||||
})
|
||||
}
|
||||
|
||||
@@ -154,6 +157,10 @@ impl Provider for LiteLLMProvider {
|
||||
)
|
||||
}
|
||||
|
||||
fn get_name(&self) -> &str {
|
||||
&self.name
|
||||
}
|
||||
|
||||
fn get_model_config(&self) -> ModelConfig {
|
||||
self.model.clone()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user