Skip to content

Commit

Permalink
Print full cost info
Browse files Browse the repository at this point in the history
  • Loading branch information
shippy committed Apr 1, 2024
1 parent bb1cb94 commit 1903c5a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion init_session.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def main() -> ChatResult:
json_resume_path=args.resume,
)
try:
print("Cost: $", round(chat.cost[0]["total_cost"], 2)) # noqa: T201
print(chat.cost) # noqa: T201
except (IndexError, KeyError):
print("Unable to retrieve cost, see full array: ", chat.cost) # noqa: T201

Expand Down

0 comments on commit 1903c5a

Please sign in to comment.