27 lines
3.9 KiB
Plaintext
27 lines
3.9 KiB
Plaintext
|
Ë
|
|||
|
|
E¶hÕ ã óø — d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
|
|||
|
|
Z
|
|||
|
|
d dl m Z d dl
|
|||
|
|
mZmZ d d lmZ ej" d
|
|||
|
|
« Z ej& d « Zej+ d d
|
|||
|
|
¬« dz Z G d„ d« Zy)é )ÚOptionalN)Úhub)ÚSQLDatabaseToolkit)Ú SQLDatabase)Ú
|
|||
|
|
ChatOpenAI)Úcreate_react_agent)ÚInvestorÚ 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 — e Zd Z d dee dee fd„Zdedee fd„Zdedee fd„Z d edefd
|
|||
|
|
„Z
|
|||
|
|
y) ÚQueryProcessorNÚ sql_sessionÚvector_db_clientc ó¬ — t t j ddd¬« | _ t t
|
|||
|
|
| j ¬« | _ t | j | j j « | j gz t t ¬« | _ || _
|
|||
|
|
t j d¬« | _
|
|||
|
|
| j j! d d
|
|||
|
|
d i¬ « | _ y )
|
|||
|
|
Nz https://openrouter.ai/api/v1z google/gemini-2.5-flash-liter )Ú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)r r ÚOPENROUTER_API_KEYr r r Útoolkitr Ú get_toolsÚquery_vector_databaseÚsystem_messager
|
|||
|
|
Úagentr ÚchromadbÚPersistentClientÚget_or_create_collectionÚ
|
|||
|
|
collection)Úselfr r s ú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 ó¦ — | j sy| j j |« }|j « j « }t |¬« S )z0Query the SQL database for investor information.N)Ú investors)r ÚexecuteÚscalarsÚallr
|
|||
|
|
)r- r1 Úresultr4 s r. Úquery_sql_databasez!QueryProcessor.query_sql_database3 sF € à× Ò Øð ×!Ñ!×)Ñ)¨%Ó0ˆØ—N‘NÓ$×(Ñ(Ó*ˆ Ü iÔ0Ð0r0 c óˆ — | j syt d« | j j |gd¬« }t |« |S )z3Query the vector database for investor information.NzVECTOR STORE WAS CALLEDé )Ú query_textsÚ n_results)r Úprintr, r1 )r- r1 Úresultss r. r& z$QueryProcessor.query_vector_database= sJ € à×$Ò$ØÜ
|
|||
|
|
Ð'Ô(ð —/‘/×'Ñ'Ø ˜Øð (ó
|
|||
|
|
ˆô ˆgŒð ˆr0 Úquestionc óF — | j j dd|fgi« }|S )zBProcess a query using the LLM and return structured investor data.ÚmessagesÚuser)r( Úinvoke)r- r@ Úresponses r. Ú
|
|||
|
|
process_queryz QueryProcessor.process_queryN s- € à—:‘:×$Ñ$Ø
|