1import typer
2from kern.agent import Agent
3from kern.knowledge.knowledge import Knowledge
4from kern.vectordb.mongodb import MongoVectorDb
5from kern.vectordb.search import SearchType
6from rich.prompt import Prompt
7
8mdb_connection_string = "mongodb://localhost:27017"
9
10vector_db = MongoVectorDb(
11 collection_name="recipes",
12 db_url=mdb_connection_string,
13 search_index_name="recipes",
14 search_type=SearchType.hybrid,
15)
16
17knowledge_base = Knowledge(
18 vector_db=vector_db,
19)
20
21knowledge_base.insert(
22 name="Recipes",
23 url="https://kern-public.s3.amazonaws.com/recipes/ThaiRecipes.pdf",
24 metadata={"doc_type": "recipe_book"},
25)
26
27def mongodb_agent(user: str = "user"):
28 agent = Agent(
29 user_id=user,
30 knowledge=knowledge_base,
31 search_knowledge=True,
32 )
33
34 while True:
35 message = Prompt.ask(f"[bold] :sunglasses: {user} [/bold]")
36 if message in ("exit", "bye"):
37 break
38 agent.print_response(message)
39
40if __name__ == "__main__":
41 typer.run(mongodb_agent)
