Annotations example (#24304)
This commit is contained in:
@@ -1,3 +1,5 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from collections.abc import Mapping, Sequence
|
from collections.abc import Mapping, Sequence
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
from enum import StrEnum
|
from enum import StrEnum
|
||||||
@@ -54,7 +56,7 @@ class LLMUsage(ModelUsage):
|
|||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_metadata(cls, metadata: dict) -> "LLMUsage":
|
def from_metadata(cls, metadata: dict) -> LLMUsage:
|
||||||
"""
|
"""
|
||||||
Create LLMUsage instance from metadata dictionary with default values.
|
Create LLMUsage instance from metadata dictionary with default values.
|
||||||
|
|
||||||
@@ -84,7 +86,7 @@ class LLMUsage(ModelUsage):
|
|||||||
latency=metadata.get("latency", 0.0),
|
latency=metadata.get("latency", 0.0),
|
||||||
)
|
)
|
||||||
|
|
||||||
def plus(self, other: "LLMUsage") -> "LLMUsage":
|
def plus(self, other: LLMUsage) -> LLMUsage:
|
||||||
"""
|
"""
|
||||||
Add two LLMUsage instances together.
|
Add two LLMUsage instances together.
|
||||||
|
|
||||||
@@ -109,7 +111,7 @@ class LLMUsage(ModelUsage):
|
|||||||
latency=self.latency + other.latency,
|
latency=self.latency + other.latency,
|
||||||
)
|
)
|
||||||
|
|
||||||
def __add__(self, other: "LLMUsage") -> "LLMUsage":
|
def __add__(self, other: LLMUsage) -> LLMUsage:
|
||||||
"""
|
"""
|
||||||
Overload the + operator to add two LLMUsage instances.
|
Overload the + operator to add two LLMUsage instances.
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user