You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
langchain/libs/partners/fireworks/tests/integration_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."""
    ...
```
4 months ago
..
__init__.py fireworks[patch]: Add Fireworks partner packages (#17694) 7 months ago
test_chat_models.py core[minor], ...: add tool calls message (#18947) 6 months ago
test_compile.py fireworks[patch]: Add Fireworks partner packages (#17694) 7 months ago
test_embeddings.py fireworks[patch]: Add Fireworks partner packages (#17694) 7 months ago
test_llms.py fireworks[patch]: remove custom async and stream implementations (#18363) 7 months ago
test_standard.py core, partners: add token usage attribute to AIMessage (#21944) 4 months ago