File size: 795 Bytes
11cc0d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import chromadb
from chromadb.utils import embedding_functions
# Adjust these as needed
CHROMA_PATH = "chroma_db"
COLLECTION_NAME = "pib_titles"
client = chromadb.PersistentClient(path=CHROMA_PATH)
collection = client.get_collection(
name=COLLECTION_NAME,
embedding_function=embedding_functions.SentenceTransformerEmbeddingFunction(
model_name="all-MiniLM-L6-v2"
)
)
# Retrieve all documents and metadata (ids are always returned)
all_docs = collection.get(include=["documents", "metadatas"])
print("Total documents:", len(all_docs["ids"]))
for i, (doc_id, doc, meta) in enumerate(zip(all_docs["ids"], all_docs["documents"], all_docs["metadatas"])):
print(f"\n--- Document {i+1} ---")
print("ID:", doc_id)
print("Document:", doc)
print("Metadata:", meta)
|