From 0d169ed3cb23b152bfa232d123fea0dc28cd165a Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Tue, 11 Nov 2025 21:28:52 +0800
Subject: [PATCH] tyro configurable

---
 fastapi_server/fake_stream_server_dashscopy.py | 10 +++++-----
 fastapi_server/server_dashscope.py             | 10 +++++-----
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/fastapi_server/fake_stream_server_dashscopy.py b/fastapi_server/fake_stream_server_dashscopy.py
index 491e411..4b64c81 100644
--- a/fastapi_server/fake_stream_server_dashscopy.py
+++ b/fastapi_server/fake_stream_server_dashscopy.py
@@ -9,12 +9,17 @@ import time
 import json
 import uvicorn
 from loguru import logger
+import tyro
 
 # Ensure we can import from project root
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from lang_agent.pipeline import Pipeline, PipelineConfig
 
+# Initialize Pipeline once
+pipeline_config = tyro.cli(PipelineConfig)
+pipeline:Pipeline = pipeline_config.setup()
+
 
 class DSMessage(BaseModel):
     role: str
@@ -43,11 +48,6 @@ app.add_middleware(
 )
 
 
-# Initialize Pipeline once
-pipeline_config = PipelineConfig()
-pipeline:Pipeline = pipeline_config.setup()
-
-
 def sse_chunks_from_text(full_text: str, response_id: str, model: str = "qwen-flash", chunk_size: int = 1000):
     created_time = int(time.time())
 
diff --git a/fastapi_server/server_dashscope.py b/fastapi_server/server_dashscope.py
index 3c9a244..8f0132c 100644
--- a/fastapi_server/server_dashscope.py
+++ b/fastapi_server/server_dashscope.py
@@ -9,12 +9,17 @@ import time
 import json
 import uvicorn
 from loguru import logger
+import tyro
 
 # Ensure we can import from project root
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from lang_agent.pipeline import Pipeline, PipelineConfig
 
+# Initialize Pipeline once
+pipeline_config = tyro.cli(PipelineConfig)
+pipeline:Pipeline = pipeline_config.setup()
+
 
 class DSMessage(BaseModel):
     role: str
@@ -43,11 +48,6 @@ app.add_middleware(
 )
 
 
-# Initialize Pipeline once
-pipeline_config = PipelineConfig()
-pipeline:Pipeline = pipeline_config.setup()
-
-
 def sse_chunks_from_stream(chunk_generator, response_id: str, model: str = "qwen-flash"):
     """
     Stream chunks from pipeline and format as SSE.