Fix bug

Author: johann-petrak

llms_wrapper/llms.py

@@ -957,7 +957,7 @@ def cost_callback(kwargs, completion_response, start_time, end_time):
                         self.cost_logger.log(
                             dict(
                                 model=llm["llm"], modelalias=llm["alias"],
-                                cost=ret["cost"], input_tokens=ret["input_tokens"], output_tokens=ret["output_tokens"]))
+                                cost=ret["cost"], input_tokens=ret["n_prompt_tokens"], output_tokens=ret["n_completion_tokens"]))
                     return ret
                 except Exception as e:
                     tb = traceback.extract_tb(e.__traceback__)

llms_wrapper/version.py

@@ -1,3 +1,3 @@
 import importlib.metadata
-__version__ = "0.9.1.11"
+__version__ = "0.9.1.12"