github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Adding max_tokens to autogenstudio LLMConfig (#1574)

Browse Source 
* Adding max_tokens to autogenstudio LLMConfig

* fix formatting issue

* fix formatting issues

---------

Co-authored-by: Victor Dibia <victordibia@microsoft.com>
This commit is contained in:
Aaron Mihalik
2024-02-07 17:14:24 -05:00
committed by GitHub
parent 620cc0ca59
commit 7f8cfed661
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
samples/apps/autogen-studio/autogenstudio/datamodel.py
View File

@@ -92,6 +92,7 @@ class LLMConfig:
temperature: float = 0
cache_seed: Optional[Union[int, None]] = None
timeout: Optional[int] = None
max_tokens: Optional[int] = None
def dict(self):
result = asdict(self)