fix: title generation issue

This commit is contained in:
Timothy Jaeryang Baek 2024-12-25 16:46:49 -07:00
parent 798886105e
commit 23bf71022e

View File

@ -613,9 +613,8 @@ async def generate_chat_completion(
payload["max_tokens"] = payload["max_completion_tokens"]
del payload["max_completion_tokens"]
# TODO: check if below is needed
# if "max_tokens" in payload and "max_completion_tokens" in payload:
# del payload["max_tokens"]
if "max_tokens" in payload and "max_completion_tokens" in payload:
del payload["max_tokens"]
# Convert the modified body back to JSON
payload = json.dumps(payload)