Anton_wireframe/app/services/__pycache__/querying.cpython-312.pyc

Ë
XçÂh‚'ãó
—dd
lm
Z
mZ
ddlZddlmZ
ddlmZ
ddlm	Z	
ddl
mZ
ddlm
Z

ddlmZ
dd	lmZmZ
dd
lmZ
ddlmZ
ej,d«
Zej0d
«
Zej5dd¬«dzZGd„d«Zy)é)ÚListÚOptionalN)
Ú
InvestorTable)
Úhub)
ÚSQLDatabaseToolkit)
ÚSQLDatabase)
Ú
ChatOpenAI)
Úcreate_react_agent)ÚInvestorDataÚInvestorList)
Úsettings)
Úselectinloadz$langchain-ai/sql-agent-system-promptzsqlite:///investors.dbÚSQLiteé)ÚdialectÚtop_kz;
 Get answers from the Sql database and the vector databasecó®—eZ
dZ		ddeedeefd„Zdedeefd„Zdedeefd„Z	d	edefd
„Z
dedeefd„Z
d	edefd
„Z	ddeededefd„Zy
)ÚQueryProcessorNÚsql_sessionÚvector_db_clientcó°
—|
|_ttjd
dd¬«|_tt|j¬«|_t|j|jj«|jg
zt¬«|_||_
tjd¬«
|_
|jj!d	d
di
¬«|_y)
Nzhttps://openrouter.ai/api/v1zgoogle/gemini-2.5-flash-liteg333333Ó?)Úapi_keyÚbase_urlÚmodelÚtemperature)ÚdbÚllm)rÚtoolsÚpromptz./chroma_db)
ÚpathÚinvestor_descriptionsÚdescriptionz1Investor descriptions and investment thesis focus)ÚnameÚmetadata)rr	r
ÚOPENROUTER_API_KEYrrrÚtoolkitr
Ú	get_toolsÚquery_vector_databaseÚsystem_messageÚagentrÚchromadbÚPersistentClientÚget_or_create_collectionÚ
collection)Úselfrrs   úL/home/oluwasanmi/Documents/Work/MKD/anton_wireframe/app/services/querying.pyÚ__init__zQueryProcessor.__init__s»€ð
'ˆÔÜÜ×/Ñ/Ø3Ø0Øô	
ˆŒô*¬R°T·X±XÔ>ˆŒÜ'Ø—(‘(Ø—,‘,×(Ñ(Ó*¨d×.HÑ.HÐ-IÑIÜ!ô
ˆŒ
ð
!1ˆÔä (× 9Ñ 9¸}Ô MˆÔØ×/Ñ/×HÑHØ(àÐRððI
ó
ˆ�óÚqueryÚreturncó¦—|js
y
|jj|
«
}|j«j«}t	|¬«
S)z0Query the SQL database for investor information.N©
Ú	investors)rÚexecuteÚscalarsÚallr)r/r3Úresultr7s    r0Úquery_sql_databasez!QueryProcessor.query_sql_database5sF€à×ÒØð×!Ñ!×)Ñ)¨%Ó0ˆØ—N‘NÓ$×(Ñ(Ó*ˆ	Ü iÔ0Ð0r2cóˆ—|js
y
td«

|jj|
g
d¬«}t|«

|S)z3Query the vector database for investor information.NzVECTOR STORE WAS CALLEDé)Úquery_textsÚ	n_results)rÚprintr.r3)r/r3Úresultss   r0r(z$QueryProcessor.query_vector_database?sJ€à×$Ò$ØÜ
Ð'Ô(ð—/‘/×'Ñ'ؘØð(ó
ˆô	ˆgŒðˆr2Úquestioncóô—|j
|
«
}|jjd
d|
fg
i
«
}|jd
«
r|d
djn
d}|j|«
}|j
||«S)zBProcess a query using the LLM and return structured investor data.ÚmessagesÚuseréÿÿÿÿÚ)Ú_extract_filters_from_queryr*ÚinvokeÚgetÚcontentÚ$_extract_investor_info_from_responseÚ#_fetch_investors_with_relationships)r/rCÚfiltersÚresponseÚai_responseÚinvestor_idss      r0Ú
process_queryzQueryProcessor.process_queryPsˆ€ð×2Ñ2°8Ó<ˆð—:‘:×$Ñ$Ø
˜6 8Ð,Ð-Ð.ó
ˆð19·±¸ZÔ0HˆH�ZÑ  Ñ$×,Ò,Èbð	ð
×@Ñ@ÀÓMˆð×7Ñ7¸ÀgÓNÐNr2rQcó´—g}	d
dl}|jd|
j««}|D�cgc]
}t|«
‘Œ}}|r|SgSc
c}w#t$r
YŒwxYw
)zGExtract investor IDs from AI response. This is a simple implementation.rNz
\bid:\s*(\d+))ÚreÚfindallÚlowerÚintÚ	Exception)r/rQrRrUÚidsÚid_strs      r0rMz3QueryProcessor._extract_investor_info_from_responseesh€ðˆð	ãà—*‘*Ð-¨{×/@Ñ/@Ó/BÓCˆCØ69Ö:¨FœC �KÐ:ˆLÐ:ñ ,ˆ|Ð3°Ð3ùò	;øÜò
	Ùð
	ús!„(A¬A¾AÁAÁ	AÁ
Acój‡—|
j
«Ši}tˆf
d
„dD««
r;d‰vrd|d<n1d‰vrd|d<n'd‰vrd	|d<nd
‰vrd|d<nd‰vrd
|d<n	d‰vrd|d<tˆf
d„dD««
r3d‰vsd‰vsd‰vrd|d<n!d‰vrd|d<nd‰vrd|d<n
d‰vsd‰vrd|d<gd¢
}|D]
}|‰vs
Œ||d<
n
d d!l}|jd"‰«}|r[|d j	d#d$«}d%‰vsd&‰vrtt
|«
d'z«
|d(<|Sd)‰vsd*‰vrtt
|«
d+z«
|d(<|S),z4Extract filter criteria from natural language query.c
3ó&•
K
—|]}
|
‰v–
—

