Milvus

Code

1import asyncio
2
3from kern.agent import Agent
4from kern.knowledge.knowledge import Knowledge
5from kern.vectordb.milvus import Milvus
6
7vector_db = Milvus(
8 collection="recipes",
9 uri="tmp/milvus.db",
10)
11
12knowledge = Knowledge(
13 name="My Milvus Knowledge Base",
14 description="This is a knowledge base that uses a Milvus DB",
15 vector_db=vector_db,
16)
17
18asyncio.run(
19 knowledge.ainsert(
20 name="Recipes",
21 url="https://kern-public.s3.amazonaws.com/recipes/ThaiRecipes.pdf",
22 metadata={"doc_type": "recipe_book"},
23 )
24)
25
26agent = Agent(knowledge=knowledge)
27agent.print_response("How to make Tom Kha Gai", markdown=True)
28
29vector_db.delete_by_name("Recipes")
30
31vector_db.delete_by_metadata({"doc_type": "recipe_book"})

Usage

Set up your virtual environment

1uv venv --python 3.12
2source .venv/bin/activate
1uv venv --python 3.12
2.venv\Scripts\activate

Install dependencies

1uv pip install -U pymilvus pypdf openai kern-ai

Run Agent

1python cookbook/08_knowledge/vector_db/milvus_db/milvus_db.py