Update api/core/llm_generator/llm_generator.py

pull/18646/head
crazywoola 1 year ago committed by GitHub
parent a1adab53ee
commit 8f144e3817
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -49,7 +49,7 @@ class LLMGenerator:
response = cast(
LLMResult,
model_instance.invoke_llm(
prompt_messages=list(prompts), model_parameters={"max_tokens": 1000, "temperature": 1}, stream=False
prompt_messages=list(prompts), model_parameters={"max_tokens": 500, "temperature": 1}, stream=False
),
)
answer = cast(str, response.message.content)

Loading…
Cancel
Save