fix benchmark.py tutorial

This commit is contained in:
ishaan-jaff 2023-09-13 11:20:12 -07:00
parent feb814a76e
commit 721d34bda2

View file

@ -51,7 +51,7 @@ def main(system_prompt):
end = time.time() end = time.time()
total_time = end - start_time total_time = end - start_time
cost = completion_cost(response) cost = completion_cost(completion_response=response)
raw_response = response['choices'][0]['message']['content'] raw_response = response['choices'][0]['message']['content']
data.append({ data.append({