From d71a7dc18aa14b9d7b94f77ac39b00bba742e249 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Wed, 3 Jul 2024 10:55:30 +0000
Subject: [PATCH] Fixing docs by re-adding missing `Prompt`.

---
 docs/openapi.json    | 6 ++++++
 router/src/server.rs | 3 ++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/docs/openapi.json b/docs/openapi.json
index b2002efe..5e0399e0 100644
--- a/docs/openapi.json
+++ b/docs/openapi.json
@@ -1734,6 +1734,12 @@
           }
         }
       },
+      "Prompt": {
+        "type": "array",
+        "items": {
+          "type": "string"
+        }
+      },
       "SimpleToken": {
         "type": "object",
         "required": [
diff --git a/router/src/server.rs b/router/src/server.rs
index ce0d6675..db8b16ad 100644
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -20,7 +20,7 @@ use crate::{
     ChatCompletion, ChatCompletionChoice, ChatCompletionChunk, ChatCompletionComplete,
     ChatCompletionDelta, ChatCompletionLogprob, ChatCompletionLogprobs, ChatCompletionTopLogprob,
     ChatRequest, Chunk, CompatGenerateRequest, Completion, CompletionComplete, CompletionFinal,
-    CompletionRequest, CompletionType, DeltaToolCall, Function, Tool, VertexRequest,
+    CompletionRequest, CompletionType, DeltaToolCall, Function, Prompt, Tool, VertexRequest,
     VertexResponse,
 };
 use crate::{FunctionDefinition, HubPreprocessorConfig, ToolCall, ToolType};
@@ -1467,6 +1467,7 @@ pub async fn run(
     Chunk,
     Completion,
     CompletionFinal,
+    Prompt,
     GenerateParameters,
     PrefillToken,
     Token,