Fix model name.

Ai00-X · Oct 22, 2023 · cc5b776 · cc5b776
1 parent 4547344
commit cc5b776
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/oai/models.rs b/src/oai/models.rs
@@ -18,12 +18,17 @@ pub struct ModelResponse {
 
 pub async fn models(State(ThreadState(sender)): State<ThreadState>) -> Json<ModelResponse> {
     let info = request_info(sender, Duration::from_secs(1)).await;
-    let model_name = info.reload.model_path.to_string_lossy().into();
+    let model_name = info
+        .reload
+        .model_path
+        .file_stem()
+        .map(|stem| stem.to_string_lossy())
+        .unwrap_or_default();
 
     Json(ModelResponse {
         data: vec![ModelChoice {
             object: "models".into(),
-            id: model_name,
+            id: model_name.into(),
         }],
     })
 }