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

Ë
ä\æhTãóô—dd
lZddl
mZ
ddlmZmZ
ddlmZmZ
ddl	m
Z

ddlmZ
ddl
mZ
ddlmZ
dd	lmZ
dd
lmZmZmZmZ
ddlmZ
e
j4d«
Zej8e«
ZGd
„d«Zy
)éN)
ÚList)ÚDATABASE_URLÚget_db)Ú	FundTableÚ
InvestorTable)
Úhub)
ÚSQLDatabaseToolkit)
ÚSQLDatabase)
Ú
ChatOpenAI)
Úcreate_react_agent)ÚCompanyMinimalÚInvestmentResponseÚPaginatedResponseÚ
SectorMinimal)
Úselectinloadz$langchain-ai/sql-agent-system-promptcóZ—eZ
dZd
„Zdedeefd„Zdedee	fd„Z
dee	deefd„Zy	)
Ú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/v1zx-ai/grok-4-fastr)Úapi_keyÚbase_urlÚmodelÚtemperature)ÚdbÚllmÚSQLiteé)ÚdialectÚtop_kzo

IMPORTANT: You must ONLY return the fund IDs (id field from the funds table) that match the user's criteria. z<Do NOT return any other information, explanations, or data. zZYour response should be ONLY a comma-separated list of numbers representing the fund IDs. zExample format: 1, 5, 12, 23)rÚtoolsÚprompt)rÚosÚgetenvrr	rÚ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Ð"Ó=ðBñ

BàLñ
M
ðk
ñ
k
ð-ñ	
-ð	ô(Ø—(‘(Ø—,‘,×(Ñ(Ó*Ø)ô
ˆ�
óÚquestionÚreturncóЗ|jjd
d|
fg
i
«
}|jd
«
r|d
djn
d}|j	|«
}|j|«
S)zBProcess a query using the LLM and return investment response data.ÚmessagesÚuseréÿÿÿÿÚ)r(ÚinvokeÚgetÚcontentÚ_extract_fund_ids_from_responseÚ_fetch_funds_by_ids)r)r.ÚresponseÚai_responseÚfund_idss     r+Ú
process_queryzQueryProcessor.process_query/st€ð—:‘:×$Ñ$Ø
˜6 8Ð,Ð-Ð.ó
ˆð19·±¸ZÔ0HˆH�ZÑ  Ñ$×,Ò,Èbð	ð
×7Ñ7¸ÓDˆð×'Ñ'¨Ó1Ð1r-r;có<
—d
dl}g}	|jd|
«}|D�cgc]
}t|«
‘Œ}}|jd|
j««}|r|D�cgc]
}t|«
‘Œ}}|Sc
c}wc
c}w#t$r}td|›�«

gcYd}~Sd}~w
wxYw
)z"Extract fund IDs from AI response.rNz\b\d+\bz\bid[:\s]*(\d+)z$Error extracting IDs from response: )ÚreÚfindallÚintÚlowerÚ	ExceptionÚprint)	r)r;r?r<ÚnumbersÚnumÚ
id_matchesÚid_strÚ
es	         r+r8z.QueryProcessor._extract_fund_ids_from_responseAs¥€ãàˆð
	ð—j‘j ¨[Ó9ˆGØ,3Ö4 Sœ˜C�Ð4ˆHÐ4ðŸ™Ð$6¸×8IÑ8IÓ8KÓLˆJÙØ6@ÖA¨FœC �KÐA�ÐAðˆùò5ùò
B
øäò	ÜÐ8¸¸Ð<Ô=Ø�Iûð	ús9ˆA8žA.°(A8ÁA3Á*A8Á.
A8Á8	BÂ
BÂ
BÂBr<c
óF—|
s#t
gd
d|
rt|
«
d
¬«Sdd
¬«Stt««
}	|j	t
«
j
tt
j«
jtj«
tt
j«
jtj«
tt
j«
jtj«
tt
j«
tt
j«
«jt