o
    h                     @   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                   @   s$   e Zd Zdd Zdd Zdd ZdS )MicrolearningDefaultsEstimatorc              	   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LLM1TEMPERATUREOPENAI_KEY1	ENDPOINT1API_VERSION1SEEDTIMEOUTllmload_promptsprompts)self r   /home/azureuser/microlearn/backend/user_journey_with_openai/agentic_workflow/src/user_journey_service/processors/defaults_estimator.py__init__   s   	z'MicrolearningDefaultsEstimator.__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+MicrolearningDefaultsEstimator.load_promptsc                 C   sv   | j d }|j|j|j|j|j|j|j|j|j	|j
|j|j|j|jd}td|  d|dg}| j|}| S )Nestimate_defaults)role
experienceSkillstopicLevelCompany_NameIndustryCompany_SizeBusiness_ModelTop_Use_CasesLearning_Focus
Tech_stack
Compliancezthe prompt is : user)r$   content)r   format	Job_Title
Experiencer&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   printr   callstrip)r   
input_dataprompt_templatepromptmessagesresponser   r   r   r#      s.   
z0MicrolearningDefaultsEstimator.estimate_defaultsN)__name__
__module____qualname__r   r   r#   r   r   r   r   r      s    r   )	crewair    user_journey_service.core.configr   r    pathlibr   r   r   r   r   r   r   <module>   s    