From fcd76c6aa7ed6c9dce62f19b2dacc98a6f1b3aeb Mon Sep 17 00:00:00 2001
From: Zhaofeng Zhang <24791380+vcfgv@users.noreply.github.com>
Date: Thu, 20 Nov 2025 15:14:11 +0800
Subject: [PATCH] refactor: update strategy type naming in schemas and router

---
 python/valuecell/server/api/routers/strategy_agent.py | 10 +++++-----
 python/valuecell/server/api/schemas/strategy.py       |  4 ++--
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/python/valuecell/server/api/routers/strategy_agent.py b/python/valuecell/server/api/routers/strategy_agent.py
index 6d55f51da..ca790c142 100644
--- a/python/valuecell/server/api/routers/strategy_agent.py
+++ b/python/valuecell/server/api/routers/strategy_agent.py
@@ -18,7 +18,7 @@
 from valuecell.core.coordinate.orchestrator import AgentOrchestrator
 from valuecell.core.types import CommonResponseEvent, UserInput, UserInputMetadata
 from valuecell.server.api.schemas.base import SuccessResponse
-from valuecell.server.api.schemas.strategy import StrategyType as StrategySummaryType
+from valuecell.server.api.schemas.strategy import StrategyType
 from valuecell.server.db.connection import get_db
 from valuecell.server.db.repositories import get_strategy_repository
 from valuecell.utils.uuid import generate_conversation_id, generate_uuid
@@ -33,8 +33,8 @@ def create_strategy_agent_router() -> APIRouter:
     @router.post("/create")
     async def create_strategy_agent(
         request: UserRequest,
-        strategy_type: StrategySummaryType = Query(
-            default=StrategySummaryType.PROMPT,
+        strategy_type: StrategyType = Query(
+            default=StrategyType.PROMPT,
             description="'prompt based strategy' or 'grid strategy'",
         ),
         db: Session = Depends(get_db),
@@ -105,9 +105,9 @@ async def create_strategy_agent(
             query = user_request.model_dump_json()
 
             # Select target agent based on strategy_type (enum)
-            if strategy_type == StrategySummaryType.PROMPT:
+            if strategy_type == StrategyType.PROMPT:
                 agent_name = "PromptBasedStrategyAgent"
-            elif strategy_type == StrategySummaryType.GRID:
+            elif strategy_type == StrategyType.GRID:
                 agent_name = "GridStrategyAgent"
             else:
                 raise HTTPException(
diff --git a/python/valuecell/server/api/schemas/strategy.py b/python/valuecell/server/api/schemas/strategy.py
index 96a350c5b..a8ac4f9d3 100644
--- a/python/valuecell/server/api/schemas/strategy.py
+++ b/python/valuecell/server/api/schemas/strategy.py
@@ -12,8 +12,8 @@
 
 
 class StrategyType(str, Enum):
-    PROMPT = "prompt based strategy"
-    GRID = "grid strategy"
+    PROMPT = "PromptBasedStrategy"
+    GRID = "GridStrategy"
 
 
 class StrategySummaryData(BaseModel):