langchain/libs/partners/ai21/tests
ccurme fbfed65fb1
core, partners: add token usage attribute to AIMessage (#21944)
```python
class UsageMetadata(TypedDict):
    """Usage metadata for a message, such as token counts.

    Attributes:
        input_tokens: (int) count of input (or prompt) tokens
        output_tokens: (int) count of output (or completion) tokens
        total_tokens: (int) total token count
    """

    input_tokens: int
    output_tokens: int
    total_tokens: int
```
```python
class AIMessage(BaseMessage):
    ...
    usage_metadata: Optional[UsageMetadata] = None
    """If provided, token usage information associated with the message."""
    ...
```
2024-05-23 14:21:58 -04:00
..
integration_tests core, partners: add token usage attribute to AIMessage (#21944) 2024-05-23 14:21:58 -04:00
unit_tests core, standard tests, partner packages: add test for model params (#21677) 2024-05-17 13:51:26 -04:00
__init__.py ai21: init package (#17592) 2024-02-15 12:25:05 -08:00