o
    hU                     @   sB   d dl mZ d dlmZ d dlZd dlmZ dZG dd dZdS )    )LLM)settingsN)Pathz(user_journey_service/config/prompts.yamlc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )SummaryGeneratorc              	   C   s@   t tj dtj tjtjtjtjtjtj	d| _
|  | _d S )N/)modeltemperatureapi_keyapi_baseapi_versionseedtimeout)r   r   PROVIDERLLM_FOR_REVIEWTEMPERATUREOPENAI_KEY2	ENDPOINT2API_VERSION2SEEDTIMEOUTllmload_promptsprompts)self r   /home/azureuser/microlearn/backend/user_journey_with_openai/agentic_workflow/src/user_journey_service/processors/summary_generator.py__init__	   s   	zSummaryGenerator.__init__c                 C   s<   t ttd}t|W  d    S 1 sw   Y  d S )Nr)openr   prompts_pathyaml	safe_load)r   fr   r   r   r      s   $zSummaryGenerator.load_promptsc                 C   P   t d | jd }|j|d}t d|  d|dg}| j|}t | |S )Nz'Inside structured learning function....structured_learningcontentThe prompt is : userroler&   printr   formatr   callr   r&   prompt_templatepromptmessagesresponser   r   r   r$         
z$SummaryGenerator.structured_learningc                 C   s   |  | d S )N)skill_gap_analysis)r   r&   r   r   r   ondemand_learning*   s   z"SummaryGenerator.ondemand_learningc                 C   r#   )Nz&Inside skill gap analysis function....skills_gap_analysisr%   r'   r(   r)   r+   r/   r   r   r   r5   .   r4   z#SummaryGenerator.skill_gap_analysisN)__name__
__module____qualname__r   r   r$   r6   r5   r   r   r   r   r      s    r   )	crewair    user_journey_service.core.configr   r    pathlibr   r   r   r   r   r   r   <module>   s    