a bg0&@sddlmZmZmZmZmZmZddlmZddl Z ddl Z ddl m Z ddlmZddlmZddlmZddlmZGd d d eZdS) )AnyIterableListOptionalTuplecast)uuid4NDocument) Embeddings) get_from_env) VectorStore)DistanceStrategyc @seZdZUdZdZeed<deZej dfee e eedddZ e ed d d Zed d d Zed ddZed ddZd,eeeeee eeedddZe e d ddZd-eeeeeedddZd.eee eedddZd/ee eeed d!d"Zd0ee eeeeefd d#d$Z d1eee eeed%d&d'Z!e"ddd(dej feee eeeee eeedd) d*d+Z#dS)2SemaDBa`SemaDB` vector store. This vector store is a wrapper around the SemaDB database. Example: .. code-block:: python from langchain_community.vectorstores import SemaDB db = SemaDB('mycollection', 768, embeddings, DistanceStrategy.COSINE) zsemadb.p.rapidapi.comHOSTzhttps://)collection_name vector_size embeddingdistance_strategyapi_keycCs,||_||_|ptdd|_||_||_dS)z#initialize the SemaDB vector store.rZSEMADB_API_KEYN)rrr r _embeddingr)selfrrrrrru/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/vectorstores/semadb.py__init__s zSemaDB.__init__)returncCsd|jtjdS)zReturn the common headers.zapplication/json)z content-typezX-RapidAPI-KeyzX-RapidAPI-Host)rrrrrrrheaders.szSemaDB.headerscCsp|jtjkrdS|jtjkr&tdnF|jtjkr6dS|jtjkrLtdn |jtjkr\dStd|jdS)z&Return the internal distance strategy.Z euclideanz,Max inner product is not supported by SemaDBdotZcosinezUnknown distance strategy N)rrEUCLIDEAN_DISTANCEZMAX_INNER_PRODUCT ValueErrorZ DOT_PRODUCTZJACCARDCOSINErrrr_get_internal_distance_strategy7s       z&SemaDB._get_internal_distance_strategycCs6|j|j|d}tjtjd||jd}|jdkS)z/Creates the corresponding collection in SemaDB.)idZ vectorSizeZdistanceMetricz /collectionsjsonr) rrr#requestspostrBASE_URLr status_code)rpayloadresponserrrcreate_collectionFszSemaDB.create_collectioncCs(tjtjd|j|jd}|jdkS)z/Deletes the corresponding collection in SemaDB. /collections/)rr')r(deleterr*rrr+)rr-rrrdelete_collectionTs zSemaDB.delete_collectionN)texts metadatas batch_sizekwargsrcKst|tst|}|j|}t|d|jkrNtdt|dd|j|jtj krt |}|t j j |ddd}tttt|}g}g}|durt|||D]<\} } } tt} || || | i| d| id qn>t||D]2\} } tt} || || | d| id qtdt||D]} || | |}tjtjd |jd d |i|jd }|jdkrtd|td|j| d}t|dkr:td|q:|S)zAdd texts to the vector store.rzEmbedding size mismatch z != T)ZaxisZkeepdimsNtext)r$vectormetadatar//pointspointsr%r'zHERE--zError adding points: Z failedRanges)! isinstancelistrZembed_documentslenrr!rrr"nparraylinalgnormrrfloattolistzipstrrappendranger(r)rr*rrr+printr8r&)rr3r4r5r6 embeddingsZ embed_matrixidsr<r8rr:Znew_idibatchr-Z failed_rangesrrr add_texts\s`           zSemaDB.add_textscCs|jS)zReturn the embeddings.)rrrrrrKszSemaDB.embeddings)rLr6rcKsHd|i}tjtjd|jd||jd}|jdkoFt|ddkS)a3Delete by vector ID or other criteria. Args: ids: List of ids to delete. **kwargs: Other keyword arguments that subclasses might use. Returns: Optional[bool]: True if deletion is successful, False otherwise, None if not implemented. rLr/r;r%r'Z failedPointsr) r(r0rr*rrr+r?r&)rrLr6r,r-rrrr0s z SemaDB.delete)rkrcCs|jtjkr8t|}|tj|}ttt | }||d}t j t jd|jd||jd}|jdkr~td|j|dS)zSearch points.)r9limitr/z/points/searchr%r'zError searching: r<)rrr"r@rArBrCrrrDrEr(r)rr*rrr+r!r8r&)rrrQZvecr,r-rrr_search_pointss   zSemaDB._search_points)queryrQr6rcKs|j|}|j||dS)z"Return docs most similar to query.rQ)r embed_querysimilarity_search_by_vector)rrTrQr6query_embeddingrrrsimilarity_searchs zSemaDB.similarity_searchcKs(|j|}|j||d}dd|DS)z$Run similarity search with distance.rUcSs,g|]$}t|dd|dd|dfqS)r:r8Z page_contentr:Zdistancer .0prrr sz7SemaDB.similarity_search_with_score..)rrVrS)rrTrQr6rXr<rrrsimilarity_search_with_scores  z#SemaDB.similarity_search_with_score)rrQr6rcKs|j||d}dd|DS)aReturn docs most similar to embedding vector. Args: embedding: Embedding to look up documents similar to. k: Number of Documents to return. Defaults to 4. Returns: List of Documents most similar to the query vector. rUcSs$g|]}t|dd|ddqS)r:r8rZr r[rrrr^sz6SemaDB.similarity_search_by_vector..)rS)rrrQr6r<rrrrWs z"SemaDB.similarity_search_by_vectorr) r3rr4rrrrr6rc KsX|s td|std|s$td||||||d} | sFtd| j||d| S)z9Return VectorStore initialized from texts and embeddings.z Collection name must be providedzVector size must be providedzAPI key must be provided)rrzError creating collection)r4)r!r.rO) clsr3rr4rrrrr6Zsemadbrrr from_textss" zSemaDB.from_texts)Nr2)N)rP)rP)rP)rP)$__name__ __module__ __qualname____doc__rrG__annotations__r*rr intr rpropertydictrr#boolr.r1rrrrrOrKr0rDrSr rYrr_rW classmethodrarrrrrs|      A      r)typingrrrrrruuidrnumpyr@r(Zlangchain_core.documentsr Zlangchain_core.embeddingsr Zlangchain_core.utilsr Zlangchain_core.vectorstoresr Z&langchain_community.vectorstores.utilsrrrrrrs