v2.3.25

ai/knowledge_base.py

@@ -3,22 +3,20 @@
 from phi.knowledge.pdf import PDFUrlKnowledgeBase, PDFKnowledgeBase
 from phi.vectordb.pgvector import PgVector2
 
+from ai.settings import ai_settings
 from db.session import db_url
 
 pdf_knowledge_base = CombinedKnowledgeBase(
     sources=[
-        PDFUrlKnowledgeBase(
-            urls=["https://www.family-action.org.uk/content/uploads/2019/07/meals-more-recipes.pdf"]
-        ),
+        PDFUrlKnowledgeBase(urls=["https://phi-public.s3.amazonaws.com/recipes/ThaiRecipes.pdf"]),
         PDFKnowledgeBase(path="data/pdfs"),
     ],
-    # Store this knowledge base in ai.pdf_documents
     vector_db=PgVector2(
-        schema="ai",
         db_url=db_url,
+        # Store the embeddings in ai.pdf_documents
         collection="pdf_documents",
-        embedder=OpenAIEmbedder(model="text-embedding-3-small"),
+        embedder=OpenAIEmbedder(model=ai_settings.embedding_model),
     ),
     # 2 references are added to the prompt
     num_documents=2,
-)
+)

ai/settings.py

@@ -10,7 +10,7 @@ class AISettings(BaseSettings):
     gpt_4: str = "gpt-4-1106-preview"
     gpt_4_vision: str = "gpt-4-vision-preview"
     gpt_3_5: str = "gpt-3.5-turbo-1106"
-    embedding_model: str = "text-embedding-ada-002"
+    embedding_model: str = "text-embedding-3-small"
     default_max_tokens: int = 1024
     default_temperature: float = 0
 

ai/storage.py

@@ -3,7 +3,6 @@
 from db.session import db_url
 
 pdf_assistant_storage = PgAssistantStorage(
-    schema="ai",
     db_url=db_url,
     table_name="pdf_assistant",
 )

pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
   # Linting and Formatting
   "ruff",
   # phidata
-  "phidata==2.3.19",
+  "phidata==2.3.25",
 ]
 
 [build-system]

requirements.txt

@@ -34,7 +34,7 @@ numpy==1.26.3
 openai==1.11.0
 packaging==23.2
 pgvector==0.2.4
-phidata==2.3.19
+phidata==2.3.25
 pluggy==1.4.0
 psycopg[binary]==3.1.17
 psycopg-binary==3.1.17