From 8bafc6d46ba62163f5d118a0b95a714947b408a8 Mon Sep 17 00:00:00 2001 From: Joschka Braun Date: Mon, 10 Jun 2024 20:23:05 -0400 Subject: [PATCH] fix: model params types --- parea/schemas/log.py | 8 ++++---- pyproject.toml | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/parea/schemas/log.py b/parea/schemas/log.py index fd638c0f..8ab5f6c2 100644 --- a/parea/schemas/log.py +++ b/parea/schemas/log.py @@ -33,10 +33,10 @@ def to_dict(self) -> Dict[str, str]: @define class ModelParams: - temp: float = 1.0 - top_p: float = 1.0 - frequency_penalty: float = 0.0 - presence_penalty: float = 0.0 + temp: Union[float, None] = 1.0 + top_p: Union[float, None] = 1.0 + frequency_penalty: Union[float, None] = 0.0 + presence_penalty: Union[float, None] = 0.0 max_length: Optional[int] = None response_format: Optional[dict] = None safe_prompt: Optional[bool] = None diff --git a/pyproject.toml b/pyproject.toml index 84317d96..23ad232f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api" [tool.poetry] name = "parea-ai" packages = [{ include = "parea" }] -version = "0.2.172" +version = "0.2.173" description = "Parea python sdk" readme = "README.md" authors = ["joel-parea-ai "]