1from os import getenv
2
3from kern.agent import Agent
4from kern.knowledge.knowledge import Knowledge
5from kern.knowledge.reader.wikipedia_reader import WikipediaReader
6from kern.vectordb.lightrag import LightRag
7
8vector_db = LightRag(
9 api_key=getenv("LIGHTRAG_API_KEY"),
10)
11
12knowledge = Knowledge(
13 name="My LightRag Knowledge Base",
14 description="This is a knowledge base that uses a LightRag Vector DB",
15 vector_db=vector_db,
16)
17
18knowledge.insert(
19 name="Recipes",
20 path="cookbook/08_knowledge/testing_resources/cv_1.pdf",
21 metadata={"doc_type": "recipe_book"},
22)
23
24knowledge.insert(
25 name="Recipes",
26 topics=["Manchester United"],
27 reader=WikipediaReader(),
28)
29
30knowledge.insert(
31 name="Recipes",
32 url="https://en.wikipedia.org/wiki/Manchester_United_F.C.",
33)
34
35agent = Agent(
36 knowledge=knowledge,
37 search_knowledge=True,
38 read_chat_history=False,
39)
40
41agent.print_response("What skills does Jordan Mitchell have?", markdown=True)
42
43agent.print_response(
44 "In what year did Manchester United change their name?", markdown=True
45)
