SYMBOL INDEX (424 symbols across 49 files) FILE: datastore/datastore.py class DataStore (line 17) | class DataStore(ABC): method upsert (line 18) | async def upsert( method _upsert (line 45) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method query (line 53) | async def query(self, queries: List[Query]) -> List[QueryResult]: method _query (line 68) | async def _query(self, queries: List[QueryWithEmbedding]) -> List[Quer... method delete (line 75) | async def delete( FILE: datastore/factory.py function get_datastore (line 5) | async def get_datastore() -> DataStore: FILE: datastore/providers/analyticdb_datastore.py class AnalyticDBDataStore (line 36) | class AnalyticDBDataStore(DataStore): method __init__ (line 37) | def __init__(self, config: Dict[str, str] = PG_CONFIG): method _initialize_db (line 57) | def _initialize_db(self): method _create_table (line 67) | def _create_table(self, cur: psycopg2.extensions.cursor): method _create_embedding_index (line 84) | def _create_embedding_index(self, cur: psycopg2.extensions.cursor): method _upsert (line 110) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _upsert_chunk (line 125) | def _upsert_chunk(self, chunk: DocumentChunk): method _query (line 169) | async def _query(self, queries: List[QueryWithEmbedding]) -> List[Quer... method delete (line 262) | async def delete( method _generate_delete_query (line 296) | def _generate_delete_query( FILE: datastore/providers/azurecosmosdb_datastore.py class AzureCosmosDBStoreApi (line 40) | class AzureCosmosDBStoreApi(ABC): method ensure (line 42) | async def ensure(self, num_lists, similarity): method upsert_core (line 46) | async def upsert_core(self, docId: str, chunks: List[DocumentChunk]) -... method query_core (line 50) | async def query_core( method drop_container (line 56) | async def drop_container(self): method delete_filter (line 60) | async def delete_filter(self, filter: DocumentMetadataFilter): method delete_ids (line 64) | async def delete_ids(self, ids: List[str]): method delete_document_ids (line 68) | async def delete_document_ids(self, documentIds: List[str]): class MongoStoreApi (line 72) | class MongoStoreApi(AzureCosmosDBStoreApi): method __init__ (line 73) | def __init__(self, mongoClient: MongoClient): method _get_metadata_filter (line 77) | def _get_metadata_filter(filter: DocumentMetadataFilter) -> dict: method ensure (line 97) | async def ensure(self, num_lists, similarity): method upsert_core (line 122) | async def upsert_core(self, docId: str, chunks: List[DocumentChunk]) -... method query_core (line 142) | async def query_core( method drop_container (line 182) | async def drop_container(self): method delete_filter (line 185) | async def delete_filter(self, filter: DocumentMetadataFilter): method delete_ids (line 189) | async def delete_ids(self, ids: List[str]): method delete_document_ids (line 192) | async def delete_document_ids(self, documentIds: List[str]): class AzureCosmosDBDataStore (line 202) | class AzureCosmosDBDataStore(DataStore): method __init__ (line 203) | def __init__(self, cosmosStore: AzureCosmosDBStoreApi): method create (line 222) | async def create(num_lists, similarity) -> DataStore: method _upsert (line 236) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 249) | async def _query( method delete (line 270) | async def delete( FILE: datastore/providers/azuresearch_datastore.py class AzureSearchDataStore (line 55) | class AzureSearchDataStore(DataStore): method __init__ (line 56) | def __init__(self): method _upsert (line 76) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method delete (line 116) | async def delete( method _query (line 164) | async def _query(self, queries: List[QueryWithEmbedding]) -> List[Quer... method _single_query (line 170) | async def _single_query(self, query: QueryWithEmbedding) -> QueryResult: method _translate_filter (line 225) | def _translate_filter(filter: DocumentMetadataFilter) -> str: method _create_index (line 262) | def _create_index(self, mgmt_client: SearchIndexClient): method _create_credentials (line 358) | def _create_credentials( FILE: datastore/providers/chroma_datastore.py class ChromaDataStore (line 36) | class ChromaDataStore(DataStore): method __init__ (line 37) | def __init__( method upsert (line 73) | async def upsert( method _upsert (line 86) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _where_from_query_filter (line 110) | def _where_from_query_filter(self, query_filter: DocumentMetadataFilte... method _process_metadata_for_storage (line 146) | def _process_metadata_for_storage(self, metadata: DocumentChunkMetadat... method _process_metadata_from_storage (line 165) | def _process_metadata_from_storage(self, metadata: Dict) -> DocumentCh... method _query (line 177) | async def _query(self, queries: List[QueryWithEmbedding]) -> List[Quer... method delete (line 220) | async def delete( FILE: datastore/providers/elasticsearch_datastore.py class ElasticsearchDataStore (line 33) | class ElasticsearchDataStore(DataStore): method __init__ (line 34) | def __init__( method _upsert (line 77) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 93) | async def _query( method delete (line 113) | async def delete( method _get_es_filters (line 164) | def _get_es_filters( method _convert_document_chunk_to_es_document_operation (line 196) | def _convert_document_chunk_to_es_document_operation( method _convert_queries_to_msearch_query (line 222) | def _convert_queries_to_msearch_query(self, queries: List[QueryWithEmb... method _convert_hit_to_document_chunk_with_score (line 242) | def _convert_hit_to_document_chunk_with_score(self, hit) -> DocumentCh... method _set_up_index (line 251) | def _set_up_index( method _recreate_index (line 285) | def _recreate_index( function connect_to_elasticsearch (line 314) | def connect_to_elasticsearch( FILE: datastore/providers/llama_datastore.py function _create_or_load_index (line 42) | def _create_or_load_index( function _create_or_load_query_kwargs (line 68) | def _create_or_load_query_kwargs( function _doc_chunk_to_node (line 80) | def _doc_chunk_to_node(doc_chunk: DocumentChunk, source_doc_id: str) -> ... function _query_with_embedding_to_query_bundle (line 91) | def _query_with_embedding_to_query_bundle(query: QueryWithEmbedding) -> ... function _source_node_to_doc_chunk_with_score (line 98) | def _source_node_to_doc_chunk_with_score( function _response_to_query_result (line 115) | def _response_to_query_result( class LlamaDataStore (line 127) | class LlamaDataStore(DataStore): method __init__ (line 128) | def __init__( method _upsert (line 134) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 152) | async def _query( method delete (line 185) | async def delete( FILE: datastore/providers/milvus_datastore.py class Required (line 47) | class Required: class MilvusDataStore (line 106) | class MilvusDataStore(DataStore): method __init__ (line 107) | def __init__( method _get_schema (line 129) | def _get_schema(self): method _create_connection (line 132) | def _create_connection(self): method _create_collection (line 174) | def _create_collection(self, collection_name, create_new: bool) -> None: method _create_index (line 222) | def _create_index(self): method _upsert (line 311) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _get_values (line 368) | def _get_values(self, chunk: DocumentChunk) -> List[any] | None: # ty... method _query (line 405) | async def _query( method delete (line 479) | async def delete( method _get_filter (line 574) | def _get_filter(self, filter: DocumentMetadataFilter) -> Optional[str]: FILE: datastore/providers/mongodb_atlas_datastore.py class MongoDBAtlasDataStore (line 31) | class MongoDBAtlasDataStore(DataStore): method __init__ (line 33) | def __init__( method client (line 74) | def client(self): method upsert (line 79) | async def upsert( method _upsert (line 89) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 108) | async def _query( method _execute_embedding_query (line 123) | async def _execute_embedding_query(self, query: QueryWithEmbedding) ->... method delete (line 159) | async def delete( method _convert_mongodb_document_to_document_chunk_with_score (line 200) | def _convert_mongodb_document_to_document_chunk_with_score( method _build_mongo_filter (line 211) | def _build_mongo_filter( method _connect_to_mongodb_atlas (line 244) | def _connect_to_mongodb_atlas(atlas_connection_uri: str): FILE: datastore/providers/pgvector_datastore.py class PGClient (line 19) | class PGClient(ABC): method upsert (line 21) | async def upsert(self, table: str, json: dict[str, Any]) -> None: method rpc (line 28) | async def rpc(self, function_name: str, params: dict[str, Any]) -> Any: method delete_like (line 35) | async def delete_like(self, table: str, column: str, pattern: str) -> ... method delete_in (line 42) | async def delete_in(self, table: str, column: str, ids: List[str]) -> ... method delete_by_filters (line 49) | async def delete_by_filters( class PgVectorDataStore (line 59) | class PgVectorDataStore(DataStore): method __init__ (line 60) | def __init__(self): method create_db_client (line 64) | def create_db_client(self) -> PGClient: method _upsert (line 73) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 100) | async def _query(self, queries: List[QueryWithEmbedding]) -> List[Quer... method delete (line 155) | async def delete( FILE: datastore/providers/pinecone_datastore.py class PineconeDataStore (line 37) | class PineconeDataStore(DataStore): method __init__ (line 38) | def __init__(self): method _upsert (line 70) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 112) | async def _query( method delete (line 179) | async def delete( method _get_pinecone_filter (line 224) | def _get_pinecone_filter( method _get_pinecone_metadata (line 252) | def _get_pinecone_metadata( FILE: datastore/providers/postgres_datastore.py class PostgresDataStore (line 24) | class PostgresDataStore(PgVectorDataStore): method create_db_client (line 25) | def create_db_client(self): class PostgresClient (line 29) | class PostgresClient(PGClient): method __init__ (line 30) | def __init__(self) -> None: method __del__ (line 37) | def __del__(self): method upsert (line 41) | async def upsert(self, table: str, json: dict[str, Any]): method rpc (line 73) | async def rpc(self, function_name: str, params: dict[str, Any]): method delete_like (line 88) | async def delete_like(self, table: str, column: str, pattern: str): method delete_in (line 99) | async def delete_in(self, table: str, column: str, ids: List[str]): method delete_by_filters (line 110) | async def delete_by_filters(self, table: str, filter: DocumentMetadata... FILE: datastore/providers/qdrant_datastore.py class QdrantDataStore (line 32) | class QdrantDataStore(DataStore): method __init__ (line 35) | def __init__( method _upsert (line 63) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 80) | async def _query( method delete (line 105) | async def delete( method _convert_document_chunk_to_point (line 133) | def _convert_document_chunk_to_point( method _create_document_chunk_id (line 152) | def _create_document_chunk_id(self, external_id: Optional[str]) -> str: method _convert_query_to_search_request (line 157) | def _convert_query_to_search_request( method _convert_metadata_filter_to_qdrant_filter (line 168) | def _convert_metadata_filter_to_qdrant_filter( method _convert_scored_point_to_document_chunk_with_score (line 233) | def _convert_scored_point_to_document_chunk_with_score( method _set_up_collection (line 245) | def _set_up_collection( method _recreate_collection (line 276) | def _recreate_collection(self, distance: rest.Distance, vector_size: i... FILE: datastore/providers/redis_datastore.py function unpack_schema (line 53) | def unpack_schema(d: dict): function _check_redis_module_exist (line 61) | async def _check_redis_module_exist(client: redis.Redis, modules: List[d... class RedisDataStore (line 76) | class RedisDataStore(DataStore): method __init__ (line 77) | def __init__(self, client: redis.Redis, redisearch_schema: dict): method init (line 89) | async def init(cls, **kwargs): method _redis_key (line 147) | def _redis_key(document_id: str, chunk_id: str) -> str: method _escape (line 161) | def _escape(value: str) -> str: method _get_redis_chunk (line 178) | def _get_redis_chunk(self, chunk: DocumentChunk) -> dict: method _get_redis_query (line 206) | def _get_redis_query(self, query: QueryWithEmbedding) -> RediSearchQuery: method _redis_delete (line 266) | async def _redis_delete(self, keys: List[str]): method _upsert (line 278) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 301) | async def _query( method _find_keys (line 345) | async def _find_keys(self, pattern: str) -> List[str]: method delete (line 348) | async def delete( FILE: datastore/providers/supabase_datastore.py class SupabaseDataStore (line 23) | class SupabaseDataStore(PgVectorDataStore): method create_db_client (line 24) | def create_db_client(self): class SupabaseClient (line 28) | class SupabaseClient(PGClient): method __init__ (line 29) | def __init__(self) -> None: method upsert (line 36) | async def upsert(self, table: str, json: dict[str, Any]): method rpc (line 45) | async def rpc(self, function_name: str, params: dict[str, Any]): method delete_like (line 57) | async def delete_like(self, table: str, column: str, pattern: str): method delete_in (line 63) | async def delete_in(self, table: str, column: str, ids: List[str]): method delete_by_filters (line 69) | async def delete_by_filters(self, table: str, filter: DocumentMetadata... FILE: datastore/providers/weaviate_datastore.py function extract_schema_properties (line 79) | def extract_schema_properties(schema): class WeaviateDataStore (line 85) | class WeaviateDataStore(DataStore): method handle_errors (line 86) | def handle_errors(self, results: Optional[List[dict]]) -> List[str]: method __init__ (line 104) | def __init__(self): method _build_auth_credentials (line 137) | def _build_auth_credentials(): method _upsert (line 149) | async def _upsert(self, chunks: Dict[str, List[DocumentChunk]]) -> Lis... method _query (line 188) | async def _query( method delete (line 264) | async def delete( method build_filters (line 316) | def build_filters(filter): method _is_valid_weaviate_id (line 354) | def _is_valid_weaviate_id(candidate_id: str) -> bool: method _is_wcs_domain (line 374) | def _is_wcs_domain(url: str) -> bool: FILE: datastore/providers/zilliz_datastore.py class ZillizDataStore (line 24) | class ZillizDataStore(MilvusDataStore): method __init__ (line 25) | def __init__(self, create_new: Optional[bool] = False): method _create_connection (line 40) | def _create_connection(self): method _create_index (line 54) | def _create_index(self): FILE: examples/authentication-methods/no-auth/main.py function upsert_file (line 40) | async def upsert_file( function upsert (line 67) | async def upsert( function query_main (line 82) | async def query_main( function query (line 100) | async def query( function delete (line 117) | async def delete( function startup (line 138) | async def startup(): function start (line 143) | def start(): FILE: examples/memory/main.py function validate_token (line 32) | def validate_token(credentials: HTTPAuthorizationCredentials = Depends(b... function upsert_file (line 56) | async def upsert_file( function upsert_main (line 83) | async def upsert_main( function upsert (line 101) | async def upsert( function query_main (line 117) | async def query_main( function query (line 137) | async def query( function delete (line 155) | async def delete( function startup (line 177) | async def startup(): function start (line 182) | def start(): FILE: examples/providers/supabase/migrations/20230414142107_init_pg_vector.sql type documents (line 3) | create table if not exists documents ( type ix_documents_document_id (line 15) | create index ix_documents_document_id on documents using btree ( documen... type ix_documents_source (line 16) | create index ix_documents_source on documents using btree ( source ) type ix_documents_source_id (line 17) | create index ix_documents_source_id on documents using btree ( source_id ) type ix_documents_author (line 18) | create index ix_documents_author on documents using btree ( author ) type ix_documents_created_at (line 19) | create index ix_documents_created_at on documents using brin ( created_at ) function match_page_sections (line 23) | create or replace function match_page_sections(in_embedding vector(256) ... FILE: local_server/main.py function get_manifest (line 44) | async def get_manifest(request): function get_logo (line 52) | async def get_logo(request): function get_openapi (line 58) | async def get_openapi(request): function upsert_file (line 67) | async def upsert_file( function upsert (line 94) | async def upsert( function query_main (line 106) | async def query_main(request: QueryRequest = Body(...)): function delete (line 121) | async def delete( function startup (line 142) | async def startup(): function start (line 147) | def start(): FILE: models/api.py class UpsertRequest (line 11) | class UpsertRequest(BaseModel): class UpsertResponse (line 15) | class UpsertResponse(BaseModel): class QueryRequest (line 19) | class QueryRequest(BaseModel): class QueryResponse (line 23) | class QueryResponse(BaseModel): class DeleteRequest (line 27) | class DeleteRequest(BaseModel): class DeleteResponse (line 33) | class DeleteResponse(BaseModel): FILE: models/models.py class Source (line 6) | class Source(str, Enum): class DocumentMetadata (line 12) | class DocumentMetadata(BaseModel): class DocumentChunkMetadata (line 20) | class DocumentChunkMetadata(DocumentMetadata): class DocumentChunk (line 24) | class DocumentChunk(BaseModel): class DocumentChunkWithScore (line 31) | class DocumentChunkWithScore(DocumentChunk): class Document (line 35) | class Document(BaseModel): class DocumentWithChunks (line 41) | class DocumentWithChunks(Document): class DocumentMetadataFilter (line 45) | class DocumentMetadataFilter(BaseModel): class Query (line 54) | class Query(BaseModel): class QueryWithEmbedding (line 60) | class QueryWithEmbedding(Query): class QueryResult (line 64) | class QueryResult(BaseModel): FILE: scripts/process_json/process_json.py function process_json_dump (line 16) | async def process_json_dump( function main (line 109) | async def main(): FILE: scripts/process_jsonl/process_jsonl.py function process_jsonl_dump (line 16) | async def process_jsonl_dump( function main (line 107) | async def main(): FILE: scripts/process_zip/process_zip.py function process_file_dump (line 19) | async def process_file_dump( function main (line 114) | async def main(): FILE: server/main.py function validate_token (line 27) | def validate_token(credentials: HTTPAuthorizationCredentials = Depends(b... function upsert_file (line 51) | async def upsert_file( function upsert (line 78) | async def upsert( function query_main (line 93) | async def query_main( function query (line 112) | async def query( function delete (line 129) | async def delete( function startup (line 150) | async def startup(): function start (line 155) | def start(): FILE: services/chunks.py function get_text_chunks (line 25) | def get_text_chunks(text: str, chunk_token_size: Optional[int]) -> List[... function create_document_chunks (line 102) | def create_document_chunks( function get_document_chunks (line 152) | def get_document_chunks( FILE: services/date.py function to_unix_timestamp (line 5) | def to_unix_timestamp(date_str: str) -> int: FILE: services/extract_metadata.py function extract_metadata_from_document (line 9) | def extract_metadata_from_document(text: str) -> Dict[str, str]: FILE: services/file.py function get_document_from_file (line 15) | async def get_document_from_file( function extract_text_from_filepath (line 25) | def extract_text_from_filepath(filepath: str, mimetype: Optional[str] = ... function extract_text_from_file (line 48) | def extract_text_from_file(file: BufferedReader, mimetype: str) -> str: function extract_text_from_form_file (line 91) | async def extract_text_from_form_file(file: UploadFile): FILE: services/openai.py function get_embeddings (line 13) | def get_embeddings(texts: List[str]) -> List[List[float]]: function get_chat_completion (line 44) | def get_chat_completion( FILE: services/pii_detection.py function screen_text_for_pii (line 5) | def screen_text_for_pii(text: str) -> bool: FILE: tests/datastore/providers/analyticdb/test_analyticdb_datastore.py function analyticdb_datastore (line 16) | def analyticdb_datastore(): function document_chunk_one (line 21) | def document_chunk_one(): function document_chunk_two (line 63) | def document_chunk_two(): function test_upsert (line 142) | async def test_upsert(analyticdb_datastore, document_chunk_one): function test_reload (line 156) | async def test_reload(analyticdb_datastore, document_chunk_one, document... function test_upsert_query_all (line 180) | async def test_upsert_query_all(analyticdb_datastore, document_chunk_two): function test_query_accuracy (line 197) | async def test_query_accuracy(analyticdb_datastore, document_chunk_one): function test_query_filter (line 215) | async def test_query_filter(analyticdb_datastore, document_chunk_one): function test_delete_with_date_filter (line 236) | async def test_delete_with_date_filter(analyticdb_datastore, document_ch... function test_delete_with_source_filter (line 259) | async def test_delete_with_source_filter(analyticdb_datastore, document_... function test_delete_with_document_id_filter (line 282) | async def test_delete_with_document_id_filter(analyticdb_datastore, docu... function test_delete_with_document_id (line 303) | async def test_delete_with_document_id(analyticdb_datastore, document_ch... FILE: tests/datastore/providers/azurecosmosdb/test_azurecosmosdb_datastore.py function create_embedding (line 20) | def create_embedding(non_zero_pos: int) -> List[float]: function azure_cosmos_db_settings_from_dot_env (line 28) | def azure_cosmos_db_settings_from_dot_env() -> dict: function initial_document_chunks (line 50) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function queries (line 66) | def queries() -> List[QueryWithEmbedding]: function azurecosmosdb_datastore (line 83) | async def azurecosmosdb_datastore() -> DataStore: function test_upsert (line 90) | async def test_upsert( function test_query (line 104) | async def test_query( function test_delete (line 130) | async def test_delete(azurecosmosdb_datastore: AzureCosmosDBDataStore) -... function test_delete_all (line 168) | async def test_delete_all(azurecosmosdb_datastore: AzureCosmosDBDataStor... FILE: tests/datastore/providers/azuresearch/test_azuresearch_datastore.py function azuresearch_mgmt_client (line 26) | def azuresearch_mgmt_client(): function test_translate_filter (line 34) | def test_translate_filter(): function test_lifecycle_hybrid (line 88) | async def test_lifecycle_hybrid(azuresearch_mgmt_client: SearchIndexClie... function test_lifecycle_vectors_only (line 95) | async def test_lifecycle_vectors_only(azuresearch_mgmt_client: SearchInd... function test_lifecycle_semantic (line 102) | async def test_lifecycle_semantic(azuresearch_mgmt_client: SearchIndexCl... function lifecycle (line 110) | async def lifecycle(azuresearch_mgmt_client: SearchIndexClient): FILE: tests/datastore/providers/chroma/test_chroma_datastore.py function ephemeral_chroma_datastore (line 19) | def ephemeral_chroma_datastore() -> ChromaDataStore: function persisted_chroma_datastore (line 26) | def persisted_chroma_datastore() -> ChromaDataStore: function get_chroma_datastore (line 35) | def get_chroma_datastore() -> ChromaDataStore: function cleanup (line 42) | def cleanup(): function create_embedding (line 51) | def create_embedding(dim: int) -> List[float]: function initial_document_chunks (line 61) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function document_chunks (line 77) | def document_chunks(initial_document_chunks) -> Dict[str, List[DocumentC... function test_add_chunks (line 103) | async def test_add_chunks(document_chunks: Dict[str, List[DocumentChunk]]): function test_upsert (line 117) | async def test_upsert( function test_add_and_query_all (line 138) | async def test_add_and_query_all(document_chunks): function test_query_accuracy (line 155) | async def test_query_accuracy(document_chunks): function test_query_filter_by_id (line 188) | async def test_query_filter_by_id(document_chunks): function test_query_filter_by_date (line 213) | async def test_query_filter_by_date(document_chunks): function test_delete_by_id (line 267) | async def test_delete_by_id(document_chunks): FILE: tests/datastore/providers/elasticsearch/test_elasticsearch_datastore.py function elasticsearch_datastore (line 19) | def elasticsearch_datastore(): function sample_embedding (line 23) | def sample_embedding(one_element_poz: int): function sample_embeddings (line 29) | def sample_embeddings(num: int, one_element_start: int = 0): function document_chunk_one (line 39) | def document_chunk_one(): function test_upsert (line 77) | async def test_upsert(elasticsearch_datastore, document_chunk_one): function test_upsert_query_all (line 92) | async def test_upsert_query_all(elasticsearch_datastore, document_chunk_... function test_delete_with_document_id (line 109) | async def test_delete_with_document_id(elasticsearch_datastore, document... function test_delete_with_source_filter (line 132) | async def test_delete_with_source_filter(elasticsearch_datastore, docume... FILE: tests/datastore/providers/llama/test_llama_datastore.py function create_embedding (line 7) | def create_embedding(non_zero_pos: int, size: int) -> List[float]: function initial_document_chunks (line 14) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function queries (line 30) | def queries() -> List[QueryWithEmbedding]: function llama_datastore (line 47) | def llama_datastore() -> LlamaDataStore: function test_upsert (line 52) | async def test_upsert( function test_query (line 62) | async def test_query( function test_delete (line 87) | async def test_delete( FILE: tests/datastore/providers/milvus/test_milvus_datastore.py function milvus_datastore (line 21) | def milvus_datastore(): function sample_embedding (line 25) | def sample_embedding(one_element_poz: int): function sample_embeddings (line 31) | def sample_embeddings(num: int, one_element_start: int = 0): function document_chunk_one (line 42) | def document_chunk_one(): function document_chunk_two (line 85) | def document_chunk_two(): function test_upsert (line 164) | async def test_upsert(milvus_datastore, document_chunk_one): function test_reload (line 174) | async def test_reload(milvus_datastore, document_chunk_one, document_chu... function test_upsert_query_all (line 198) | async def test_upsert_query_all(milvus_datastore, document_chunk_two): function test_query_accuracy (line 218) | async def test_query_accuracy(milvus_datastore, document_chunk_one): function test_query_filter (line 238) | async def test_query_filter(milvus_datastore, document_chunk_one): function test_delete_with_date_filter (line 261) | async def test_delete_with_date_filter(milvus_datastore, document_chunk_... function test_delete_with_source_filter (line 286) | async def test_delete_with_source_filter(milvus_datastore, document_chun... function test_delete_with_document_id_filter (line 311) | async def test_delete_with_document_id_filter(milvus_datastore, document... function test_delete_with_document_id (line 334) | async def test_delete_with_document_id(milvus_datastore, document_chunk_... FILE: tests/datastore/providers/mongodb_atlas/test_integration.py function documents (line 22) | def documents(): function client (line 35) | def client(): function delete (line 44) | def delete(client) -> bool: function upsert (line 52) | def upsert(delete, documents, client) -> bool: function test_delete (line 59) | def test_delete(delete) -> None: function test_upsert (line 65) | def test_upsert(upsert) -> None: function test_query (line 71) | def test_query(upsert, client) -> None: # upsert, function test_required_vars (line 103) | def test_required_vars() -> None: function test_mongodb_connection (line 112) | def test_mongodb_connection() -> None: function test_openai_connection (line 118) | def test_openai_connection() -> None: FILE: tests/datastore/providers/mongodb_atlas/test_mongodb_datastore.py function assert_when_ready (line 41) | async def assert_when_ready(callable: Callable, tries: int = 5, interval... function collection_size_callback_factory (line 57) | def collection_size_callback_factory(collection, num: int): function _mongodb_datastore (line 67) | def _mongodb_datastore(): function mongodb_datastore (line 72) | async def mongodb_datastore(_mongodb_datastore): function sample_embedding (line 81) | def sample_embedding(one_element_poz: int): function sample_embeddings (line 88) | def sample_embeddings(num: int, one_element_start: int = 0): function document_id (line 93) | def document_id(): function chunk_ids (line 98) | def chunk_ids(document_id): function one_documents_chunks (line 104) | def one_documents_chunks(document_id, chunk_ids): function test_upsert (line 145) | async def test_upsert(mongodb_datastore: MongoDBAtlasDataStore, one_docu... function test_upsert_query_all (line 154) | async def test_upsert_query_all(mongodb_datastore, one_documents_chunks,... function test_delete_with_document_id (line 172) | async def test_delete_with_document_id(mongodb_datastore, one_documents_... function test_delete_with_source_filter (line 186) | async def test_delete_with_source_filter(mongodb_datastore, one_document... function build_mongo_filter (line 213) | def build_mongo_filter(): function test_build_mongo_filter_with_no_filter (line 217) | async def test_build_mongo_filter_with_no_filter(build_mongo_filter): function test_build_mongo_filter_with_start_date (line 222) | async def test_build_mongo_filter_with_start_date(build_mongo_filter): function test_build_mongo_filter_with_end_date (line 234) | async def test_build_mongo_filter_with_end_date(build_mongo_filter): function test_build_mongo_filter_with_metadata_field (line 246) | async def test_build_mongo_filter_with_metadata_field(build_mongo_filter): FILE: tests/datastore/providers/postgres/test_postgres_datastore.py function create_embedding (line 15) | def create_embedding(non_zero_pos: int) -> List[float]: function initial_document_chunks (line 23) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function queries (line 39) | def queries() -> List[QueryWithEmbedding]: function postgres_datastore (line 56) | def postgres_datastore() -> PostgresDataStore: function test_upsert (line 61) | async def test_upsert( function test_query (line 71) | async def test_query( function test_delete (line 96) | async def test_delete( function test_upsert_new_chunk (line 108) | async def test_upsert_new_chunk(postgres_datastore): function test_upsert_existing_chunk (line 121) | async def test_upsert_existing_chunk(postgres_datastore): function test_query_score (line 154) | async def test_query_score(postgres_datastore): function test_query_filter (line 182) | async def test_query_filter(postgres_datastore): function test_delete (line 234) | async def test_delete(postgres_datastore): function test_delete_all (line 269) | async def test_delete_all(postgres_datastore): FILE: tests/datastore/providers/qdrant/test_qdrant_datastore.py function create_embedding (line 17) | def create_embedding(non_zero_pos: int, size: int) -> List[float]: function qdrant_datastore (line 24) | def qdrant_datastore() -> QdrantDataStore: function client (line 31) | def client() -> qdrant_client.QdrantClient: function initial_document_chunks (line 36) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function document_chunks (line 52) | def document_chunks() -> Dict[str, List[DocumentChunk]]: function test_datastore_creates_payload_indexes (line 82) | async def test_datastore_creates_payload_indexes( function test_upsert_creates_all_points (line 98) | async def test_upsert_creates_all_points( function test_upsert_does_not_remove_existing_documents_but_store_new (line 110) | async def test_upsert_does_not_remove_existing_documents_but_store_new( function test_query_returns_all_on_single_query (line 129) | async def test_query_returns_all_on_single_query(qdrant_datastore, docum... function test_query_returns_closest_entry (line 146) | async def test_query_returns_closest_entry(qdrant_datastore, document_ch... function test_query_filter_by_document_id_returns_this_document_chunks (line 168) | async def test_query_filter_by_document_id_returns_this_document_chunks( function test_query_start_date_converts_datestring (line 197) | async def test_query_start_date_converts_datestring( function test_query_end_date_converts_datestring (line 219) | async def test_query_end_date_converts_datestring( function test_delete_removes_by_ids (line 240) | async def test_delete_removes_by_ids( function test_delete_removes_by_document_id_filter (line 254) | async def test_delete_removes_by_document_id_filter( function test_delete_removes_all (line 270) | async def test_delete_removes_all( FILE: tests/datastore/providers/redis/test_redis_datastore.py function redis_datastore (line 17) | async def redis_datastore(): function create_embedding (line 21) | def create_embedding(i, dim): function create_document_chunk (line 27) | def create_document_chunk(i, dim): function create_document_chunks (line 38) | def create_document_chunks(n, dim): function test_redis_upsert_query (line 44) | async def test_redis_upsert_query(redis_datastore): function test_redis_filter_query (line 60) | async def test_redis_filter_query(redis_datastore): function test_redis_delete_docs (line 74) | async def test_redis_delete_docs(redis_datastore): FILE: tests/datastore/providers/supabase/test_supabase_datastore.py function create_embedding (line 15) | def create_embedding(non_zero_pos: int) -> List[float]: function initial_document_chunks (line 23) | def initial_document_chunks() -> Dict[str, List[DocumentChunk]]: function queries (line 39) | def queries() -> List[QueryWithEmbedding]: function supabase_datastore (line 56) | def supabase_datastore() -> SupabaseDataStore: function test_upsert (line 61) | async def test_upsert( function test_query (line 71) | async def test_query( function test_delete (line 96) | async def test_delete( function test_upsert_new_chunk (line 108) | async def test_upsert_new_chunk(supabase_datastore): function test_upsert_existing_chunk (line 121) | async def test_upsert_existing_chunk(supabase_datastore): function test_query_score (line 154) | async def test_query_score(supabase_datastore): function test_query_filter (line 182) | async def test_query_filter(supabase_datastore): function test_delete (line 234) | async def test_delete(supabase_datastore): function test_delete_all (line 269) | async def test_delete_all(supabase_datastore): FILE: tests/datastore/providers/weaviate/test_weaviate_datastore.py function weaviate_client (line 26) | def weaviate_client(): function test_db (line 37) | def test_db(weaviate_client, documents): function documents (line 52) | def documents(): function caplog (line 105) | def caplog(caplog: LogCaptureFixture): function test_upsert (line 114) | def test_upsert(weaviate_client, document_id): function test_upsert_no_metadata (line 197) | def test_upsert_no_metadata(weaviate_client): function test_upsert_invalid_documents (line 234) | def test_upsert_invalid_documents(weaviate_client, test_document, expect... function test_query (line 276) | def test_query(test_db, query, expected_num_results): function test_delete (line 286) | def test_delete(test_db, weaviate_client, caplog): function test_build_auth_credentials (line 318) | def test_build_auth_credentials(monkeypatch): function test_extract_schema_properties (line 352) | def test_extract_schema_properties(): function test_reuse_schema (line 379) | def test_reuse_schema(weaviate_client, caplog): function test_build_date_filters (line 391) | def test_build_date_filters(): function test_is_valid_weaviate_id (line 430) | def test_is_valid_weaviate_id(test_input, expected_result): function test_upsert_same_docid (line 435) | def test_upsert_same_docid(test_db, weaviate_client): function test_is_wcs_domain (line 537) | def test_is_wcs_domain(url, expected_result): FILE: tests/datastore/providers/zilliz/test_zilliz_datastore.py function zilliz_datastore (line 20) | def zilliz_datastore(): function test_zilliz (line 25) | async def test_zilliz(zilliz_datastore):