s3-knowledge-cookbooks-phi-2020

phidatahq · Nov 20, 2024 · e08c22d · e08c22d
1 parent aa4ecaf
commit e08c22d
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/cookbook/knowledge/s3_pdf.py b/cookbook/knowledge/s3_pdf.py
@@ -0,0 +1,15 @@
+from phi.agent import Agent
+from phi.knowledge.s3.pdf import S3PDFKnowledgeBase
+from phi.vectordb.pgvector import PgVector
+
+db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
+
+knowledge_base = S3PDFKnowledgeBase(
+    bucket_name="phi-public",
+    key="recipes/ThaiRecipes.pdf",
+    vector_db=PgVector(table_name="recipes", db_url=db_url),
+)
+knowledge_base.load(recreate=False)  # Comment out after first run
+
+agent = Agent(knowledge=knowledge_base, search_knowledge=True)
+agent.print_response("How to make Thai curry?", markdown=True)
diff --git a/cookbook/knowledge/s3_text.py b/cookbook/knowledge/s3_text.py
@@ -0,0 +1,15 @@
+from phi.agent import Agent
+from phi.knowledge.s3.text import S3TextKnowledgeBase
+from phi.vectordb.pgvector import PgVector
+
+db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
+
+knowledge_base = S3TextKnowledgeBase(
+    bucket_name="phi-public",
+    key="recipes/recipes.docx",
+    vector_db=PgVector(table_name="recipes", db_url=db_url),
+)
+knowledge_base.load(recreate=False)  # Comment out after first run
+
+agent = Agent(knowledge=knowledge_base, search_knowledge=True)
+agent.print_response("How to make Hummus?", markdown=True)