^i.2dZddlmZddlmZmZddlmZmZGddeZ GddeZ Gd d eZ Gd d eZ Gd deZ GddeZGddeZGddeZGddeZdS)uPydantic models for the research (深度研究) API. 深度研究接口的请求/响应数据模型。 定义研究任务描述、研究计划、SSE 流式数据块、 以及 plan/run/rerun 各阶段的请求和响应结构。 ) annotations)AnyLiteral) BaseModelFieldceZdZUdZeddddZded<ed d Zd ed <ed d Zded<d S)ResearchRequestzPayload for ``POST /research``..u 研究问题 min_length max_length descriptionstrquestionNu'可选的会话 ID，用于多轮对话r str | None session_id显式导入的文档 ID 列表list[str] | None seed_doc_ids) __name__ __module__ __qualname____doc__rr__annotations__rr2D:\work\zm-rag\backend\app\api\schemas\research.pyr r s~))E#!.YYYHYYYY"U45^___J____%*U4=^%_%_%_L______rr ceZdZUdZeddZded<eddd d Zd ed <edddd Zd ed<edddZ ded<edddZ ded<edddZ ded<edddZ ded<eddZ ded <edd!Zd"ed#<eed$%Zd&ed'<eed(%Zd&ed)<dS)* ResearchTaskuStructured task definition for Deep Research. 深度研究任务定义，包含研究模式、主题、范围限定、输出模板等参数。 deepu 研究模式rzLiteral['quick', 'deep']mode.r iu 研究主题r rtopicr u核心研究问题rNu 研究目标rrrgoalu 时间范围 time_rangeu 地域范围 region_scopeu 部门范围 org_scope comprehensiveu 输出模板zTLiteral['comprehensive', 'policy_brief', 'timeline', 'comparison', 'implementation']output_templateu 研究深度z,Literal['standard', 'deep', 'comprehensive'] depth_levelu必须纳入的文档default_factoryr list[str]required_doc_idsu必须关注的事项required_matter_ids)rrrrrr#rr$rr&r(r)r*r,r-listr1r2rrrr!r!s &+U6~%N%N%NDNNNNsqSnUUUEUUUUE#!J^___H____uTdOOODOOOO"U4C^TTTJTTTT$uTc~VVVLVVVV!E$3NSSSISSSS o>::: ;;;;AF"AAAK#(%Ja"b"b"bbbbb%*U4Md%e%e%eeeeeeerr!cXeZdZUdZeddZded<eedZd ed <eed Z d ed <eed Z d ed<eedZ d ed<eedZ d ed<eedZ d ed<eedZd ed<eedZd ed<dS) ResearchPlanuPlan produced before the research run starts. 研究计划：由 LLM 生成，包含目标拆解、子问题、检索重点和章节大纲， 用户确认后作为 run 阶段的执行蓝图。 .u研究计划摘要rrsummaryu 目标拆解r.r0 objectivesu 子问题 sub_questionsu 检索重点retrieval_focusu预计章节结构section_outlineu 预期交付expected_deliverablesu风险与提示notesu明确纳入的文档included_doc_idsu明确纳入的事项included_matter_idsN)rrrrrr6rr3r7r8r9r:r;r<r=r>rrrr5r53sF 5*>???G????!E$NSSSJSSSS$uT{SSSMSSSS!&t!X!X!XOXXXX!&tI]!^!^!^O^^^^',uT~'^'^'^^^^^uT?PQQQEQQQQ"'%Ja"b"b"bbbbb%*U4Md%e%e%eeeeeeerr5cfeZdZUdZeddZded<ded<edd Zd ed <dS) ResearchPlanRequestz'Payload for generating a research plan.N可选的会话 IDrrrr!taskrrrrrrrrrrrrrrr@r@Esf11"U45IJJJJJJJJ%*U4=^%_%_%_L______rr@c6eZdZUdZdZded<ded<ded<dS) ResearchPlanResponsez,Response body for a generated research plan.Nrrr!rBr5plan)rrrrrrrrrrErEMsA66!J!!!!rrEcpeZdZUdZeddZded<ded<d ed <edd Zd ed <dS)ResearchRunRequestz0Payload for executing a confirmed research plan.NrArrrr!rBr5rFrrrrCrrrrHrHUsr::"U45IJJJJJJJJ%*U4=^%_%_%_L______rrHceZdZUdZeddZded<ded<d ed <ed d d dZded<edddZded<eddZ ded<eddZ ded<dS)ResearchSectionRerunRequestz1Payload for re-running a specific report section.NrArrrr!rBr5rF.r r'u要重跑的章节标题r r section_titler u当前章节摘要r%section_summaryu!当前章节已绑定的文档 IDrsource_doc_idsrr) rrrrrrrrKrLrMrrrrrJrJ^s;;"U45IJJJJJJJJsqSNhiiiMiiii"'%K_"`"`"`O````',uT?b'c'c'cNcccc%*U4=^%_%_%_L______rrJceZdZUdZeddZded<dZded <dZded <dZ ded <dZ ded <dZ ded <dZ ded<dZ ded<dZded<dZded<dZded<dZded<dS) ResearchChunkuFA single SSE chunk sent during streaming research output. ``type`` indicates the kind of content: - ``thinking`` – intermediate reasoning steps - ``text`` – final answer text - ``reference`` – a cited document reference - ``done`` – signals end of stream - ``error`` – an error occurred .zthinking | text | reference | progress | plan | summary | finding | conflict | section | open_question | source_group | follow_up | done | errorrrtypeNrcontentstagestatustitleseveritystrengthdoc_id doc_numberz int | None chunk_index float | Nonerelevance_scorezdict[str, Any] | Nonepayload)rrrrrrPrrQrRrSrTrUrVrWrXrYr[r\rrrrOrOjs  [DGEFEHHF!J!!!!"K""""$(O((((%)G))))))rrOcDeZdZUdZded<ded<dZded<dZded <dS) ResearchReferencez2A document reference cited by the research answer.rrWrTNrrXrZr[)rrrrrrXr[rrrr^r^sL<<KKKJJJ!J!!!!$(O((((((rr^N)r __future__rtypingrrpydanticrrr r!r5r@rErHrJrOr^rrrrbs#"""""%%%%%%%%`````i```fffff9fff8fffff9fff$`````)```9```````` ` ` ` ` `) ` ` `*****I***>))))) )))))r