Add the model name to the LLM usage metrics

2024-08-11 12:08:46 -04:00
parent 32640e054d
commit a97775bff3
1 changed files with 1 additions and 0 deletions
--- a/src/pipecat/services/openai.py
+++ b/src/pipecat/services/openai.py
@@ -137,6 +137,7 @@ class BaseOpenAILLMService(LLMService):
            if chunk.usage:
                tokens = {
                    "processor": self.name,
+                    "model": self._model,
                    "prompt_tokens": chunk.usage.prompt_tokens,
                    "completion_tokens": chunk.usage.completion_tokens,
                    "total_tokens": chunk.usage.total_tokens