This commit is contained in:
Krrish Dholakia 2023-08-03 06:57:30 -07:00
parent 231a3e727b
commit 6ae8384fb5
14 changed files with 59 additions and 36 deletions

View file

@ -189,7 +189,7 @@ def completion(
if max_tokens != float('inf'):
max_tokens_to_sample = max_tokens
else:
max_tokens_to_sample = 300 # default in Anthropic docs https://docs.anthropic.com/claude/reference/client-libraries
max_tokens_to_sample = litellm.max_tokens # default in Anthropic docs https://docs.anthropic.com/claude/reference/client-libraries
## LOGGING
logging(model=model, input=prompt, azure=azure, additional_args={"max_tokens": max_tokens}, logger_fn=logger_fn)
## COMPLETION CALL