Hackathons/Hoya26
1
0
Fork 0
mirror of https://github.com/SirBlobby/Hoya26.git synced 2026-02-04 03:34:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

gemeni endpoints

Browse Source
This commit is contained in:
KasaNick
2026-01-24 06:33:29 -05:00
parent d6471afab2
commit c46ec76027
7 changed files with 134 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/src/chroma/vector_store.py
View File

@@ -17,7 +17,9 @@ def get_collection(collection_name=COLLECTION_NAME):
def insert_documents(texts, embeddings, collection_name=COLLECTION_NAME, metadata_list=None):
collection = get_collection(collection_name)
ids = [f"doc_{i}_{hash(text)}" for i, text in enumerate(texts)]
if metadata_list:
collection.add(
ids=ids,
@@ -31,17 +33,22 @@ def insert_documents(texts, embeddings, collection_name=COLLECTION_NAME, metadat
embeddings=embeddings,
documents=texts
)
return len(texts)
def search_documents(query_embedding, collection_name=COLLECTION_NAME, num_results=5, filter_metadata=None):
collection = get_collection(collection_name)
query_params = {
"query_embeddings": [query_embedding],
"n_results": num_results
}
if filter_metadata:
query_params["where"] = filter_metadata
results = collection.query(**query_params)
output = []
if results and results["documents"]:
for i, doc in enumerate(results["documents"][0]):
@@ -50,6 +57,7 @@ def search_documents(query_embedding, collection_name=COLLECTION_NAME, num_resul
"text": doc,
"score": score
})
return output
def delete_documents_by_source(source_file, collection_name=COLLECTION_NAME):