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

Ë
ˆyíhãó
—dd
lZddl
mZmZ
ddlmZmZ
ddlmZm	Z	m
Z

ddlmZ
ddl
mZ
ddlmZ
ddlmZ
dd	lmZ
dd
lmZmZmZmZ
ddlmZ
ddlmZ
ej<d
«
Zej@e«
Z!Gd„d«Z"y
)éN)ÚListÚOptional)ÚDATABASE_URLÚget_db)Ú	FundTableÚ
InvestorTableÚProjectTable)
Úhub)
ÚSQLDatabaseToolkit)
ÚSQLDatabase)
Ú
ChatOpenAI)
Úcreate_react_agent)ÚCompanyMinimalÚInvestmentResponseÚPaginatedResponseÚ
SectorMinimal)
Úselectinload)
Ú(calculate_project_investor_compatibilityz$langchain-ai/sql-agent-system-promptcóv—eZ
dZd
„Z	ddedeedeefd„Z	dede
efd„Z	dd	e
edeedeefd
„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/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Ú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Ð"Ó=ðBñ

BàLñ
M
ðk
ñ
k
ð-ñ	
-ð	ô(Ø—(‘(Ø—,‘,×(Ñ(Ó*Ø)ô
ˆ�
óNÚquestionÚ
project_idÚreturncóÒ—|jjd
d|
fg
i
«
}|jd
«
r|d
djn
d}|j	|«
}|j||«S)zÜProcess a query using the LLM and return investment response data.

        Args:
            question: The natural language query to process
            project_id: Optional project ID for compatibility scoring
        ÚmessagesÚuseréÿÿÿÿÚ)r+ÚinvokeÚgetÚcontentÚ_extract_fund_ids_from_responseÚ_fetch_funds_by_ids)r,r1r2ÚresponseÚai_responseÚfund_idss      r.Ú
process_queryzQueryProcessor.process_query1sv€ð—:‘:×$Ñ$Ø
˜6 8Ð,Ð-Ð.ó
ˆð19·±¸ZÔ0HˆH�ZÑ  Ñ$×,Ò,Èbð	ð
×7Ñ7¸ÓDˆð×'Ñ'¨°*Ó=Ð=r0r?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?rCr@ÚnumbersÚnumÚ
id_matchesÚid_strÚ
es	         r.r<z.QueryProcessor._extract_fund_ids_from_responseJs¥€ãàˆð
	ð—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
óD—|
s#t
gd
d|
rt|
«
d
¬«Sdd
¬«Stt««
}	d}|�i|j	t
«
j
tt
j«
«
jt
j|k(«
j«}|j	t«
j
ttj«
jtj«
ttj«
jtj «
ttj«
jtj"«
ttj$«
ttj"«
«jtjj'|
«
«
j)«}g}|D�
]o}|j}d}	|�t+||d¬«}	|jdd	D�
cgc]#}
t-|
j|
j.¬
«‘Œ%}}
|j$r3dj1|j$D�cgc]}|j.‘Œc}«
n
d}