Milvus

pip install "upsonic[rag]"

from upsonic import Agent, Task, KnowledgeBase
from upsonic.embeddings.openai_provider import OpenAIEmbeddingProvider
from upsonic.vectordb import MilvusProvider, MilvusConfig, ConnectionConfig, Mode, HNSWIndexConfig

# Setup embedding provider
embedding = OpenAIEmbeddingProvider(api_key="your-api-key")

# Embedded mode (Milvus Lite)
config = MilvusConfig(
    collection_name="my_collection",
    vector_size=1536,
    connection=ConnectionConfig(mode=Mode.EMBEDDED, db_path="./milvus_db"),
    index=HNSWIndexConfig(m=16, ef_construction=200),
    consistency_level="Bounded"
)
vectordb = MilvusProvider(config)

# Create knowledge base
kb = KnowledgeBase(
    sources="document.pdf",
    embedding_provider=embedding,
    vectordb=vectordb
)

# Use with Agent
agent = Agent("openai/gpt-4o")
task = Task(
    description="Search the knowledge base",
    context=[kb]
)
result = agent.do(task)