浏览代码

Annotations example (#24304)

tags/1.8.0
Asuka Minato 2 个月前
父节点
当前提交
1459fded08
没有帐户链接到提交者的电子邮件
共有 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.


正在加载...
取消
保存