Œ
y­w
©
N©)Ú.0ÚstageÚquestion_lowers  €r0ú	<genexpr>z=QueryProcessor._extract_filters_from_query.<locals>.<genexpr>~søèø€ò

àð
�^Ô#ñ

ùóƒ
)Úseedúseries aúseries búseries cÚgrowthú
late stagereÚSEEDrarfÚSERIES_ArgÚSERIES_BrhÚSERIES_CriÚGROWTHrjÚ
LATE_STAGEc
3ó&•
K
—|]}
|
‰v–
—

Œ
y­w
r^r_)r`Úgeorbs  €r0rcz=QueryProcessor._extract_filters_from_query.<locals>.<genexpr>—søèø€ò
àð
�>Ô!ñ
ùrd)ÚusÚusaú
united statesÚeuropeÚasiaúsilicon valleyúbay arearsrtruÚUSÚ	geographyrvÚEuroperwÚAsiarxryzSilicon Valley)	ÚfintechÚ
healthcareÚsaasÚaiÚbiotechÚconsumerÚ
enterpriseÚcryptoÚ
blockchainÚsectorrNz8\$?(\d+(?:,\d{3})*(?:\.\d+)?)\s*(?:million|m|k|thousand)ú
,rHÚmillionÚ
mi@BÚmin_check_sizeÚthousandÚ
kiè)rWÚanyrUrVÚreplacerXÚfloat)	r/rCrOÚsectorsr‡rUÚamountsÚamountrbs	        @r0rIz*QueryProcessor._extract_filters_from_queryxsÓ
ø€à!Ÿ™Ó)ˆØˆôó

ðô

ô

ð˜Ñ'Ø#)�˜Ò ؘ~Ñ-Ø#-�˜Ò ؘ~Ñ-Ø#-�˜Ò ؘ~Ñ-Ø#-�˜Ò ؘ^Ñ+Ø#+�˜Ò Ø Ñ/Ø#/�˜Ñ ôó
ðô
ô
ð˜Ñ&ؘNÑ*Ø" nÑ4à'+�˜Ò$ؘ^Ñ+Ø'/�˜Ò$ؘ>Ñ)Ø'-�˜Ò$Ø! ^Ñ3°zÀ^Ñ7SØ'7�˜Ñ$ò

ˆðò	ˆFؘÒ'Ø$*�˜Ñ!Ùð	ó	à—*‘*ØGÈó
ˆñؘQ‘Z×'Ñ'¨¨RÓ0ˆFؘNÑ*¨c°^Ñ.CÜ,/´°f³
ÀÑ0GÓ,H�Ð(Ñ)ðˆð˜~Ñ-°¸Ñ1FÜ,/´°f³
ÀÑ0DÓ,E�Ð(Ñ)àˆr2rRrOcóP—|jstg¬
«
Sddlm}
|jj	t
«
j
tt
j«
tt