NgddlmZmZmZmZmZmZddlmZddl m Z deee efde fdZ Gdde ZGd d e Zd S) )AnyCallableIteratorListOptionalTuple)Document) BaseLoaderdocsreturnc@dd|DS)z#Default joiner for content columns. cg|] }|d S)).0docs j/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/document_loaders/rocksetdb.py z"default_joiner.. s---c!f---)join)r s rdefault_joinerrs# 99----- . ..rc,eZdZdZdedeffd ZxZS)ColumnNotFoundErrorzColumn not found error. missing_keyquerycVtd|d|dS)NzColumn "z" not selected in query: )super__init__)selfrr __class__s rrzColumnNotFoundError.__init__s2 RKRR5RRSSSSSr)__name__ __module__ __qualname____doc__strr __classcell__)r!s@rrr s]!!TCTTTTTTTTTTTrrceZdZdZdefdededeedeeede ee eefgeff dZ d e e fd ZdS) RocksetLoaderaLoad from a `Rockset` database. To use, you should have the `rockset` python package installed. Example: .. code-block:: python # This code will load 3 records from the "langchain_demo" # collection as Documents, with the `text` column used as # the content from langchain_community.document_loaders import RocksetLoader from rockset import RocksetClient, Regions, models loader = RocksetLoader( RocksetClient(Regions.usw2a1, ""), models.QueryRequestSql( query="select * from langchain_demo limit 3" ), ["text"] ) ) Nclientr content_keys metadata_keyscontent_columns_joinerc ddlm}m}ddlm}n#t $rt dwxYwt ||stdt|t ||stdt|||_ ||_ ||_ ||_ ||_ ||_||_ |j ddS#t"$rYdSwxYw) aKInitialize with Rockset client. Args: client: Rockset client object. query: Rockset query object. content_keys: The collection columns to be written into the `page_content` of the Documents. metadata_keys: The collection columns to be written into the `metadata` of the Documents. By default, this is all the keys in the document. content_columns_joiner: Method that joins content_keys and its values into a string. It's method that takes in a List[Tuple[str, Any]]], representing a list of tuples of (column name, column value). By default, this is a method that joins each column value with a new line. This method is only relevant if there are multiple content_keys. r)QueryPaginator RocksetClient)QueryRequestSqlz]Could not import rockset client python package. Please install it with `pip install rockset`.z;client should be an instance of rockset.RocksetClient, got zBquery should be an instance of rockset.model.QueryRequestSql, got langchainN)rocksetr/r0rockset.modelsr1 ImportError isinstance ValueErrortyper*rr+r-r, paginator request_modelset_applicationAttributeError) r r*rr+r,r-r/r0r1s rrzRocksetLoader.__init__-s].  = = = = = = = = 6 6 6 6 6 6 6   @   &-00 &F||&&  %11 %E{{%%    (&<#*',  K ' ' 4 4 4 4 4    DD s+>C C('C(r c#rK|jj|jj}|D] t |fd|jD|jfd|jDnVU#t$r%}t|j d|jd}~wwxYwdS)N)sqlc$g|] }||f Srrrcolrs rrz+RocksetLoader.lazy_load..os!FFFS#s3xFFFrc"i|] }|| Srrr@s r z+RocksetLoader.lazy_load..qsJJJc3s8JJJr) page_contentmetadatar) r*Queriesrresultsr r-r+r,KeyErrorrargs)r query_resultsers @r lazy_loadzRocksetLoader.lazy_loadgs +11 2   !  C !%!rQsAAAAAAAAAAAAAAAA------@@@@@@/eCHo./3//// TTTTT)TTTfffffJfffffr