fix server types for client gen

Stonelinks · Stonelinks · commit f50b87f1490a · 2023-05-22T13:36:28.000-05:00
diff --git a/llama_cpp/server/app.py b/llama_cpp/server/app.py
@@ -191,7 +191,7 @@ def get_settings():
 
 
 class CreateCompletionRequest(BaseModel):
-    prompt: Union[str, List[str]] = Field(
+    prompt: str = Field(
         default="", description="The prompt to generate completions for."
     )
     suffix: Optional[str] = Field(
@@ -276,7 +276,7 @@ async def server_sent_events(
 
 class CreateEmbeddingRequest(BaseModel):
     model: Optional[str] = model_field
-    input: Union[str, List[str]] = Field(description="The input to embed.")
+    input: str = Field(description="The input to embed.")
     user: Optional[str]
 
     class Config:
