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

Ë
®×h‰ãóè—dd
lZddl
mZ
ddlmZmZ
ddlmZ
ddlm	Z	
ddl
mZ
ddlm
Z

ddlmZ
dd	lmZ
dd
lmZmZ
ddlmZ
e	j.d«
Ze
j2e«
ZGd
„d«Zy
)éN)
ÚList)ÚDATABASE_URLÚget_db)
Ú
InvestorTable)
Úhub)
ÚSQLDatabaseToolkit)
ÚSQLDatabase)
Ú
ChatOpenAI)
Úcreate_react_agent)ÚInvestorDataÚInvestorList)
Úselectinloadz$langchain-ai/sql-agent-system-promptcóN—eZ
dZd
„Zdedefd„Zdedeefd„Z	deedefd„Z
y	)
ÚQueryProcessorc
óB
—t
tjd
«
ddd¬«|_t	t
|j¬«|_tjdd¬	«d
zdzdzd
z}
t|j|jj«|
¬«|_y)NÚOPENROUTER_API_KEYzhttps://openrouter.ai/api/v1zopenai/gpt-4o-minir)Úapi_keyÚbase_urlÚmodelÚtemperature)ÚdbÚllmÚSQLiteé)ÚdialectÚtop_kz^

IMPORTANT: You must ONLY return the investor IDs (id field) that match the user's criteria. z<Do NOT return any other information, explanations, or data. z^Your response should be ONLY a comma-separated list of numbers representing the investor IDs. zExample format: 1, 5, 12, 23)rÚtoolsÚprompt)r
ÚosÚgetenvrrrÚtoolkitÚprompt_templateÚformatrÚ	get_toolsÚagent)ÚselfÚsystem_message_updateds  úL/home/oluwasanmi/Documents/Work/MKD/anton_wireframe/app/services/querying.pyÚ__init__zQueryProcessor.__init__s¢€ÜÜ—I‘IÐ2Ó3Ø3Ø&Øô	
ˆŒô*¬R°T·X±XÔ>ˆŒô
×"Ñ"¨8¸1Ð"Ó=Øpñ

q
àLñ
M
ðo
ñ
o
ð-ñ	
-ð	ô(Ø—(‘(Ø—,‘,×(Ñ(Ó*Ø)ô
ˆ�
óÚquestionÚreturncóЗ|jjd
d|
fg
i
«
}|jd
«
r|d
djn
d}|j	|«
}|j|«
S)z7Process a query using the LLM and return investor data.ÚmessagesÚuseréÿÿÿÿÚ)r%ÚinvokeÚgetÚcontentÚ#_extract_investor_ids_from_responseÚ_fetch_investors_by_ids)r&r+ÚresponseÚai_responseÚinvestor_idss     r(Ú
process_queryzQueryProcessor.process_query*st€ð—:‘:×$Ñ$Ø