o <¹hÀã@szddlmZmZmZmZddlmZddlZddlmZddl m Z ddlmZddl mZddlmZGd d „d ƒZdS)é)ÚDictÚListÚAnyÚOptional)ÚdatetimeN)ÚRoleplayScenario)ÚConversationTurn)ÚGroqService)Ú OllamaService)ÚJSONHandlerc@súeZdZddd„Zdefdd„Zdded ed edefdd „Z ddededed edede efdd„Zded edefdd„Zded ede e eeeffdd„Zded ede e efdd„Zded edefdd„Zded ede eeeffdd„ZdS) ÚRoleplayEngineNcCs$|ptƒ|_|p tƒ|_tƒ|_dS)N)r Úgroq_servicer Úollama_servicerÚjson_handler)Úselfr r©rú\/var/www/eduai.edurigo.com/doc_train/edurigo_ai/roleplay/testing/services/roleplay_engine.pyÚ__init__szRoleplayEngine.__init__Úuse_groqcCs|r|jS|jS)z9Get the appropriate AI service based on groqRoleplay flag)r r)rrrrrÚ_get_ai_serviceszRoleplayEngine._get_ai_serviceFÚscenarioÚis_adminÚ force_restartÚreturnc Cs~|j}|j ||¡}|rtdƒn|rtdƒ|Stdƒ||jt ¡ ¡ddd|jt ¡ ¡dœgdœ}|j |||¡|S) zStart a new roleplay sessionzFDEBUG: Force restarting conversation - replacing existing conversationz8DEBUG: Conversation already exists, not creating new onez7DEBUG: No existing conversation found, creating new oneÚactiveéÚsystem©Úturn_numberÚspeakerÚmessageÚ timestamp)Ú session_idÚscenario_idÚ started_atÚstatusÚturns) r"rÚload_conversationÚprintrÚnowÚ isoformatÚscenario_introÚsave_conversation)rrrrr"Úexisting_conversationÚconversation_datarrrÚ start_sessions* üÿûzRoleplayEngine.start_sessionTr"Úlearner_messagecCsâ|j ||¡}|sdSt|dƒd}|d|t ¡ ¡dœ}|d |¡dd„|dDƒ} t| ƒdk} dd„|ddd …Dƒ}| |¡}| | ¡||¡} | sSdS|dd | t ¡ ¡dœ}|d |¡|j |||¡| S)z%Add learner response and get AI replyNr&rÚlearnerrcSsg|] }|ddkr|‘qS)rr1r©Ú.0ÚturnrrrÚ Lsz7RoleplayEngine.add_learner_response..cSs*g|]}|ddkr|d|ddœ‘qS)rrr )rr rr2rrrr5QsýþéÿÿÿÿÚai)rr'Úlenrr)r*ÚappendrÚplay_characterÚto_dictr,)rr"rr0rrr.rÚlearner_turnÚ learner_turnsÚis_first_learner_responseÚconversation_historyÚ ai_serviceÚai_responseÚai_turnrrrÚadd_learner_response9s> üû ý üz#RoleplayEngine.add_learner_responsecCs>|j ||¡}|sdSd|d<t ¡ ¡|d<|j |||¡S)zEnd a roleplay sessionFÚ completedr%Úended_at)rr'rr)r*r,©rr"rr.rrrÚend_sessionrszRoleplayEngine.end_sessioncCs|j ||¡}|sdS|dS)zGet full conversation historyNr&)rr'rFrrrÚget_conversation_history}sz'RoleplayEngine.get_conversation_historycCsV|j ||¡}|sdSg}|dD]}t|d|d|d|dd}| |¡q|S)z/Get conversation turns formatted for assessmentNr&rrr r!r)rr'rr9)rr"rr.r&Ú turn_datar4rrrÚ%get_conversation_turns_for_assessment…süz4RoleplayEngine.get_conversation_turns_for_assessmentcCs&|j ||¡}|sdS| dd¡dkS)z"Check if a session is still activeFr%r)rr'ÚgetrFrrrÚis_session_active—sz RoleplayEngine.is_session_activecCsJ|j ||¡}|sdS|d|d|d| dd¡t|dƒ| d¡d œS) zGet basic session informationNr"r#r$r%rr&rE)r"r#r$r%Ú turn_countrE)rr'rKr8rFrrrÚget_session_infoŸs úzRoleplayEngine.get_session_info)NN)F)T)Ú__name__Ú __module__Ú__qualname__rÚboolrrÚintÚstrr/rrCrGrrrrHrrJrLrNrrrrr s ($9&&r)ÚtypingrrrrrÚuuidÚmodels.scenariorÚmodels.assessmentrÚservices.groq_servicer Úservices.ollama_servicer Úutils.json_handlerrrrrrrÚs