File size: 1,044 Bytes
6e1a53e |
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 26 27 28 29 30 31 32 |
from embedchain import App
from fastapi import APIRouter
from utils.embedchain import EC_APP_CONFIG
router = APIRouter()
ec_app = App.from_config(config=EC_APP_CONFIG)
@router.get("/api/v1/admin/collections")
async def get_all_collections():
# Currently only works for ChromaDB but can be extended easily
# for other vector stores as well
collections = ec_app.db.client.list_collections()
responses = [c.dict() for c in collections]
return responses
@router.get("/api/v1/admin/collections/chromadb/{collection_name}")
async def get_collection_details(collection_name: str):
collection = ec_app.db.client.get_collection(collection_name)
collection_data = collection.get()
metadatas, documents = collection_data['metadatas'], collection_data['documents']
collated_data = []
for i in zip(metadatas, documents):
collated_data.append({
"metadata": i[0],
"document": i[1]
})
response = {"details": collection.dict(), "data": collated_data}
return response
|