o sg @sdZddlZejdddlZddlZddlmZeddlm Z ddl m Z ddl m Z ddlmZmZdd lmZGd d d eZd ed eeeeBeBfddZedkroddlmZegdZeeejdSdS)z(Format Public Comps as a table using LLMN.) load_dotenv)JsonOutputParser)ChatPromptTemplate) ChatOpenAI) BaseModelField)OPENAI_MODEL_35c@s2eZdZUedddZeeeeBeBe d<dS) TableOutput.z!List of list for the table's data)titledataN) __name__ __module__ __qualname__rr liststrfloatint__annotations__rrN/var/www/html/XCapMarket/services/stock_info/public_comparable_format_table.pyr s "r r returnc Csttdtdd}i}|D]T}||d}|D]E}t||tr2||dkr1t||d||<qt||tr]||D]}t|||tr\|||dkr\|||d|||<q=q|||<qt gd}t t d}||B|B} | t ||d } | S) Ng?OPENAI_API_KEY)model temperatureapi_key comp_tableiiʚ;))systemz.You are an expert with the Python PPTX library)rzXYou are going to create a table for the data provided. Return the data as a list of list)rzQDon't provide the ticker column and add an empty column between Name and the data)rzIf there are units such as Billions or Millions consistent across the column, add it to the header. Don't add it to the Multiples like EV / Revenue)humanz)The data you have to work with is: {data})rz!Return the in JSON: {json_format}) placeholderz{agent_scratchpad})pydantic_model)r json_format)rr osgetenv isinstancerrdictr from_messagesrr invokejsondumpsget_format_instructions) r llm llm_inputtickerrkeysub_keypromptparserchainoutputrrrformat_for_tables6       r4__main__)PublicComparables)AAPLTSLAMETAMSFT) __doc__syspathappendr(r"dotenvrlangchain_core.output_parsersrlangchain_core.promptsrlangchain_openairpydanticrrconfigs.configr r r%rrrrr4r %services.stock_info.public_comparabler6pcprintr rrrrs&      "4