jimi
a842f1861f
Some checks failed
Pre-commit / run (ubuntu-latest) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_en (ubuntu-latest, 3.10) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_zh (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.12) (push) Has been cancelled
27 lines
677 B
Python
27 lines
677 B
Python
# -*- coding: utf-8 -*-
|
|
"""The model usage class in agentscope."""
|
|
from dataclasses import dataclass, field
|
|
from typing import Literal, Any
|
|
|
|
from .._utils._mixin import DictMixin
|
|
|
|
|
|
@dataclass
|
|
class ChatUsage(DictMixin):
|
|
"""The usage of a chat model API invocation."""
|
|
|
|
input_tokens: int
|
|
"""The number of input tokens."""
|
|
|
|
output_tokens: int
|
|
"""The number of output tokens."""
|
|
|
|
time: float
|
|
"""The time used in seconds."""
|
|
|
|
type: Literal["chat"] = field(default_factory=lambda: "chat")
|
|
"""The type of the usage, must be `chat`."""
|
|
|
|
metadata: dict[str, Any] | None = field(default_factory=lambda: None)
|
|
"""The metadata of the usage."""
|