From b65ce87a39eb33251e7cb920590aee64338808f3 Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
Date: Tue, 29 Apr 2025 17:44:11 +0800
Subject: [PATCH] fix: `current_stage` in `Query`

---
 ruff.toml => .ruff.toml | 1 +
 pkg/core/entities.py    | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)
 rename ruff.toml => .ruff.toml (98%)

diff --git a/ruff.toml b/.ruff.toml
similarity index 98%
rename from ruff.toml
rename to .ruff.toml
index 27159c90..992262ff 100644
--- a/ruff.toml
+++ b/.ruff.toml
@@ -8,6 +8,7 @@ ignore = [
     "E741", # Ambiguous variable name: `l`
     "E722", # bare-except
     "E721", # type-comparison
+    "F821", # undefined-all
     "FURB113", # repeated-append
     "FURB152", # math-constant
     "UP007", # non-pep604-annotation
diff --git a/pkg/core/entities.py b/pkg/core/entities.py
index 5ffd0029..9eddc935 100644
--- a/pkg/core/entities.py
+++ b/pkg/core/entities.py
@@ -96,7 +96,7 @@ class Query(pydantic.BaseModel):
     """回复消息链，从resp_messages包装而得"""
 
     # ======= 内部保留 =======
-    current_stage = None  # pkg.pipeline.pipelinemgr.StageInstContainer
+    current_stage: typing.Optional['pkg.pipeline.pipelinemgr.StageInstContainer'] = None
     """当前所处阶段"""
 
     class Config: