fabricatio_milvus.actions.rag

Inject data into the database.

Classes

InjectToDB

Inject data into the database.

MilvusRAGTalk

RAG-enabled conversational action that processes user questions based on a given task.

Module Contents

class fabricatio_milvus.actions.rag.InjectToDB(/, **data: Any)

Bases: fabricatio_core.models.action.Action, fabricatio_milvus.capabilities.milvus.MilvusRAG

Inject data into the database.

output_key: str = 'collection_name'

The key used to store this action’s output in the context dictionary.

collection_name: str = 'my_collection'

The name of the collection to inject data into.

class fabricatio_milvus.actions.rag.MilvusRAGTalk(/, **data: Any)

Bases: fabricatio_core.models.action.Action, fabricatio_milvus.capabilities.milvus.MilvusRAG

RAG-enabled conversational action that processes user questions based on a given task.

This action establishes an interactive conversation loop where it retrieves context-relevant information to answer user queries according to the assigned task briefing.

Notes

task_input: Task briefing that guides how to respond to user questions collection_name: Name of the vector collection to use for retrieval (default: “my_collection”)

Returns:

Number of conversation turns completed before termination

output_key: str = 'task_output'

The key used to store this action’s output in the context dictionary.