From 1911d8db945999f5f71e1e40b8860f1e159d8576 Mon Sep 17 00:00:00 2001
From: Sashin <sashinchetty8@gmail.com>
Date: Tue, 5 Dec 2023 22:36:45 +0200
Subject: [PATCH] implementing chroma

---
 playground/memory/chroma_usage.py | 23 +++--------------------
 1 file changed, 3 insertions(+), 20 deletions(-)

diff --git a/playground/memory/chroma_usage.py b/playground/memory/chroma_usage.py
index 64aa347c..21ae475a 100644
--- a/playground/memory/chroma_usage.py
+++ b/playground/memory/chroma_usage.py
@@ -1,27 +1,10 @@
 from swarms.memory import chroma
 
-# loader = CSVLoader(
-#     file_path="../document_parsing/aipg/aipg.csv",
-#     encoding="utf-8-sig",
-# )
-# docs = loader.load()
+chromadbcl = chroma.ChromaClient()
 
+chromadbcl.add_vectors(["This is a document", "BONSAIIIIIII", "the walking dead"])
 
-# Initialize the Qdrant instance
-# See qdrant documentation on how to run locally
-qdrant_client = chroma.ChromaClient()
-
-qdrant_client.add_vectors(["This is a document", "BONSAIIIIIII", "the walking dead"])
-
-results = qdrant_client.search_vectors("zombie", limit=1)
+results = chromadbcl.search_vectors("zombie", limit=1)
 
 print(results)
 
-# qdrant_client.add_vectors(docs)
-#
-# # Perform a search
-# search_query = "Who is jojo"
-# search_results = qdrant_client.search_vectors(search_query)
-# print("Search Results:")
-# for result in search_results:
-#     print(result)