From 6c7aad11f37bc2ba274f39921d4e14e3e355ad01 Mon Sep 17 00:00:00 2001 From: hronoas Date: Wed, 8 Nov 2023 17:23:51 +0300 Subject: [PATCH] openai extension: wrong frequency_penalty type (#4512) --- extensions/openai/typing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions/openai/typing.py b/extensions/openai/typing.py index 11fd5f65..e03358d8 100644 --- a/extensions/openai/typing.py +++ b/extensions/openai/typing.py @@ -51,7 +51,7 @@ class CompletionRequestParams(BaseModel): logprobs: int | None = None max_tokens: int | None = 16 n: int | None = Field(default=1, description="Unused parameter.") - presence_penalty: int | None = 0 + presence_penalty: float | None = 0 stop: str | List[str] | None = None stream: bool | None = False suffix: str | None = None @@ -82,7 +82,7 @@ class ChatCompletionRequestParams(BaseModel): logit_bias: dict | None = None max_tokens: int | None = None n: int | None = Field(default=1, description="Unused parameter.") - presence_penalty: int | None = 0 + presence_penalty: float | None = 0 stop: str | List[str] | None = None stream: bool | None = False temperature: float | None = 1