refactor(api): move model_runtime into dify_graph (#32858)

2026-05-05 18:08:07 +08:00 · 2026-03-02 20:15:32 +08:00
parent e985e73bdc
commit 4fd6b52808
253 changed files with 557 additions and 589 deletions
--- a/api/dify_graph/model_runtime/entities/rerank_entities.py
+++ b/api/dify_graph/model_runtime/entities/rerank_entities.py
@ -0,0 +1,20 @@
+from pydantic import BaseModel
+
+
+class RerankDocument(BaseModel):
+    """
+    Model class for rerank document.
+    """
+
+    index: int
+    text: str
+    score: float
+
+
+class RerankResult(BaseModel):
+    """
+    Model class for rerank result.
+    """
+
+    model: str
+    docs: list[RerankDocument]