@idZddlZddlmZmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlmZej0eZdedefd Zy) z#API routes for FastAPI application.N) APIRouter HTTPException)DictAny) QueryRequest QueryResponseFeedbackRequestFeedbackResponse TrainRequest TrainResponseTrainingHistoryResponseIterationStartRequestIterationStartResponseIterationNextRequestIterationNextResponseIterationFinalizeRequestIterationFinalizeResponse) WebServiceservicereturnctddg}|jdtdtdtffd }|jd tdt dtffd }|jd t dtdt ffd }|jd tdtffd }|jdtdtdtffd }|jdtdtdtffd }|jdtdtdtffd }|S)zCreate API router with all endpoints. Args: service: WebService instance Returns: Configured APIRouter z/apiapi)prefixtagsz/query)response_modelrequestrcK j|j|j}tdi|S#t$r5}t j d|dtdt|d}~wwxYww) aProcess a natural language query and return patterns. Args: request: Query request with natural language query string and optional iteration rounds Returns: Query response with patterns, rules, and explanation )iteration_roundszError in query endpoint: Texc_info status_codedetailN) run_queryqueryrr Exceptionloggererrorrstrrresulters A/home/ubuntu/codebase/yexijia/保研/colocation_mvp/web/router.pyr'zcreate_router..query st @&&w}}wG_G_&`F *6* * @ LL4QC84L HCA? ? @s%A817A8 A50A00A55A8z /feedbackc@K |jdvr tddj|j|j|j}t d i|S#t $r5}tjd|dtd t|d }~wwxYww) zSave user feedback for a pattern. Args: request: Feedback request with pattern and feedback type Returns: Feedback response with status and counts )positivenegativeiz)Feedback must be 'positive' or 'negative'r") pattern_idpatternfeedbackzError in feedback endpoint: Trr!Nr%) r5r add_feedbackr3r4r r(r)r*r+r,s r/r5zcreate_router..feedback1s @'??# #F ))"-- ))*F $-f- - @ LL7s;dL KCA? ? @)BAAB B&0BBBz/trainc@K j|jxsd|jxsd|jxsd|jxsd}t d i|S#t $r5}tjd|dtd t| d }~wwxYww) zTrigger preference model training. Args: request: Training request with hyperparameters Returns: Training response with loss history gMbP? g333333?)epochs learning_rate batch_sizemarginzError in train endpoint: Trr!r"Nr%) run_trainingr;r<r=r>r r(r)r*rr+r,s r/trainzcreate_router..trainLs @))~~+%33.get_training_historybsc @224G*5W5 5 @ LL?sCdL SCA? ? @s$A! A! A0AAA!z/iteration/startcK j|j|j}tdi|S#t$r5}t j d|dtdt|d}~wwxYww)zStart an interactive iteration session (Stage4). Args: request: Iteration start request with query and rounds Returns: Iteration start response with session_id and first round results z#Error in iteration start endpoint: Trr!r"Nr%) start_iterationr'rrr(r)r*rr+r,s r/rEz&create_router..start_iterationqsq @,,W]]GqcBTL RCA? ? @$A706A7 A40A//A44A7z/iteration/nextcK j|j|j}tdi|S#t$r5}t j d|dtdt|d}~wwxYww)a Process feedback, train model, and return next iteration round (Stage4). Args: request: Iteration next request with session_id and feedback Returns: Iteration next response with next round results z"Error in iteration next endpoint: Trr!r"Nr%) next_iteration session_idr5rr(r)r*rr+r,s r/rHz%create_router..next_iterationss @++G,>,>@P@PQF(262 2 @ LL=aSADL QCA? ? @rFz/iteration/finalizecK j|j}tdi|S#t$r5}tj d|dt dt|d}~wwxYww)zGet final results after all iterations (Stage4). Args: request: Iteration finalize request with session_id Returns: Iteration finalize response with final results z&Error in iteration finalize endpoint: Trr!r"Nr%)finalize_iterationrIrr(r)r*rr+r,s r/rKz)create_router..finalize_iterationsl @//0B0BCF,6v6 6 @ LLA!EPTL UCA? ? @s$A,%+A, A)0A$$A))A,)rpostrrr r r r getr rrrrrr) rrouterr'r5r@rBrErHrKs ` r/ create_routerrOsfE7 3F [[-[8@\@m@9@  [[-=[>@@4D@?@4 [[-[8@\@m@9@* ZZ 1HZI @(? @J @ [[#4J[K@'<@AW@L@  [["3H[I@&:@?T@J@  [[&7P[Q@*B@G`@R@ M)__doc__loggingfastapirrtypingrr web.schemasrrr r r r r rrrrrr web.servicesr getLogger__name__r)rOr%rPr/rYsR),$   8 $O:O)OrP