o sg@sbdZddlZddlZddlmZddlmZddlmZddl m Z m Z e Z GdddZdS) zSlide researcherN)ChatPromptTemplate)JsonOutputParser) ChatOpenAI)OPENAI_API_KEYOPENAI_MODEL_MINIc@sFeZdZdZeededZeddedZdZ dZ d dd Z d d Z dS) SlideResearchz%Conduct research on the slide content)model temperatureapi_keyzgpt-4o10z20-40NcCs||_||_||_||_dS)z#Initialize the slide research classN)slide primary_data parsing_classprompt_addition)selfr rrrrA/var/www/html/XCapMarket/services/ppt_generator/slide_research.py__init__s zSlideResearch.__init__c Csvd}|rz`ddd|jdfdd|jdfddd d d g}d |jjvr(|d |jr3|d|jft|}t|jd}||j B|B}| |j j |j j |jd|d}|jdi|}d}WnRtytjdddtd||jB|B}| |j j |j j |jd|d}z|jdi|}d}WntytjdddtdYnwYnw|s|S)zFormat the slide contentT)systemz[You are an expert in converting the context provided and fit it within the format asked forrzAll titles should be z tokens or shorterzAll helper text should be )humanzSlide title: {title})rz&Slide highlevel takeaway is: {content})rzSlide research data to be used: Rely on primary data here first: {primary_data} Rely on secondary data here second if needed: {secondary_data})rz%convert to json format: {json_format})rzMake the slide content specific to the company. If you don't have specific data points or information, you can use [XXX] as a placeholder Examples: graph)rzOIf you don't have actual numbers, use 0 as the placeholder for the data points.r)pydantic_object)titlecontentrsecondary_data json_formatFz&Error in formatting the slide content.)exc_infoz%Error in formatting the slide content Nr) title_tokens text_tokensr__name__appendrr from_messagesr llm_model3invoker rrrget_format_instructionsdict Exceptionloggingerrortimesleep llm_model4)r error_output prompt_temp chat_promptparserchain chain_outputrrr_format#sx          TzSlideResearch._format)N) r# __module__ __qualname____doc__rrrr&r/r!r"rr6rrrrrs  r)r9r+r-langchain.promptsrlangchain_core.output_parsersrlangchain_openairconfigs.configrr getLogger XCM_loggerrrrrrs