Skip to content

Commit 7f8cfed

Browse files
Adding max_tokens to autogenstudio LLMConfig (#1574)
* Adding max_tokens to autogenstudio LLMConfig * fix formatting issue * fix formatting issues --------- Co-authored-by: Victor Dibia <[email protected]>
1 parent 620cc0c commit 7f8cfed

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

samples/apps/autogen-studio/autogenstudio/datamodel.py

+1
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ class LLMConfig:
9292
temperature: float = 0
9393
cache_seed: Optional[Union[int, None]] = None
9494
timeout: Optional[int] = None
95+
max_tokens: Optional[int] = None
9596

9697
def dict(self):
9798
result = asdict(self)

0 commit comments

Comments
 (0)