Asuka Minato 2 месяцев назад
Родитель
Сommit
1459fded08
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 5 добавлений и 3 удалений
  1. 5
    3
      api/core/model_runtime/entities/llm_entities.py

+ 5
- 3
api/core/model_runtime/entities/llm_entities.py Просмотреть файл

@@ -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.


Загрузка…
Отмена
Сохранить