Bladeren bron

Annotations example (#24304)

tags/1.8.0
Asuka Minato 2 maanden geleden
bovenliggende
commit
1459fded08
No account linked to committer's email address
1 gewijzigde bestanden met toevoegingen van 5 en 3 verwijderingen
  1. 5
    3
      api/core/model_runtime/entities/llm_entities.py

+ 5
- 3
api/core/model_runtime/entities/llm_entities.py Bestand weergeven

@@ -1,3 +1,5 @@
from __future__ import annotations

from collections.abc import Mapping, Sequence
from decimal import Decimal
from enum import StrEnum
@@ -54,7 +56,7 @@ class LLMUsage(ModelUsage):
)

@classmethod
def from_metadata(cls, metadata: dict) -> "LLMUsage":
def from_metadata(cls, metadata: dict) -> LLMUsage:
"""
Create LLMUsage instance from metadata dictionary with default values.

@@ -84,7 +86,7 @@ class LLMUsage(ModelUsage):
latency=metadata.get("latency", 0.0),
)

def plus(self, other: "LLMUsage") -> "LLMUsage":
def plus(self, other: LLMUsage) -> LLMUsage:
"""
Add two LLMUsage instances together.

@@ -109,7 +111,7 @@ class LLMUsage(ModelUsage):
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.


Laden…
Annuleren
Opslaan