Skip to content

Commit 8bafc6d

Browse files
committed
fix: model params types
1 parent 8f29b81 commit 8bafc6d

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

parea/schemas/log.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@ def to_dict(self) -> Dict[str, str]:
3333

3434
@define
3535
class ModelParams:
36-
temp: float = 1.0
37-
top_p: float = 1.0
38-
frequency_penalty: float = 0.0
39-
presence_penalty: float = 0.0
36+
temp: Union[float, None] = 1.0
37+
top_p: Union[float, None] = 1.0
38+
frequency_penalty: Union[float, None] = 0.0
39+
presence_penalty: Union[float, None] = 0.0
4040
max_length: Optional[int] = None
4141
response_format: Optional[dict] = None
4242
safe_prompt: Optional[bool] = None

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
66
[tool.poetry]
77
name = "parea-ai"
88
packages = [{ include = "parea" }]
9-
version = "0.2.172"
9+
version = "0.2.173"
1010
description = "Parea python sdk"
1111
readme = "README.md"
1212
authors = ["joel-parea-ai <[email protected]>"]

0 commit comments

Comments
 (0)