You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
678 B
28 lines
678 B
1 year ago
|
from swarms.memory import chroma
|
||
|
|
|
||
|
|
# loader = CSVLoader(
|
||
|
|
# file_path="../document_parsing/aipg/aipg.csv",
|
||
|
|
# encoding="utf-8-sig",
|
||
|
|
# )
|
||
|
|
# docs = loader.load()
|
||
|
|
|
||
|
|
|
||
|
|
# 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)
|
||
|
|
|
||
|
|
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)
|