103 lines
14 KiB
Plaintext
103 lines
14 KiB
Plaintext
|
Ë
|
|||
|
|
&âÂh- ã óª — d dl Z d dlmZmZ d dlZd dlZd dlmZm Z m
|
|||
|
|
Z
|
|||
|
|
m Z d dl m
|
|||
|
|
Z
|
|||
|
|
d dlmZ d dlmZ d dlmZ d dlmZ G d „ d
|
|||
|
|
e« Z G d „ d « Zy)
|
|||
|
|
é N)ÚListÚOptional)Ú CompanyTableÚ
|
|||
|
|
InvestorTableÚInvestorTeamMemberÚ SectorTable)ÚPromptTemplate)Ú
|
|||
|
|
ChatOpenAI)Ú InvestorData)Ú BaseModel)Úsettingsc ó — e Zd ZU dZeed< y)ÚInvestorOutputz Schema for LLM structured outputÚ
|
|||
|
|
investor_dataN)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__r Ú__annotations__© ó úQ/home/oluwasanmi/Documents/Work/MKD/anton_wireframe/app/services/openrouter_v2.pyr r s … Ù*à Ô r r c óº — e Zd Z ddee dee fd„Zdej dedee fd„Z
|
|||
|
|
d e e ddfd
|
|||
|
|
„Z d e e ddfd „Z
|
|||
|
|
dd ej d
|
|||
|
|
ede e fd„Zy)ÚInvestorProcessorNÚ sql_sessionÚvector_db_clientc ór — d| _ t | j dg¬« | _ t t j
|
|||
|
|
ddd¬« | _ | j j t « | _ || _
|
|||
|
|
|| _ t j d¬ « | _ | j j d
|
|||
|
|
d d i¬
|
|||
|
|
« | _ y )Na„ You are an expert data extraction assistant. Extract investor information from the provided CSV data and return it as a structured record.
|
|||
|
|
|
|||
|
|
Given the following CSV data row:
|
|||
|
|
{question}
|
|||
|
|
|
|||
|
|
Extract and structure the following fields for the investor:
|
|||
|
|
- name: The investor's full name
|
|||
|
|
- description: Description of the investor
|
|||
|
|
- aum: Assets under management (as integer, use 0 if not available)
|
|||
|
|
- check_size_lower: Lower bound of investment check size (as integer)
|
|||
|
|
- check_size_upper: Upper bound of investment check size (as integer)
|
|||
|
|
- geographic_focus: Geographic region focus
|
|||
|
|
- stage_focus: Investment stage focus (must be one of: seed, series_a, series_b, series_c, growth, late_stage)
|
|||
|
|
- number_of_investments: Number of investments made (default 0)
|
|||
|
|
|
|||
|
|
Also extract related data:
|
|||
|
|
- portfolio_companies: List of companies they've invested in
|
|||
|
|
- team_members: List of team members with name, role, email
|
|||
|
|
- sectors: List of sectors they focus on
|
|||
|
|
|
|||
|
|
Important:
|
|||
|
|
- If a field is not available, use appropriate defaults
|
|||
|
|
- stage_focus must be one of the valid enum values
|
|||
|
|
- Return clean, valid JSON only
|
|||
|
|
|
|||
|
|
Return the data as a single comprehensive investor data record.Úquestion)ÚtemplateÚinput_variablesz https://openrouter.ai/api/v1z google/gemini-2.5-flash-liter )Úapi_keyÚbase_urlÚmodelÚ temperaturez ./chroma_db)ÚpathÚinvestor_descriptionsÚ descriptionz1Investor descriptions and investment thesis focus)ÚnameÚmetadata)r r Úpromptr
|
|||
|
|
r
|
|||
|
|
ÚOPENROUTER_API_KEYÚllmÚwith_structured_outputr Ústructured_llmr r ÚchromadbÚPersistentClientÚget_or_create_collectionÚ
|
|||
|
|
collection)Úselfr r s r Ú__init__zInvestorProcessor.__init__ s € ð
|
|||
|
|
CˆŒ
|
|||
|
|
ô6 %Ø—]‘]°Z°Lô
|
|||
|
|
ˆŒ ô Ü ×/Ñ/Ø3Ø0Øô
|
|||
|
|
ˆŒð #Ÿh™h× =Ñ =¼nÓ MˆÔØ&ˆÔØ 0ˆÔ ä (× 9Ñ 9¸}Ô MˆÔ Ø ×/Ñ/×HÑHØ(à Ð Rðð Ió
|
|||
|
|
ˆ�r ÚrowÚrow_idxÚreturnc ƒ óŠ K — i }|j « D ]s \ }}t j |« sŒ t |« j dd« j dd« j dd« }dj d„ |D « « }|||< Œu dj |j « D ��cg c] \ }}|› d|› �‘Œ c}}« } t
|
|||
|
|
d |d
|
|||
|
|
z › d �« | j j |« ƒ d { –—† }|j r |j S y c c}}w 7 Œ## t $ r } t
|
|||
|
|
d
|
|||
|
|
|d
|
|||
|
|
z › d| › �« Y d } ~ y d } ~ ww xY ww)z Process a single row of dataú
|
|||
|
|
ú ú
|
|||
|
|
ú Ú c 3 óD K — | ] }t |« d k\ s|dv r|–— Œ yw)é )r9 r; r< N)Úord)Ú.0Úchars r ú <genexpr>z1InvestorProcessor._process_row.<locals>.<genexpr>\ s+ è ø€ ò &à ܘ4“y B’¨$Ð2DÑ*Dô ñ&ùs ‚ ú, z: zProcessing row é z...NzError processing row ) ÚitemsÚpdÚnotnaÚstrÚreplaceÚjoinÚprintr. Úainvoker Ú Exception)
|
|||
|
|
r3 r5 r6 Ú cleaned_rowÚkeyÚvalueÚ clean_valueÚrow_strÚresultÚes
|
|||
|
|
r Ú _process_rowz InvestorProcessor._process_rowL sG è ø€ ð
|
|||
|
|
ˆ Ø Ÿ)™)›+ò /‰JˆC�Ü�x‰x˜�ô ˜“Jß ‘W˜T 3Ó'ß ‘W˜T 3Ó'ß ‘W˜T 3Ó'ð ð !Ÿg™gñ &à +ô&ó � ð
|
|||
|
|
$/� ˜CÒ ð /ð" —)‘)Ø1<×1BÑ1BÓ1D× E¡: 3¨�ˆu�B�u�gÒ
|
|||
|
|
Ó Eó
|
|||
|
|
ˆð Ü �O G¨a¡K =°Ð4Ô 5Ø ×.Ñ.×6Ñ6°wÓ?×?ˆFØ×#Ò#Ø ×+Ñ+Ð+Øùó
|
|||
|
|
Fð
|
|||
|
|
@ùô ò Ü Ð)¨'°A©+¨ °b¸¸Ð<Ô =Üûð üsS ‚.E±A8EÂ)D
|
|||
|
|
Â:EÃ0D Ã3DÃ4D ÄEÄD Ä E Ä"D;Ä6EÄ;E Å EÚinvestor_data_listc ƒ ó¤ K — | j sy |D �]Ì }t |j j |j j |j j
|
|||
|
|
|j j |j j |j j |j j |j j ¬« }| j j |« | j j « |j D ]ì }| j j t « j! t j |j k( « j# « }|sgt |j ¬« }| j j |« | j j « |j j% |« ŒÃ||j vsŒÒ|j j% |« Œî |j& D �] }| j j t( « j! t( j |j k( « j# « }|s“t) |j |j* |j, |j. |j0 ¬« } | j j | « | j j « |j& j% | « Œð||j& vsŒÿ|j& j% |« �Œ |j2 D ]® }
|
|||
|
|
| j j t4 « j! t4 j6 |
|
|||
|
|
j6 k( « j# « } | rŒ]t5 |
|
|||
|
|
j |
|
|||
|
|
j8 |
|
|||
|
|
j6 |j: ¬« } | j j | « Œ° �ŒÏ | j j= « t? dtA |« › d�« y# tB $ r.}
|
|||
|
|
| j jE « t? d|
|
|||
|
|
› �« ‚ d}
|
|||
|
|
~
|
|||
|
|
ww xY ww) z/Save investors and related data to SQL databaseN)r( r' ÚaumÚcheck_size_lowerÚcheck_size_upperÚgeographic_focusÚ stage_focusÚnumber_of_investments)r( )r( ÚindustryÚlocationÚ founded_yearÚwebsite)r( ÚroleÚemailÚ investor_idúSuccessfully saved z investors to databasez Error saving to SQL database: )#r r Úinvestorr( r' rY rZ r[ r\ r] r^ ÚaddÚflushÚsectorsÚqueryr ÚfilterÚfirstÚappendÚportfolio_companiesr r_ r` ra rb Ú team_membersr rd rc ÚidÚcommitrL ÚlenrN Úrollback)r3 rW r Ú db_investorÚ sector_dataÚexisting_sectorÚ db_sectorÚ company_dataÚexisting_companyÚ
|
|||
|
|
db_companyÚteam_member_dataÚexisting_memberÚdb_team_memberrU s r Ú _save_to_sqlz InvestorProcessor._save_to_sqlq s” è ø€ à× Ò Ø ðW Ø!3ó N
|
|||
|
|
=�
|
|||
|
|
ä +Ø&×/Ñ/×4Ñ4Ø -× 6Ñ 6× BÑ BØ%×.Ñ.×2Ñ2Ø%2×%;Ñ%;×%LÑ%LØ%2×%;Ñ%;×%LÑ%LØ%2×%;Ñ%;×%LÑ%LØ -× 6Ñ 6× BÑ BØ*7×*@Ñ*@×*VÑ*Vô � ð × Ñ ×$Ñ$ [Ô1Ø× Ñ ×&Ñ&Ô(ð $1×#8Ñ#8ò H�Kð ×(Ñ(×.Ñ.¬{Ó;ß ™¤ × 0Ñ 0°K×4DÑ4DÑ DÓEß ™›ð $ñ +Ü$/°[×5EÑ5EÔ$F˜ Ø ×(Ñ(×,Ñ,¨YÔ7Ø ×(Ñ(×.Ñ.Ô0à#×+Ñ+×2Ñ2°9Õ=ð +°+×2EÑ2EÒEØ '× /Ñ /× 6Ñ 6°Õ Gð#Hð( %2×$EÑ$Eó U�Lð ×(Ñ(×.Ñ.¬|Ó<ß ™¤ × 1Ñ 1°\×5FÑ5FÑ FÓGß ™›ð %ñ ,Ü%1Ø!-×!2Ñ!2Ø%1×%:Ñ%:Ø%1×%:Ñ%:Ø)5×)BÑ)BØ$0×$8Ñ$8ô &˜
|
|||
|
|
ð ×(Ñ(×,Ñ,¨ZÔ8Ø ×(Ñ(×.Ñ.Ô0ð $×7Ñ7×>Ñ>¸zÕJð ,°;×3RÑ3RÒRØ '× ;Ñ ;× BÑ BÐCSÖ Tð1Uð6 )6×(BÑ(Bò =Ð$ð ×(Ñ(×.Ñ.Ô/AÓBß ™Ô 2× 8Ñ 8Ð<L×<RÑ<RÑ RÓSß ™›ð $ò +Ü);Ø!1×!6Ñ!6Ø!1×!6Ñ!6Ø"2×"8Ñ"8Ø(3¯©ô *˜ð ×(Ñ(×,Ñ,¨^Õ<ò =ðN
|
|||
|
|
=ð`
|
|||
|
|
× Ñ × #Ñ #Ô %Ü Ð'¬Ð,>Ó(?Ð'@Ð@VÐWÕ Xøäò Ø × Ñ × %Ñ %Ô 'Ü Ð2°1°#Ð6Ô 7Ø ûð üs> ‚Q‘G P Ç D'P ÌBP Î B P ÐQÐ Q
|
|||
|
|
Ð )QÑQ
|
|||
|
|
Ñ
|
|||
|
|
Qc ƒ óJ K — | j syg }g }g }t |« D �]• \ }}|j }dj |j D �cg c] }|j
|
|||
|
|
‘Œ c}« } dj |j D �
|
|||
|
|
cg c] }
|
|||
|
|
|
|
|||
|
|
j
|
|||
|
|
‘Œ c}
|
|||
|
|
« } d|j
|
|||
|
|
› d|j xs d› d|j d›d|j d›d |j d›d
|
|||
|
|
|j › d |j j › d | › d
|
|||
|
|
| › d�j « } |j | « |j |j
|
|||
|
|
|j j |j |j dœ« |j d|› d|j
|
|||
|
|
j! dd« j! dd« › �« �Œ˜ |r8 | j" j% |||¬« t' dt) |« › d�« yyc c}w c c}
|
|||
|
|
w # t* $ r}
|
|||
|
|
t' d|
|
|||
|
|
› �« Y d}
|
|||
|
|
~
|