fix: missing model name in response; 404 for model not found (#6)

traceloop · Nov 12, 2024 · dad45aa · dad45aa
1 parent b0900f4
commit dad45aa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/models/chat.rs b/src/models/chat.rs
@@ -42,6 +42,7 @@ pub struct ChatCompletionResponse {
     pub id: String,
     pub object: String,
     pub created: u64,
+    pub model: String,
     pub choices: Vec<ChatCompletionChoice>,
     pub usage: Usage,
 }

diff --git a/src/pipelines/pipeline.rs b/src/pipelines/pipeline.rs
@@ -53,7 +53,7 @@ pub async fn chat_completions(
         }
     }
 
-    Err(StatusCode::SERVICE_UNAVAILABLE)
+    Err(StatusCode::NOT_FOUND)
 }
 
 pub async fn completions(
@@ -71,7 +71,7 @@ pub async fn completions(
         }
     }
 
-    Err(StatusCode::SERVICE_UNAVAILABLE)
+    Err(StatusCode::NOT_FOUND)
 }
 
 pub async fn embeddings(
@@ -89,5 +89,5 @@ pub async fn embeddings(
         }
     }
 
-    Err(StatusCode::SERVICE_UNAVAILABLE)
+    Err(StatusCode::NOT_FOUND)
 }