aiagent.smnetstores package

Submodules

aiagent.smnetstores.elastic module

class ElasticSearchSemanticNetworkStore(options, user, mappings, embedder)

Bases: object

add_relationships(relationships, metadata)

add_relationships_v2(relationships, metadata)

check_for_index_exist(store_name=None)

check_item_exists(source)

get_es()

class ExtendedElasticVectorSearch(elasticsearch_url, index_name, embedding)

Bases: ElasticVectorSearch

query(query: str, k: int = 14, minscore=1.7, **kwargs: Any) → List[Document]

Return docs most similar to query. Args:

query: Text to look up documents similar to. k: Number of Documents to return. Defaults to 4.

Returns:

List of Documents most similar to the query.

aiagent.smnetstores.postgres module

class PostgresSemanticNetworkStore(config, user_name)

Bases: BaseSemanticNetworkStore

add_relationships(relationships, metadata)

add_relationships_v2(relationships, metadata)

query(query)

Module contents

class ElasticSearchSemanticNetworkStore(options, user, mappings, embedder)

Bases: object

add_relationships(relationships, metadata)

add_relationships_v2(relationships, metadata)

check_for_index_exist(store_name=None)

check_item_exists(source)

get_es()

class PostgresSemanticNetworkStore(config, user_name)

Bases: BaseSemanticNetworkStore

add_relationships(relationships, metadata)

add_relationships_v2(relationships, metadata)

query(query)