Skip to content

Commit faaa477

Browse files
committed
make pinecone the default
1 parent db5913b commit faaa477

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

llm-complete-guide/steps/populate_index.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -634,7 +634,7 @@ class IndexType(Enum):
634634
@step(enable_cache=False)
635635
def index_generator(
636636
documents: str,
637-
index_type: IndexType = IndexType.POSTGRES,
637+
index_type: IndexType = IndexType.PINECONE,
638638
) -> None:
639639
"""Generates an index for the given documents.
640640
@@ -869,7 +869,7 @@ def _index_generator_pinecone(documents: str) -> None:
869869
"values": doc["embedding"],
870870
"metadata": {
871871
"filename": doc["filename"],
872-
"parent_section": doc["parent_section"],
872+
"parent_section": doc["parent_section"] or "",
873873
"url": doc["url"],
874874
"page_content": doc["page_content"],
875875
"token_count": doc["token_count"]

llm-complete-guide/utils/llm_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -536,7 +536,7 @@ def find_vectorstore_name() -> str:
536536
return model_version.run_metadata["vector_store"]["name"]
537537
except KeyError:
538538
logger.error("Vector store metadata not found in model version")
539-
return "pgvector" # Fallback to default
539+
return "pinecone" # Fallback to default
540540

541541

542542
def rerank_documents(

0 commit comments

Comments
 (0)