feat(core+web): store raw usage data on streams (and visualize properly in ADE) (#6452)
* feat(core): store raw usage data on streams * fix(web): various fixes to deal w/ hardcoding against openai
This commit is contained in:
committed by
Caren Thomas
parent
88a3743cc8
commit
4af6465226
@@ -234,7 +234,10 @@ class SimpleLLMStreamAdapter(LettaLLMStreamAdapter):
|
||||
# "stop_reason": "",
|
||||
# "stop_sequence": None,
|
||||
"type": "message",
|
||||
"usage": {
|
||||
# Use raw_usage if available for transparent provider trace logging, else fallback
|
||||
"usage": self.interface.raw_usage
|
||||
if hasattr(self.interface, "raw_usage") and self.interface.raw_usage
|
||||
else {
|
||||
"input_tokens": self.usage.prompt_tokens,
|
||||
"output_tokens": self.usage.completion_tokens,
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user