iU, S SK JrJrJr S SKrS SKJrJrJrJ r J
r
J r J r S SK
Jr S SKJr S SKJr S SKJrJr S SKJrJrJr S S
KJr S S K J r J r J r " S S
\5 r g) )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 % ^ \ rS rSrSS\SSSSSSSSSSSSS4S\\ S\\ S\\ S \\ S
\\ S \S \\ S
\\ S\\ S\S\\ R S\\ R S\\ S\\ S\\
S\\ S\SS4$U 4S jjjr\S\\ 4S j5 rS\\ S\4S jr S*S \S\\ S \\ S \\ S\4
S jjrS\S\4S jrS\\ S\4S! jr S*S \S\\ S \\ S \\ S\4
S" jjrS\S\4S# jr S$\ S\ 4S% jr S*S&\ S\\ S\\ S\\
S\\ 4
S' jjr! S*S&\ S\\ S\\ S\\
S\\ 4
S( jjr"S)r#U =r$$ )+ LlamaCloudCompositeRetriever NF< Tname retriever_id project_name
project_idorganization_idcreate_if_not_existsapi_keybase_urlapp_urltimeout httpx_clientasync_httpx_clientmode rerank_top_n
rerank_config persistedkwargsreturnc > [ XxXU 5 U l [ XxXU 5 U l [ U R X4U5 U l Xl U R
R U l UU l [ U R U R
XU5 U l
U R cw U( ap U( aP U R R R U R
R [ U R / S9S9U l
O[ SU R S35 eU
b U
O[ U l Ub UO[ U l Ub UO[ U l [( TU ]U UR- S5 UR- SS5 S 9 g)
z#Initialize the Composite Retriever.N)r pipelines)r! requestzRetriever with name 'z ' does not exist in project.callback_managerverboseF)r3 r4 )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 h/home/james-whalen/.local/lib/python3.13/site-packages/llama_cloud_services/index/composite_retriever.pyrB %LlamaCloudCompositeRetriever.__init__ sF 6 "'W|T # w1C
' LL,O
LL - -#* LL$,,I
>> !i#!% !8!8!I!I #|| +b I "J "
!+DII;6RS
" -T4
-9-E\4/>+U+U $ / #(C
6:5M5M5SQS5S/
5SMM8 #5S , /
, # #
.. 4;;= >
/
s "B1c U R R =( d / nU Vs/ s H nUR U:w d M UPM nn[ U5 [ U5 :X a gU R U5 gs snf NFT)r9 r1 r lenrS rD r current_retriever_pipeline_namesr[ new_retriever_pipeliness rF remove_index)LlamaCloudCompositeRetriever.remove_index su +/>>+C+C+Ir( =#
<}}$
< #
& '3/O+P P ''(?@#
s
A/A/c * # U R ( aQ U R R R U R R
US9I S h vN U l U R $ U R R
SU0S9U l U R $ NA7frN )r8 r6 r: rP r9 r< rQ rR s rF aupdate_retriever_pipelines8LlamaCloudCompositeRetriever.aupdate_retriever_pipelines s ??#'==#;#;#L#L!!Y $M $ DN ~~ "^^ 0 0i8P 0 QDN~~
s A
BBABc # U=( d UR nU=( d UR R n[ UR UUUS9nU R
=( d / Vs0 s H ofR U_M nnUUUR ' U R
[ UR 5 5 5 I S h vN $ s snf N 7frY ) r r[ rW r
r< rK rl r\ r] r^ s rF async_add_index,LlamaCloudCompositeRetriever.async_add_index s !uzz ' U5>>+U+U $ / #(C
6:5M5M5SQS5S/
5SMM8 #5S , /
, # #
55 4;;= >
/
s A"C$B;:>+C+C+Ir( =#
<}}$
< #
& '3/O+P P../FGGG#
Hs! $B A9A9/B 3A>4 B composite_retrieval_nodec [ [ UR R UR R UR R
S9UR S9$ )N)r< textmetadata)nodescore)r r rx r< rv rw ry )rD rt s rF _result_nodes_to_node_with_score=LlamaCloudCompositeRetriever._result_nodes_to_node_with_score sQ +0033-2277166??
+00
rH query_bundlec Ub UO U R nUb UO U R nUb UO U R nU R ( aF U R R
R
U R R UUUUR S9nOZU R R
R U R R UUUUR U R R S9nUR Vs/ s H o`R U5 PM nn[ U R UR U R R" 5 n[% Xx- S SS9$ s snf )N)r* r+ r, queryr! r* r+ r, r~ r1 c U R $ rJ ry xs rF 8LlamaCloudCompositeRetriever._retrieve.. rH Tkeyreverse)r> r? r@ r8 r5 r: retriever9 r< query_strdirect_retriever7 r1 nodesrz r image_nodesr! sorted rD r| r* r+ r, resultrx
node_w_scoresimage_nodes_w_scoress rF _retrieve&LlamaCloudCompositeRetriever._retrieve sC 'tTZZ'3'?|TEWEW * 6MD." r rH Tr )r> r? r@ r8 r6 r: r r9 r< r r r7 r1 r rz r r r! r r s rF
_aretrieve'LlamaCloudCompositeRetriever._aretrieve s[ 'tTZZ'3'?|TEWEW * 6MD r8 r@ r? r r7 r r9 )NNN)%__name__
__module__ __qualname____firstlineno__r r strboolinthttpxClient AsyncClientr r r rB propertyr r
rK r rS r r ra ri rl ro rr r r rz r r r __static_attributes__
__classcell__)rE s @rF 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
'
X 26&*04 '
!'
-.'
sm '
- '
m
'
'
rH 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 rH rF r sL & & 8 > ; H H H ! H
= H
rH