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

Ë
Z
i&ãóØ—dd
lZdd
l
Z
dd
lZddlmZmZ
ddlmZ
ddlm	Z	m
Z
mZ
ddlm
Z

ddlmZ
ddlmZmZmZmZ
ddlmZ
dd	lmZ
dd
lmZ
e
j6e«
ZGd„d«Zy
)
éN)ÚListÚOptional)
Úget_db)Ú	FundTableÚ
InvestorTableÚProjectTable)
ÚChatPromptTemplate)
Ú
ChatOpenAI)ÚCompanyMinimalÚInvestmentResponseÚPaginatedResponseÚ
SectorMinimal)
Útext)
Úselectinload)
Ú(calculate_project_investor_compatibilitycóp—eZ
dZd
„Zdedefd„Z	d
dedeedee	fd„Z
	d
deedeedee	fd	„Zy)ÚQueryProcessorc
ó˜—t
tjd
«
ddd¬«|_i|_tjddg«
|_y)NÚOPENROUTER_API_KEYzhttps://openrouter.ai/api/v1zopenai/gpt-4o-minir)Úapi_keyÚbase_urlÚmodelÚtemperature)Úsystemu\You are a SQL expert. Generate a SQLite query to find fund IDs based on user requirements.

Database Schema:
- funds: id, fund_name, investor_id, check_size_lower, check_size_upper, geographic_focus
- fund_sectors: fund_id, sector_id
- fund_investment_stages: fund_id, stage_id
- sectors: id, name
- investment_stages: id, name
- investors: id, name, aum

IMPORTANT RULES:
1. ALWAYS return ONLY fund IDs (funds.id) - use SELECT DISTINCT f.id
2. For geography: use LIKE '%keyword%' for partial matching
   - 'Europe' → WHERE geographic_focus LIKE '%Europe%'
   - 'America' → WHERE geographic_focus LIKE '%America%'
   - 'Asia' → WHERE geographic_focus LIKE '%Asia%'
3. For stages: JOIN fund_investment_stages and investment_stages
   - Use LIKE for stage matching: s.name LIKE '%Seed%'
4. For sectors: JOIN fund_sectors and sectors
   - Use LIKE for sector matching: sec.name LIKE '%Fintech%'
5. For check size filters:
   - "under X" → WHERE check_size_upper <= X
   - "over X" → WHERE check_size_lower >= X
   - "between X and Y" → WHERE check_size_lower >= X AND check_size_upper <= Y
6. Use DISTINCT to avoid duplicates from joins
7. Return a single, complete SELECT query

Example Queries:
Q: "Seed stage investors in Europe"
A: SELECT DISTINCT f.id FROM funds f 
   JOIN fund_investment_stages fis ON f.id = fis.fund_id 
   JOIN investment_stages s ON fis.stage_id = s.id 
   WHERE s.name LIKE '%Seed%' AND f.geographic_focus LIKE '%Europe%'

Q: "Fintech investors with check size under 5 million"
A: SELECT DISTINCT f.id FROM funds f 
   JOIN fund_sectors fs ON f.id = fs.fund_id 
   JOIN sectors sec ON fs.sector_id = sec.id 
   WHERE sec.name LIKE '%Fintech%' AND f.check_size_upper <= 5000000

Q: "Growth stage investors"
A: SELECT DISTINCT f.id FROM funds f 
   JOIN fund_investment_stages fis ON f.id = fis.fund_id 
   JOIN investment_stages s ON fis.stage_id = s.id 
   WHERE s.name LIKE '%Growth%'

Q: "Investors in North America"
A: SELECT DISTINCT f.id FROM funds f 
   WHERE f.geographic_focus LIKE '%North America%' OR f.geographic_focus LIKE '%America%'

Return ONLY the SQL query, no explanations or markdown.)Úuserz
{question})r
ÚosÚgetenvÚllmÚquery_cacher	Ú
from_messagesÚ
sql_prompt)
Úselfs
 úL/home/oluwasanmi/Documents/Work/MKD/anton_wireframe/app/services/querying.pyÚ__init__zQueryProcessor.__init__sS€ÜÜ—I‘IÐ2Ó3Ø3Ø&Øô	
ˆŒðˆÔô-×:Ñ:ð5ðl
'ðo
8
ó:
ˆ�óÚquestionÚreturncóœ—t
j|
j«j«j	««
j«S)
z,Generate cache key from normalized question.)ÚhashlibÚmd5ÚlowerÚstripÚencodeÚ	hexdigest)r"r&s  r#Ú_get_cache_keyzQueryProcessor._get_cache_keyas3€ä�{‰{˜8Ÿ>™>Ó+×1Ñ1Ó3×:Ñ:Ó<Ó=×GÑGÓIÐIr%NÚ
project_idc	ó2—|j
|
«
}||jvr(|j|}tjd
|›�«

n¨|jj|
¬«
}|jj|«
}|jj«}|jdd«jdd«j«}||j|<tjd|›�«

tt««
}	|jt|«
«
}|j«D�	cgc]}	|	d‘Œ	}
}	tjdt!|
«
›d	|
d
d›t!|
«
dkDrdn
d›�«

|j#|
|«|j%«
Sc
c}	w#t&$rZ}tj)d
|›�«

tj)d|›�«

t+gdddd¬«cYd
}~|j%«
Sd
}~w
wxYw
#|j%«
wxYw
)zÓProcess a query by generating and executing SQL directly.

        Args:
            question: The natural language query to process
            project_id: Optional project ID for compatibility scoring
        zUsing cached SQL: )
r&z```sqlÚz```zGenerated SQL: rzFound z fund IDs: Né
z...zSQL execution error: zFailed SQL: é
©ÚitemsÚtotalÚpageÚ	page_sizeÚtotal_pages)r/rÚloggerÚinfor!Úformat_messagesrÚinvokeÚcontentr,ÚreplaceÚnextrÚexecuterÚfetchallÚlenÚ_fetch_funds_by_idsÚcloseÚ	ExceptionÚerrorr
)r"r&r0Ú	cache_keyÚ	sql_queryÚmessagesÚresponseÚ
db_sessionÚresultÚrowÚfund_idsÚ
es            r#Ú
process_queryzQueryProcessor.process_queryesç
€ð×'Ñ'¨Ó1ˆ	ð˜×(Ñ(Ñ(Ø×(Ñ(¨Ñ3ˆIÜ�K‰KÐ,¨Y¨KÐ8Õ9ð—‘×6Ñ6ÀÐ6ÓIˆHØ—x‘x—‘ xÓ0ˆHØ ×(Ñ(×.Ñ.Ó0ˆIð"×)Ñ)¨(°BÓ7×?Ñ?ÀÀrÓJ×PÑPÓRˆIð+4ˆD×Ñ˜YÑ'Ü�K‰K˜/¨)¨Ð5Ô6ôœ&›(“^ˆ
ð	Ø×'Ñ'¬¨Y«Ó8ˆFØ*0¯/©/Ó*;Ö< 3˜˜A›Ð<ˆHÐ<Ü�K‰KØœ˜X›˜ {°8¸C¸R°=°/Ì3ÈxË=Ð[]ÒK]Á%ÐceÐAfÐgô
ð×+Ñ+¨H°jÓAð
×ÑÕùò=øôò	Ü�L‰LÐ0°°Ð4Ô5Ü�L‰L˜<¨	 {Ð3Ô4ä$Ø 
¨
°RÀQôô
ð
×ÑÕûð	ûð
×ÑÕúsCÄ,FÄ0FÄ<AFÆFÆ	H
Æ'?G<Ç&
H
Ç'HÇ<H
È
HÈHrPc
ó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}
|j"r|j"dd	n
gD�cgc]#}t3|j|j.¬