How to handle types from the acompletion in python sdk?

[Nikola-Milovic]

Hey! i had a typesafe implementation with openai python sdk, but when I switched to litellm I've noticed having issues with types since everything is type | Unknown

This example worked perfectly before

            parsed = None
            usage = None

            # Check if choices are available
            if (
                chat_completion.choices[0]
                and chat_completion.choices[0].message.content
            ):
                parsed = chat_completion.choices[0].message.parsed

            # Check if usage data is available
            if chat_completion.choices:
                usage_data = chat_completion.usage

                # Ensure all usage attributes are available
                if (
                    usage_data.prompt_tokens is not None
                    and usage_data.completion_tokens is not None
                    and usage_data.total_tokens is not None
                ):
                    usage = Usage(
                        prompt_tokens=usage_data.prompt_tokens,
                        completion_tokens=usage_data.completion_tokens,
                        total_tokens=usage_data.total_tokens,
                    )

            return (parsed, usage)

Now its full of type errors

Diagnostics:
    2   │   │   │   if (                                Cannot access attribute "message" for class "StreamingChoices"
    1   │   │   │   │   chat_completion.choices[0]        Attribute "message" is unknown [reportAttributeAccessIssue]
 229   │   │   │   │   and chat_completion.choices[0].message.content     ● Pyright: Cannot access attribute "message" for c
    1   │   │   │   ):
   2   │   │   │   │   parsed = chat_completion.choices[0].message.parsed     ●● Pyright: Cannot access attribute "message"
    3   │   │   │
    4   │   │   │   # Check if usage data is available
    5   │   │   │   if chat_completion.choices:
   6   │   │   │   │   usage_data = chat_completion.usage     ● Pyright: Cannot access attribute "usage" for class "ModelRes
    7   │   │   │   │
    8   │   │   │   │   # Ensure all usage attributes are available
    9   │   │   │   │   if (

And I couldnt find any example that actually handles types properly