h>dZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl Z ddl Z ddlmZddlmZddlmZddlmZdd lmZmZmZmZmZmZmZdd lm Z m!Z!dd l"m#Z#eZ$ d%d e e efd Z% d%de efdZ&e! d&de e'de e e'de ede ede ee'e'fde ee'efde ee'efde ee'efde e e(e jRfde e'defdZ*e! d&de e'de e e'de ede ede ee'e'fde ee'efde ee'efde ee'efde e e(e jRfde e'de eeeeefffdZ+e! d'de'de e'e e'fde ed e e,d!e ed"e e-de ee'efde ee'efde ee'efde e e(e jRfde e'defd#Z.e! d'de'de e'e e'fde ed e e,d!e ed"e e-de ee'efde ee'efde ee'efde e e(e jRfde e'de eeeeefffd$Z/y)(z@ LiteLLM SDK Functions for Creating and Searching Vector Stores N)partial)Any CoroutineDictListOptionalUnion)request_timeout)Logging)BaseLLMHTTPHandler)GenericLiteLLMParams)&VectorStoreCreateOptionalRequestParamsVectorStoreCreateResponseVectorStoreFileCountsVectorStoreResultContent&VectorStoreSearchOptionalRequestParamsVectorStoreSearchResponseVectorStoreSearchResult)ProviderConfigManagerclient)VectorStoreRequestUtils mock_resultscV|tdtddgg}tdd|S) z%Mock response for vector store searchgffffff?z5This is a sample search result from the vector store.text)rtype)scorecontentz vector_store.search_results.pagez sample query)object search_querydata)rrrrs V/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/vector_stores/main.py!mock_vector_store_search_responser#"sG #,T#    %1#   mock_responsecR|$tdddddtdddddddddd }|S) z%Mock response for vector store createN vs_mock123 vector_storeiEezMock Vector Storer) in_progress completedfailed cancelledtotalr*) idr created_atnamebytes file_countsstatus expires_after expires_atlast_active_atmetadata)rrr%s r"!mock_vector_store_create_responser9:sS1!!$- # ( r$r0file_idsr4chunking_strategyr7 extra_headers extra_query extra_bodytimeoutcustom_llm_providerreturnc Kt} tj} d| d<| d} ttf|||||||||| d | } t j }t|j| }| jd|d{}tj|r |d{}|S|}|S7)7 #t$r}tjd| || | d}~wwxYww)z' Async: Create a vector store. TacreateNopenai) r0r:r4r;r7r<r=r>r?r@modelr@original_exceptioncompletion_kwargs extra_kwargs) localsasyncioget_event_looprcreate contextvars copy_contextrunrun_in_executor iscoroutine Exceptionlitellmexception_type)r0r:r4r;r7r<r=r>r?r@kwargs local_varsloopfuncctxfunc_with_context init_responseresponsees r"rCrCVs(J( %%' y  &"*    '/'#! 3    &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA7B3B/B3$B1%B3*C+B3.C/B31B33 C<CCCc  t} | jd} | jdd} | jdddu}tdi| }|jr9t |jt rttdi|jS| d} tjtj|  }|td | | j| tj | }| j#dd |i|d| i| t$j'||| || |||xst(|| jd  }|S#t*$r}tj,d| || | d}~wwxYw)a Create a vector store. Args: name: The name of the vector store. file_ids: A list of File IDs that the vector store should use. expires_after: The expiration policy for the vector store. chunking_strategy: The chunking strategy used to chunk the file(s). metadata: Set of 16 key-value pairs that can be attached to an object. Returns: VectorStoreCreateResponse containing the created vector store details. litellm_logging_objlitellm_call_idNrCFTr8rDproviderz)Vector store create is not supported for r0rFoptional_paramslitellm_paramsr@r) #vector_store_create_optional_paramsvector_store_provider_configr@rf logging_objr<r>r? _is_asyncrrE)rJgetpopr r% isinstancedictr9rr!get_provider_vector_stores_configrT LlmProviders ValueErrorupdater0get_requested_vector_store_create_optional_paramupdate_environment_variablesbase_llm_http_handlervector_store_create_handlerr rSrU)r0r:r4r;r7r<r=r>r?r@rVrWr`rarjrfrhrgr]r^s r"rMrMs<JI 17( $ ( /;r?r@rE) rJrKrLrsearchrNrOrPrQrRrSrTrU)rxryrzr{r|r}r<r=r>r?r@rVrWrXrYrZr[r\r]r^s r"rrs*J) %%' y  &"*   +++''#! 3   &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY CA8B4B0B4%B2&B4+C,B4/C0B42B44 C=CCCc  &t} | jd} | jdd}| jdddu}tdi| }|jrTt |jt tfr4d}t |jtr |j}t|S| d} tjtj|  }|td | | j| tj | }| j#d||d |||d | t$j'||||| || ||| xst(|| jd }|S#t*$r}tj,d| || | d}~wwxYw)ag Search a vector store for relevant chunks based on a query and file attributes filter. Args: vector_store_id: The ID of the vector store to search. query: A query string or array for the search. filters: Optional filter to apply based on file attributes. max_num_results: Maximum number of results to return (1-50, default 10). ranking_options: Optional ranking options for search. rewrite_query: Whether to rewrite the natural language query for vector search. Returns: VectorStoreSearchResponse containing the search results. r`raNrFTr!rDrbz)Vector store search is not supported for )rxry)rarxrdr) rxry#vector_store_search_optional_paramsrhr@rfrir<r>r?rjrrErk)rJrlrmr r%rnstrlistr#rrprTrqrrrsr0get_requested_vector_store_search_optional_paramrurvvector_store_search_handlerr rSrU)rxryrzr{r|r}r<r=r>r?r@rVrWr`rarjrfrrhrr]r^s r"rrCs@JN 17( $ ( /;rs>> -SI58?+, =A4 7892:>568$($((,)-/3,0+/59)-< 3-< tCy!< D><  ~ < tCH~& < DcN+< $sCx.)< c3h(< eE5==01 2< "#< < < ~$($((,)-/3,0+/59)-g 3-g tCy!g D>g  ~ g tCH~& g DcN+g $sCx.)g c3h(g eE5==01 2g "#g  $iS:S0S&T TUg g T#%)&*$(/3,0+/59)-> > d3i > d^> c] > d^ > D> > DcN+> $sCx.)> c3h(> eE5==01 2> "#> !> > B#%)&*$(/3,0+/59)-n n d3i n d^n c] n d^ n D> n DcN+n $sCx.)n c3h(n eE5==01 2n "#n  $iS:S0S&T TU!n n r$