Update models.py

Author: ysymyth

src/tot/models.py

@@ -41,5 +41,5 @@ def gpt_usage(backend="gpt-4"):
  if backend == "gpt-4":
  cost = completion_tokens / 1000 * 0.06 + prompt_tokens / 1000 * 0.03
  elif backend == "gpt-3.5-turbo":
- cost = (completion_tokens + prompt_tokens) / 1000 * 0.0002
- return {"completion_tokens": completion_tokens, "prompt_tokens": prompt_tokens, "cost": cost}
+ cost = completion_tokens / 1000 * 0.002 + prompt_tokens / 1000 * 0.0015
+ return {"completion_tokens": completion_tokens, "prompt_tokens": prompt_tokens, "cost": cost}