rakam_systems.generation.agents.ClassifyQuery

class rakam_systems.generation.agents.ClassifyQuery(agent, trigger_queries: Series, class_names: Series, embedding_model: str = 'all-MiniLM-L6-v2')

__init__(agent, trigger_queries: Series, class_names: Series, embedding_model: str = 'all-MiniLM-L6-v2')

Methods

__init__(agent, trigger_queries, class_names)
build_vector_store(trigger_queries, class_names)	Builds a VectorStores object from the trigger queries and class names.
execute(query)	Classifies the query by finding the closest match in the FAISS index.