remove eronous print

fix access of class

Author: Daniel OBrien

ldai/client.py

@@ -43,7 +43,7 @@ def model_config(self, key: str, context: Context, default_value: AIConfig, vari
         if variables:
             all_variables.update(variables)
         all_variables['ldctx'] = context
-        print(variation)
+        
         if isinstance(variation['prompt'], list) and all(isinstance(entry, dict) for entry in variation['prompt']):
             variation['prompt'] = [
                 LDMessage(

ldai/tracker.py

@@ -116,9 +116,9 @@ def track_bedrock_converse(self, res: dict) -> dict:
 
     def track_tokens(self, tokens: Union[TokenUsage, BedrockTokenUsage]) -> None:
         token_metrics = tokens.to_metrics()
-        if token_metrics['total'] > 0:
+        if token_metrics.total > 0:
             self.ld_client.track('$ld:ai:tokens:total', self.context, self.get_track_data(), token_metrics.total)
-        if token_metrics['input'] > 0:
+        if token_metrics.input > 0:
             self.ld_client.track('$ld:ai:tokens:input', self.context, self.get_track_data(), token_metrics.input)
-        if token_metrics['output'] > 0:
+        if token_metrics.output > 0:
             self.ld_client.track('$ld:ai:tokens:output', self.context, self.get_track_data(), token_metrics.output)