Skip to content

Commit 534c0c4

Browse files
author
chengjie
committed
feat: 添加 only_need_prompt 参数用于调试 user_prompt 是否生效
1 parent 9c9e64a commit 534c0c4

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

api/models.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,12 @@ class QueryRequest(BaseModel):
158158
example=False
159159
)
160160

161+
only_need_prompt: bool = Field(
162+
False,
163+
description="是否仅返回最终提示词(用于调试,查看发送给 LLM 的完整提示)",
164+
example=False
165+
)
166+
161167
# ===== 关键词提取 =====
162168
hl_keywords: Optional[List[str]] = Field(
163169
None,

api/query.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ async def query_rag(
9393
"chunk_top_k": DEFAULT_CHUNK_TOP_K,
9494
"enable_rerank": True,
9595
"only_need_context": request.only_need_context,
96+
"only_need_prompt": request.only_need_prompt,
9697
}
9798

9899
# 添加高级参数(如果提供)

0 commit comments

Comments
 (0)