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

Ë
•·hÇãóô—dd
lm
Z

ddlZddlmZ
ddlmZ
ddlmZ
ddl	m
Z

ddlmZ
ddl
mZ
dd	lmZ
ej d
«
Zej$d«
Zej)dd
¬«dzZGd„d«Zy)é)
ÚOptionalN)
Úhub)
ÚSQLDatabaseToolkit)
ÚSQLDatabase)
Ú
ChatOpenAI)
Úcreate_react_agent)
ÚInvestorList)
Úsettingsz$langchain-ai/sql-agent-system-promptzsqlite:///investors.dbÚSQLiteé)ÚdialectÚtop_kz;
 Get answers from the Sql database and the vector databasecój—eZ
dZ		ddeedeefd„Zdedeefd„Zdedeefd„Z	d	edefd
„Z
y
)ÚQueryProcessorNÚsql_sessionÚvector_db_clientcó¬
—t
tjd
dd¬«|_t	t
|j¬«|_t|j|jj«|jg
ztt¬«|_||_
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ÚpromptÚresponse_formatz./chroma_db)
ÚpathÚinvestor_descriptionsÚdescriptionz1Investor descriptions and investment thesis focus)ÚnameÚmetadata)rr
ÚOPENROUTER_API_KEYrrrÚtoolkitrÚ	get_toolsÚquery_vector_databaseÚsystem_messager	Ú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,r0Úresultr3s    r-Úquery_sql_databasez!QueryProcessor.query_sql_database3sF€à×ÒØð×!Ñ!×)Ñ)¨%Ó0ˆØ—N‘NÓ$×(Ñ(Ó*ˆ	Ü iÔ0Ð0r/cóˆ—|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+r0)r,r0Úresultss   r-r%z$QueryProcessor.query_vector_database=sJ€à×$Ò$ØÜ
Ð'Ô(ð—/‘/×'Ñ'ؘØð(ó
ˆô	ˆgŒðˆr/ÚquestioncóF—|jjd
d|
fg
i
«
}|S)zBProcess a query using the LLM and return structured investor data.ÚmessagesÚuser)r'Úinvoke)r,r?Úresponses   r-Ú
process_queryzQueryProcessor.process_queryNs-€à—:‘:×$Ñ$Ø