Saad0KH commited on
Commit
1a2d238
·
verified ·
1 Parent(s): 7598e01

Update chromedb_service.py

Browse files
Files changed (1) hide show
  1. 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', port=80)
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', port=80)
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', port=80)
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', port=80)
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', port=80)
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],