fix up types

Daniel OBrien · Daniel OBrien · commit 60f735354487 · 2024-11-07T09:09:15.000-05:00
diff --git a/ldai/types.py b/ldai/types.py
@@ -59,6 +59,7 @@ def to_metrics(self) -> TokenMetrics:
             'output': self.completion_tokens,
         }
  
+@dataclass
 class BedrockTokenUsage:
     def __init__(self, data: dict):
         self.totalTokens = data.get('totalTokens', 0)

Original file line number	Diff line number	Diff line change
`@@ -59,6 +59,7 @@ def to_metrics(self) -> TokenMetrics:`
`59`	`59`	`'output': self.completion_tokens,`
`60`	`60`	`}`
`61`	`61`
	`62`	`+@dataclass`
`62`	`63`	`class BedrockTokenUsage:`
`63`	`64`	`def __init__(self, data: dict):`
`64`	`65`	`self.totalTokens = data.get('totalTokens', 0)`