code cleanup

This commit is contained in:
Krrish Dholakia 2023-08-07 10:44:02 -07:00
parent 10bb5f80df
commit 5e564ef5de
5 changed files with 16 additions and 3 deletions

View file

@ -229,7 +229,7 @@ def completion(
prompt_tokens = anthropic.count_tokens(prompt)
completion_tokens = anthropic.count_tokens(completion_response)
## RESPONSE OBJECT
print(f"model_response: {model_response}")
print_verbose(f"raw model_response: {model_response}")
model_response["choices"][0]["message"]["content"] = completion_response
model_response["created"] = time.time()
model_response["model"] = model