aiagent.agentstores package

Submodules

aiagent.agentstores.elastic module

class ElasticSearchAgentStore(store_name, base_store_name, smnet, agent_llm, options, chain_type='stuff', mappings={'properties': {'context': {'fields': {'keyword': {'ignore_above': 256, 'type': 'keyword'}}, 'type': 'text'}, 'metadata': {'properties': {'source': {'type': 'keyword'}}}, 'text': {'fields': {'keyword': {'ignore_above': 256, 'type': 'keyword'}}, 'type': 'text'}, 'vector': {'dims': 1536, 'type': 'dense_vector'}}})

Bases: BaseAgentStore

add_content(chunks, metadatas)
check_existence_of_store_name(store_name=None)
direct_query(input)
get_document(source)
get_es()
prompt_query(prompt_id, **kwargs)
query(input)
split_content(base_name, content)

Module contents

class ElasticSearchAgentStore(store_name, base_store_name, smnet, agent_llm, options, chain_type='stuff', mappings={'properties': {'context': {'fields': {'keyword': {'ignore_above': 256, 'type': 'keyword'}}, 'type': 'text'}, 'metadata': {'properties': {'source': {'type': 'keyword'}}}, 'text': {'fields': {'keyword': {'ignore_above': 256, 'type': 'keyword'}}, 'type': 'text'}, 'vector': {'dims': 1536, 'type': 'dense_vector'}}})

Bases: BaseAgentStore

add_content(chunks, metadatas)
check_existence_of_store_name(store_name=None)
direct_query(input)
get_document(source)
get_es()
prompt_query(prompt_id, **kwargs)
query(input)
split_content(base_name, content)