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

Ë
4êïh¶ãó<
—dd
lZdd
l
Z
dd
lZddlmZmZ
ddlmZmZ
ddl	m
Z
mZmZ
ddl
mZ
ddlmZ
ddlmZ
ddlmZ
dd	lmZ
dd
lmZmZmZmZ
ddlmZ
ddlmZ
e
j@e!«
Z"ejFd
«
Z$ejJe«
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/v1zopenai/gpt-4o-minir)Ú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
ddi
¬«}tj|›«

|d
dj}tjd|›�«

|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Úrecursion_limité2)
ÚconfigéÿÿÿÿzAI Response: 
)r+ÚinvokeÚloggerÚinfoÚcontentÚ_extract_fund_ids_from_responseÚ_fetch_funds_by_ids)r,r1r2ÚresponseÚfinal_message_contentÚfund_idss      r.Ú
process_queryzQueryProcessor.process_query4s–€ð—:‘:×$Ñ$Ø
˜6 8Ð,Ð-Ð.Ø% rÐ*ð%ó
ˆô	�‰�x�jÔ"Ø (¨Ñ 4°RÑ 8× @Ñ @ÐÜ�‰�oÐ&;Ð%<Ð=Ô>à×7Ñ7Ð8MÓNˆð×'Ñ'¨°*Ó=Ð=r0Úai_responsecó<
—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,rErGrCÚnumbersÚnumÚ
id_matchesÚid_strÚ
es	         r.r?z.QueryProcessor._extract_fund_ids_from_responseMs¥€ãàˆð
	ð—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ÂBrCc
ó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.¬
«‘Œ%}}