Spaces:
Running
Running
Update chromedb_service.py
Browse files- chromedb_service.py +5 -5
chromedb_service.py
CHANGED
@@ -28,7 +28,7 @@ def encode_image_from_url(image_url):
|
|
28 |
# Ajouter une image dans ChromaDB
|
29 |
def add_image_to_chroma(collection_name, id, image_url, metadata):
|
30 |
vector = encode_image_from_url(image_url)
|
31 |
-
chroma_client = chromadb.HttpClient(host='stable-diffusion-engine.oneiro-lego.com'
|
32 |
collection = chroma_client.get_or_create_collection(
|
33 |
name=collection_name, dimension=len(vector)
|
34 |
)
|
@@ -41,7 +41,7 @@ def add_image_to_chroma(collection_name, id, image_url, metadata):
|
|
41 |
|
42 |
# Ajouter un document dans ChromaDB
|
43 |
def add_document(collection_name, id, text, metadata):
|
44 |
-
chroma_client = chromadb.HttpClient(host='stable-diffusion-engine.oneiro-lego.com'
|
45 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
46 |
collection.upsert(
|
47 |
documents=[text],
|
@@ -52,20 +52,20 @@ def add_document(collection_name, id, text, metadata):
|
|
52 |
|
53 |
# Supprimer un document dans ChromaDB
|
54 |
def delete_document(collection_name, id):
|
55 |
-
chroma_client = chromadb.HttpClient(host='stable-diffusion-engine.oneiro-lego.com'
|
56 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
57 |
collection.delete(ids=[id])
|
58 |
print(f"Document {id} supprimé avec succès !")
|
59 |
|
60 |
# Supprimer une collection dans ChromaDB
|
61 |
def delete_collection(collection_name):
|
62 |
-
chroma_client = chromadb.HttpClient(host='stable-diffusion-engine.oneiro-lego.com'
|
63 |
chroma_client.delete_collection(name=collection_name)
|
64 |
print(f"Collection {collection_name} supprimée avec succès !")
|
65 |
|
66 |
# Recherche dans une collection
|
67 |
def search(collection_name, query, n_results):
|
68 |
-
chroma_client = chromadb.HttpClient(host='stable-diffusion-engine.oneiro-lego.com'
|
69 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
70 |
results = collection.query(
|
71 |
query_texts=[query],
|
|
|
28 |
# Ajouter une image dans ChromaDB
|
29 |
def add_image_to_chroma(collection_name, id, image_url, metadata):
|
30 |
vector = encode_image_from_url(image_url)
|
31 |
+
chroma_client = chromadb.HttpClient(host='https://stable-diffusion-engine.oneiro-lego.com')
|
32 |
collection = chroma_client.get_or_create_collection(
|
33 |
name=collection_name, dimension=len(vector)
|
34 |
)
|
|
|
41 |
|
42 |
# Ajouter un document dans ChromaDB
|
43 |
def add_document(collection_name, id, text, metadata):
|
44 |
+
chroma_client = chromadb.HttpClient(host='https://stable-diffusion-engine.oneiro-lego.com')
|
45 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
46 |
collection.upsert(
|
47 |
documents=[text],
|
|
|
52 |
|
53 |
# Supprimer un document dans ChromaDB
|
54 |
def delete_document(collection_name, id):
|
55 |
+
chroma_client = chromadb.HttpClient(host='https://stable-diffusion-engine.oneiro-lego.com')
|
56 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
57 |
collection.delete(ids=[id])
|
58 |
print(f"Document {id} supprimé avec succès !")
|
59 |
|
60 |
# Supprimer une collection dans ChromaDB
|
61 |
def delete_collection(collection_name):
|
62 |
+
chroma_client = chromadb.HttpClient(host='https://stable-diffusion-engine.oneiro-lego.com')
|
63 |
chroma_client.delete_collection(name=collection_name)
|
64 |
print(f"Collection {collection_name} supprimée avec succès !")
|
65 |
|
66 |
# Recherche dans une collection
|
67 |
def search(collection_name, query, n_results):
|
68 |
+
chroma_client = chromadb.HttpClient(host='https://stable-diffusion-engine.oneiro-lego.com')
|
69 |
collection = chroma_client.get_or_create_collection(name=collection_name)
|
70 |
results = collection.query(
|
71 |
query_texts=[query],
|