From 7bb4b678857a4b1ed3e53e93f018b3512277259a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Yanli=20=E7=9B=90=E7=B2=92?= <yanli@dify.ai>
Date: Tue, 23 Jun 2026 17:35:10 +0800
Subject: [PATCH] fix: relax knowledge retrieval usage schema

---
 api/services/entities/knowledge_retrieval_inner.py        | 8 --------
 .../services/test_knowledge_retrieval_inner_service.py    | 2 ++
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/api/services/entities/knowledge_retrieval_inner.py b/api/services/entities/knowledge_retrieval_inner.py
index 86276b80177..aaf686da44c 100644
--- a/api/services/entities/knowledge_retrieval_inner.py
+++ b/api/services/entities/knowledge_retrieval_inner.py
@@ -173,14 +173,6 @@ class InnerKnowledgeRetrieveRequest(BaseModel):
 class InnerKnowledgeRetrieveUsage(ResponseModel):
     """Serialized LLM usage payload returned by dataset retrieval."""
 
-    model_config = ConfigDict(
-        from_attributes=True,
-        extra="forbid",
-        populate_by_name=True,
-        serialize_by_alias=True,
-        protected_namespaces=(),
-    )
-
     prompt_tokens: int
     completion_tokens: int
     total_tokens: int
diff --git a/api/tests/unit_tests/services/test_knowledge_retrieval_inner_service.py b/api/tests/unit_tests/services/test_knowledge_retrieval_inner_service.py
index 287d787ad70..11f9dfedc1a 100644
--- a/api/tests/unit_tests/services/test_knowledge_retrieval_inner_service.py
+++ b/api/tests/unit_tests/services/test_knowledge_retrieval_inner_service.py
@@ -98,6 +98,8 @@ class TestInnerKnowledgeRetrievalService:
             "total_price": "0",
             "currency": "USD",
             "latency": 0,
+            "time_to_first_token": None,
+            "time_to_generate": None,
         }
         mock_rag_cls.return_value = rag