From 1459fded0836b6c39f432303269250390c8bde88 Mon Sep 17 00:00:00 2001
From: Asuka Minato <i@asukaminato.eu.org>
Date: Fri, 22 Aug 2025 11:14:17 +0900
Subject: [PATCH] Annotations example (#24304)

---
 api/core/model_runtime/entities/llm_entities.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/api/core/model_runtime/entities/llm_entities.py b/api/core/model_runtime/entities/llm_entities.py
index ace2c1f77..0e1277bc8 100644
--- a/api/core/model_runtime/entities/llm_entities.py
+++ b/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.