Add set max_tokens attribute in OpenAIClient initialization (#330)
This commit is contained in:
parent
87fbf31a78
commit
807a402ba4
1 changed files with 2 additions and 0 deletions
|
|
@ -87,6 +87,8 @@ class OpenAIClient(LLMClient):
|
||||||
else:
|
else:
|
||||||
self.client = client
|
self.client = client
|
||||||
|
|
||||||
|
self.max_tokens = max_tokens
|
||||||
|
|
||||||
async def _generate_response(
|
async def _generate_response(
|
||||||
self,
|
self,
|
||||||
messages: list[Message],
|
messages: list[Message],
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue