Ji. d dl mZmZmZ d dlZd dlmZmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZmZ d dlmZmZmZ d d
lmZ d d l m Z m Z m Z G d d
e Z y) )AnyListOptionalN)CompositeRetrievalMode#CompositeRetrievedTextNodeWithScoreRetrieverCreate RetrieverRetrieverPipelinePresetRetrievalParams ReRankConfig)OMIT)
BaseRetriever)DEFAULT_PROJECT_NAME) get_aclient
get_client)
NodeWithScore QueryBundleTextNode )LlamaCloudIndex)resolve_projectresolve_retriever(page_screenshot_nodes_to_node_with_scorec % b e Zd Zddedddddddddddddfdee dee dee dee d ee d
ed ee d ee d
ee dedeej deej dee dee dee dee de
ddf$ fd
Zedee fd Zdee defdZ d(dedee d ee d ee def
d Zdedefd Zdee defd Z d(dedee d ee d ee def
d!Zdedefd"Zd#e de fd$Z d(d%e dee dee dee dee f
d&Z d(d%e dee dee dee dee f
d'Z! xZ"S )) LlamaCloudCompositeRetrieverNF< Tname retriever_id project_name
project_idorganization_idcreate_if_not_existsapi_keybase_urlapp_urltimeout httpx_clientasync_httpx_clientmode rerank_top_n
rerank_config persistedkwargsreturnc t ||| |
| | _ t ||| |
| | _ t | j ||| | _ || _ | j
j | _ || _ t | j | j
||| | _
| j s|rq|rV| j j j | j
j t | j g | _
nt d| j d |
|
nt | _ ||nt | _ ||nt | _ t( | U |j- d |j- dd y)
z#Initialize the Composite Retriever.N)r pipelines)r requestzRetriever with name 'z ' does not exist in project.callback_managerverboseF)r2 r3 )r _clientr _aclientr projectr r
_persistedr retriever
retrieversupsert_retrieveridr
ValueErrorr
_mode
_rerank_top_n_rerank_configsuper__init__get)selfr r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- __class__s /var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/llama_cloud_services/index/composite_retriever.pyrA z%LlamaCloudCompositeRetriever.__init__ sT 6 "'8Wg|T # Xw1C
' LL,
O
LL - -#* LL$,,lI
>> !i#!% !8!8!I!I #|| +b I "J "
!+DII;6RS
" -T4
-9-E\4/>+U+U $ / #(C
6:5M5M5SQS/
(0HMM8 #/
+ /
, # #
.. 4;;= >
/
s B#c | j j xs g }|D cg c] }|j |k7 r| }}t | t | k( ry| j | yc c}w NFT)r8 r0 r lenrQ rC r current_retriever_pipeline_namesrX new_retriever_pipeliness rE remove_indexz)LlamaCloudCompositeRetriever.remove_index ss +/>>+C+C+Ir( =#
}}$
#
#
& '3/O+P P ''(?@#
s A#c , K | j rT| j j j | j j
| d { | _ | j S | j j
d|i | _ | j S 7 DwrK )r7 r5 r9 rN r8 r; rO rP s rE aupdate_retriever_pipelinesz8LlamaCloudCompositeRetriever.aupdate_retriever_pipelines s ??#'==#;#;#L#L!!Y $M $ DN ~~ "^^ 0 0i8P 0 QDN~~
s A B
BABc l K |xs |j }|xs |j j }t |j ||| }| j
xs g D ci c] }|j | }}|||j < | j
t |j d { S c c}w 7 wrV ) r rX rT r
r; rI rg rY rZ r[ s rE async_add_indexz,LlamaCloudCompositeRetriever.async_add_index s !uzz ' U5>>+U+U $ / #(C
6:5M5M5SQS/
(0HMM8 #/
+ /
, # #
55 4;;= >
/
s AB4B-,>+C+C+Ir( =#
}}$
#
#
& '3/O+P P../FGGG#
Hs A4A-.A4'A2( A4composite_retrieval_nodec t t |j j |j j |j j
|j S )N)r; textmetadata)nodescore)r r rp r; rn ro rq )rC rl s rE _result_nodes_to_node_with_scorez=LlamaCloudCompositeRetriever._result_nodes_to_node_with_score sQ +0033-2277166??
+00
rF query_bundlec ||n | j }||n | j }||n | j }|5|t k7 r,| |t k( r
t | }n|j d|i }| j rH| j j j | j j |||j }n\| j j j | j j |||j | j j }|j D cg c] }| j# | }}t% | j |j& | j j( }t+ ||z d d S c c}w ) Ntop_nrv rL r) r+ queryr r) r+ rx r0 c | j S rH rq xs rE z8LlamaCloudCompositeRetriever._retrieve..
rF Tkeyreverse)r= r> r? r
r rO r7 r4 r9 retriever8 r; query_strdirect_retriever6 r0 nodesrr r image_nodesr sorted rC rs r) r* r+ resultrp
node_w_scoresimage_nodes_w_scoress rE _retrievez&LlamaCloudCompositeRetriever._retrieve s~ 'tTZZ'3'?|TEWEW * 6MD.. r rF Tr )r= r> r? r
r rO r7 r5 r9 r r8 r; r r r6 r0 r rr r r r r r s rE
_aretrievez'LlamaCloudCompositeRetriever._aretrieve s 'tTZZ'3'?|TEWEW * 6MDF" FA$F"%F&F"8F A
F"F" F")NNN)#__name__
__module__ __qualname__r r strboolinthttpxClient AsyncClientr r r rA propertyr r
rI r rQ r r r^ re rg ri rk r r rr r r r
__classcell__)rD s @rE r r s
#&*&:$()-%*!%"&!%/3:>15&*04$(-?
sm?
sm ?
sm
?
SM?
"# ?
#?
# ?
3-?
# ?
?
u||,!?
" %U%6%6 7#?
& -.'?
( sm)?
* -+?
, D>-?
. /?
0
1?
B .T*;%< . .
/0
#%)GK
sm
c]
&..C%D
6
/0
#%)GK
sm
c]
&..C%D
6
(K
26&*04 -
!-
-.-
sm -
- -
m
-
d 26&*04 -
!-
-.-
sm -
- -
m
-
rF r )!typingr r r r llama_cloudr r r r r
r r &llama_cloud.resources.pipelines.clientr
$llama_index.core.base.base_retrieverr llama_index.core.constantsr $llama_index.core.ingestion.api_utilsr r llama_index.core.schemar r r baser api_utilsr r r r rF rE r sL & & 8 > ; H H H ! T
= T
rF