How I can capture usage

[poojayad]

This I'm trying to figure out a way to capture usage. And in _raw_response I'm got-

ChatCompletion(id='chatcmpl-AJHni8VbinfvMlISa7oqM9ayC2UxQ', choices=[Choice(finish_reason='stop', index=0, logprobs=None, message=ChatCompletionMessage(content=None, refusal=None, role='assistant', function_call=None, tool_calls=[ChatCompletionMessageToolCall(id='call_IYAnmrlFFREWtjOGHJrq53xM', function=Function(arguments='{"name":"John Doe","age":30}', name='UserInfo'), type='function')]))], created=1729160022, model='gpt-3.5-turbo-0125', object='chat.completion', service_tier=None, system_fingerprint=None, usage=CompletionUsage(completion_tokens=10, prompt_tokens=79, total_tokens=89, completion_tokens_details=None, prompt_tokens_details=None))

which is instance of "openai.types.chat.chat_completion.ChatCompletion"

usage=CompletionUsage(completion_tokens=10, prompt_tokens=79, total_tokens=89, completion_tokens_details=None, prompt_tokens_details=None))

Now here I'm looking out a authentic way to read usage and token.
is there any exposed method, or can utilise hooks for this. please help

Refer Sample Test Code Here

import instructor
from pydantic import BaseModel
from openai import OpenAI

# Define the desired output structure
class UserInfo(BaseModel):
    name: str
    age: int

# Patch the OpenAI client
client = instructor.from_openai(OpenAI())

# Define a logging hook to print kwargs, including token usage if available
def log_kwargs(**kwargs):
    if "usage" in kwargs:
        print(f"Usage info: {kwargs['usage']}")  # Log token usage data
    else:
        print("No usage data found.")

# Attach the logging hook to the client
client.on("completion:kwargs", log_kwargs)

# Extract structured data from natural language
response = client.chat.completions.create(
    model="gpt-3.5-turbo",
    response_model=UserInfo,
    messages=[{"role": "user", "content": "John Doe is 30 years old."}],
    # return_raw_response=True  # Ensure full response with metadata
)
print(response)
raw_response = response._raw_response
print("raw_response")
print(raw_response)
print(type(raw_response))

if raw_response and "usage" in raw_response:
    print(f"Prompt Tokens: {raw_response['usage']['prompt_tokens']}")
    print(f"Completion Tokens: {raw_response['usage']['completion_tokens']}")
    print(f"Total Tokens: {raw_response['usage']['total_tokens']}")
else:
    print("Not able to capture usage data found.")

[poojayad]

if (
hasattr(raw_response, "usage")
and raw_response.usage is not None
):
prompt_tokens = raw_response.usage.prompt_tokens
completion_tokens = raw_response.usage.completion_tokens
total_tokens = raw_response.usage.total_tokens

                    print(f"Prompt Tokens: {prompt_tokens}")
                    print(f"Completion Tokens: {completion_tokens}")
                    print(f"Total Tokens: {total_tokens}")
                else:
                    print("No usage information available.")
                    
                    
                    For I choose to go ahead with this, lmk if we have better solutions