From 91062a834f3b586beb58fe5ef20bb637ff8c3c27 Mon Sep 17 00:00:00 2001
From: lazymio <mio@lazym.io>
Date: Mon, 24 Feb 2025 21:38:01 +0800
Subject: [PATCH] Default values

---
 ktransformers/server/schemas/endpoints/chat.py     | 7 +++----
 ktransformers/server/schemas/legacy/completions.py | 5 ++---
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/ktransformers/server/schemas/endpoints/chat.py b/ktransformers/server/schemas/endpoints/chat.py
index 5507266..821b3b9 100644
--- a/ktransformers/server/schemas/endpoints/chat.py
+++ b/ktransformers/server/schemas/endpoints/chat.py
@@ -25,10 +25,9 @@ class ChatCompletionCreate(BaseModel):
     messages: List[Message]
     model : str
     stream : bool = False
-    temperature: Optional[float]
-    top_p: Optional[float]
-    frequency_penalty: Optional[float]
-
+    temperature: Optional[float] = None
+    top_p: Optional[float] = None
+    
     def get_tokenizer_messages(self):
         return [m.to_tokenizer_message() for m in self.messages]
 
diff --git a/ktransformers/server/schemas/legacy/completions.py b/ktransformers/server/schemas/legacy/completions.py
index ca4b89c..ea936ea 100644
--- a/ktransformers/server/schemas/legacy/completions.py
+++ b/ktransformers/server/schemas/legacy/completions.py
@@ -9,9 +9,8 @@ class CompletionCreate(BaseModel):
     model: str
     prompt: str | List[str]
     stream: bool = False
-    temperature: Optional[float]
-    top_p: Optional[float]
-    frequency_penalty: Optional[float]
+    temperature: Optional[float] = None
+    top_p: Optional[float] = None
 
     def get_tokenizer_messages(self):
         if isinstance(self.prompt,List):