fix: Cost calculation enhancement (#207)

2024-10-31 13:58:51 +11:00
parent 0ab1966e93
commit aa324ce507
6 changed files with 89 additions and 27 deletions
@@ -2,18 +2,19 @@ import json
 import traceback
 from copy import deepcopy
 from typing import Mapping
-from attrs import define, evolve, field, Factory
-from exchange.langfuse_wrapper import observe_wrapper
+
+from attrs import Factory, define, evolve, field
 from tiktoken import get_encoding

 from exchange.checkpoint import Checkpoint, CheckpointData
 from exchange.content import Text, ToolResult, ToolUse
+from exchange.langfuse_wrapper import observe_wrapper
 from exchange.message import Message
 from exchange.moderators import Moderator
 from exchange.moderators.truncate import ContextTruncate
 from exchange.providers import Provider, Usage
-from exchange.tool import Tool
 from exchange.token_usage_collector import _token_usage_collector
+from exchange.tool import Tool


 def validate_tool_output(output: str) -> None: