@i.dZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZej eZGddZy) z/Service layer for encapsulating business logic.N)DictAnyOptionalList)PipelineManager) MemoryStore)PatternEmbedderc deZdZdZdefdZddedeede ee ffdZ d ed e d ede ee ffd Z dd edededede ee ff dZde ee ffdZde ee fde ee ffdZde ee fde ee ffdZde de fdZdedede ee ffdZded e eeefde ee ffdZdede ee ffdZy) WebServicezService layer for web API.managercN||_t|_t|_y)zeInitialize web service. Args: manager: PipelineManager instance N)r rmemoryr embedder)selfr s C/home/ubuntu/codebase/yexijia/保研/colocation_mvp/web/services.py__init__zWebService.__init__s  !m ') Nqueryiteration_roundsreturnc |jj||}|j|}d|vr|jd|d<|S#t$r }t j d|dd}~wwxYw)a Execute a query and return results. Args: query: Natural language query string iteration_rounds: Optional number of iteration rounds (Stage4) Returns: Dictionary containing query results )rextracted_parametersparamszError processing query: Texc_infoN)r process_query _clean_resultpop Exceptionloggererrorrrrresultcleaned_resultes r run_queryzWebService.run_querys \\//HX/YF!//7N%7+9+=+=>T+Ux(! !  LL3A37$L G  sAA A3A..A3 pattern_idpatternfeedbackc |jj}|jj|}|dk(r|jj ||n0|dk(r|jj ||ntd||jj|}t|jdg}t|jdg}dd|||dS#t$r } tjd| d d } ~ wwxYw) a,Add user feedback for a pattern. Args: pattern_id: Pattern identifier pattern: Pattern content (list of feature types) feedback: "positive" or "negative" Returns: Dictionary with feedback status and counts positivenegativezInvalid feedback type: okzFeedback saved: )statusmessagepositive_countnegative_countzError adding feedback: TrN)r stage3_user_idrencode_patternr add_positive add_negative ValueErrorget_user_profilelengetrr r!) rr'r(r)user_id pattern_vec user_profiler0r1r%s r add_feedbackzWebService.add_feedback2s  ll11G--66w?K:% ((+>Z' ((+> #:8*!EFF ;;77@L !1!1*b!ABN !1!1*b!ABN-hZ8"0"0     LL21#6L F  sCC D )DD epochs learning_rate batch_sizemarginc |jjsddgddS|jjsddgddS|jjddgddSddlm}|j |jjjj||jj_ |jjj|||jj }|jj|jj}|rz#Contrastive learning is not enabledNzTrainer is not initialized)lr)r>r@ save_pathr+r,stage3_model_path)PreferenceEncoderzconfig/config.yamlrutf-8encodingstage3 hidden_dim) input_dimrMcpu) map_locationmodel_state_dictz(Preference model reloaded after trainingz#Failed to reload preference model: r-zTraining completed successfullylosszError during training: Tr)(r stage3_enabledstage3_use_contrastivetrainer torch.optimoptimAdammodel parameters optimizertrainrFrr7r2r8r9_save_training_historyhasattrospathexistsembedding.encoderrGyamlopen safe_loadr get_sentence_embedding_dimensionpreference_modeltorchloadload_state_dictevalr inforwarningr!str)rr>r?r@rArXhistoryr<r0r1total_feedbackrGrd config_pathfconfigrOrMri checkpointr%s r run_trainingzWebService.run_trainingZsY <<..%6$& <<66%D$& ||##+%;$&  (-2ZZ $$**557 .8.DLL * ll**00%,,881G ;;77 8S8STL!$\%5%5j"%E!F!$\%5%5j"%E!F!/.!@!" LL / / Pt||%89bggnnT\\MkMk>lNC"6Kk3AQ!%!2B $ 3 3 T T VI!'Hb!9!=!=lC!PJ4E"+#-5DLL1 !!&DLL,J,JY^!_JLL11AA*M_B`aLL11668KK JK < ' FB 7  #BA!NNN%H#LMMN  LL21#6L F!q6 "   sqL<L<L<E1L< L%L7C3L*L<L L L9L4/L<4L99L<< M9)M4.M94M9c,ddl}ddl}d}|jj|sdgiS t |dd5}|j |}ddd|S#1swYSxYw#t $r(}tjd|d dgicYd}~Sd}~wwxYw) zuGet training history from logs. Returns: Dictionary containing training history rNzlogs/training_history.json trainingsrHrIrJz Error reading training history: Tr) jsonr`rarbrerjrr r!)rryr` history_pathrsdatar%s rget_training_historyzWebService.get_training_historys 3 ww~~l+$ $ %lC':ayy|;K;K % LL;A3?$L O$ $ %s:A"A A"AA"A"" B+BBBr#c4i}|jD]\}}t|tjr|j ||<4t|t r|j |||<Yt|tr|j|||<~|||<|S)zClean result dictionary, converting numpy arrays to lists. Args: result: Raw result dictionary Returns: Cleaned result dictionary items isinstancenpndarraytolistdict _clean_dictlist _clean_list)rr#cleanedkeyvalues rrzWebService._clean_results ,,.JC%,$||~ E4(#//6 E4(#//6 $ )rdc4i}|jD]\}}t|tjr|j ||<4t|t r|j |||<Yt|tr|j|||<~|||<|S)zRecursively clean dictionary.r~)rrrkvs rrzWebService._clean_dictsGGIDAq!RZZ(XXZ At$!--a0 At$!--a0  rlcrg}|D]}t|tjr |j|j =t|t r!|j|j |nt|tr!|j|j||j||S)zRecursively clean list.) rrrappendrrrrr)rrritems rrzWebService._clean_listsD$ +t{{}-D$'t//56D$'t//56t$rc |jj||}|j|}|S#t$r }tj d|dd}~wwxYw)a Start an interactive iteration session (Stage4). Args: query: User query iteration_rounds: Number of iteration rounds Returns: Dictionary containing session_id and first round results zError starting iteration: TrN)r start_iterationrrr r!r"s rrzWebService.start_iterations_ \\11%9IJF!//7N! !  LL5aS9DL I  .1 AAA session_idc |jj||}|j|}|S#t$r }tj d|dd}~wwxYw)a6Process feedback, train model, and return next iteration round (Stage4). Args: session_id: Iteration session ID feedback: Dictionary with 'positive' and 'negative' pattern lists Returns: Dictionary containing next round results zError in next iteration: TrN)r next_iterationrrr r!)rrr)r#r$r%s rrzWebService.next_iterations^ \\00XFF!//7N! !  LL4QC84L H  rc |jj|}|j|}|S#t$r }tj d|dd}~wwxYw)zGet final results after all iterations (Stage4). Args: session_id: Iteration session ID Returns: Dictionary containing final results and explanation zError finalizing iteration: TrN)r finalize_iterationrrr r!)rrr#r$r%s rrzWebService.finalize_iteration,s\ \\44Z@F!//7N! !  LL7s;dL K  s-0 AAA)N) gMbP? g333333?)__name__ __module__ __qualname____doc__rrrorintrrr&rr=floatrvr|rrrrrrrrrr r s$**shsmtTWY\T\~,&s&T&S&TRUWZRZ^&PEJ:=f3fEf #f27fBFsCx.fP%d38n%*DcNtCH~* T#s(^ S#X  T d SCDcN$S$s)^8LQUVY[^V^Q_$ST#s(^rr )rloggingr`typingrrrrnumpyrcontroller.managerr memory.storerlearning.embedderr getLoggerrr r rrrrs>5 ,,.$-   8 $llr