Merge pull request #930 from parea-ai/fix-model-params-types

fix: model params types
parea-ai · Jun 11, 2024 · a5d0198 · a5d0198
2 parents 8f29b81 + 8bafc6d
commit a5d0198
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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 <[email protected]>"]