o sg1@sdZddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZdd lmZe dd lmZmZeZGdddZedkreddZ eZ!e"e!#e Z$ee$dSdS)Class to run the LLM modelN)datetime) load_dotenv)ChatPromptTemplate)JsonOutputParser) ChatOpenAI)OPENAI_TOKEN_LIMIT)LeadershipResults)SICCode)check_token_limit)load_sic_codes)OPENAI_API_KEYOPENAI_MODEL_35c@s@eZdZdZddZddZdddZd efd d Zd d Z dS)LLMChatrcCstdttd|_dS)r皙? temperature model_nameapi_keyN)rrr llm)selfr5/var/www/html/XCapMarket/utils/llm_company_profile.py__init__s zLLMChat.__init__c st|trt|ttd}d}d}|D]J}z ||||IdH}Wn2tjyU}z%t|t dt d|td||||IdH}WYd}~nd}~ww|dur_|j}|}q|S)z3 Request a summary from OpenAI API g@Nz%s:OpenAI Rate Limit Error: %sz%Y-%m-%D %H:%M:%S) isinstancestrr intropenai_request_callopenaiRateLimitErrorprintloggingerrorrtodaystrftimetimesleepcontent)r page_text openai_prompt response_text seed_textpg_textresponseerrropenai_request%s<     zLLMChat.openai_requestNc sz.ddddd|fg}|r|dt|}tdttd}||B}|||d Id H}|WStyF} z t| WYd } ~ d Sd } ~ ww) z:Call the OpenAI API to get the response for the given text)systemzYou are an expert consultant. Step 1 based on the text, become an expert in that company's industry and then follow the instructions below.)r2zEYour responses should be formatted in bullets each 20 tokens or less.)r2zYou will communicate in a formal tone, ensuring clear, succinct and professional dialogue. It will provide communication with precision, avoiding colloquial language and maintaining a serious demeanor.)humanz {page_text}r3)userzYou have already provided a summary here * {seed_text}*. Only provide a summary if you have more to add. Otherwise, return the same summary.rr)r*r-N) appendr from_messagesrrr ainvoke Exceptionr") rr*r+r- prompt_temppromptrchainr/r0rrrrGs8  zLLMChat.openai_request_call product_infocsb|dur tdttd}tgd}||jB|B}|tt || dIdH}|dS)z.Get the SIC code for the provided product_infoNzProduct info is requiredpydantic_object))r2z6You are a helpful AI bot in company SIC code analysis.)r2zOUse these SIC codes and the related industry and sector. {sic_dict_json_string})r2zmTake in the product information and provide the SIC code for the company. Do not provide NAIC or other codes.)r3z'Product Description: {product_info} )r3z {JSONoutput})sic_dict_json_stringr< JSONoutputcode) ValueErrorrr rr6rr7jsondumpssic_dictget_format_instructions)rr<parser chat_templater;chain_responserrrproduct_to_sicws  zLLMChat.product_to_siccsV|dur tdttd}tgd}||jB|B}|||dIdH}|S)z Map the leadership using the LLMNzCompany leadership is requiredr=))r2z8You are a helpful AI bot in company leadership analysis.)r2aGiven a list of leadership names and titles. Return a structured output with leader names, titles and linkedin links if provided. If there are multiple linkedin links that match, choose the first one. If there isn't a match, leave it blank)r2zGReturn a json list including the leader name, title, and linkedin link.)r3z$Leadership: {company_leadership} )r2z;Structure the JSON output using this template. {JSONoutput})company_leadershipr@)rBrr rr6rr7rF)rrKrGrHr;rIrrrmap_leaderships  zLLMChat.map_leadership)N) __name__ __module__ __qualname____doc__rr1rrrJrLrrrrrs "0 #r__main__hiaZ Elon Musk: Founder and CEO - Elon Musk: CEO - Max Hodak: President - Vanessa Tolosa: Vice President of Engineering - Philip Sabes: Vice President of Neuroscience - Tim Hanson: Vice President of Manufacturing - Jared Birchall: Vice President of Legal Affairs - Steve Petrou: Vice President of Finance - DJ Seo: Vice President of Implant Systems - Ben Rapoport: Vice President of Hardware - Will Brandler: Vice President of Software - Eric Whitmire: Vice President of Clinical Research - Matthew MacDougall: Vice President of Business Development )%rPasynciorCr#osr'rr dotenvrlangchain.promptsrlangchain_core.output_parsersrlangchain_openairconfigs.configr1services.company_profile.data_classes.llm_resultsr .services.company_profile.data_classes.SIC_coder utils.check_token_limitr utils.load_sic_codesr r rrErrMr"rKrrunrLmappingrrrrs:          &