Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

added dotenv in .toml, added an example to use qdrant, fixed the code in main.py #1653

Merged
merged 9 commits into from
Aug 14, 2024
Merged

added dotenv in .toml, added an example to use qdrant, fixed the code in main.py #1653

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift click to select a range
a9fd00d
added litellm and dotenv in .toml, updated main.py in the init method…
rajib76 Aug 4, 2024
63983ce
Merge branch 'main' of https://github.com/rajib76/mem0 into main
rajib76 Aug 6, 2024
f0e3da7
added dotenv, added example for qdrant cloud, fixed main.py in init
rajib76 Aug 6, 2024
29dfa9c
removed python dot env
rajib76 Aug 8, 2024
e9908e0
changed to qdrant_host and changed user to test
rajib76 Aug 9, 2024
8771400
changed qdrant
rajib76 Aug 9, 2024
e448fcc
changed qdrant
rajib76 Aug 9, 2024
d78be60
changed qdrant
rajib76 Aug 9, 2024
b3615c1
removed line 37 from main.py per Dev's comment
rajib76 Aug 14, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 40 additions & 0 deletions cookbooks/add_memory_using_qdrant_cloud.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 1,40 @@
# This example shows how to use vector config to use QDRANT CLOUD
import os
from dotenv import load_dotenv
from mem0 import Memory

# Loading OpenAI API Key
load_dotenv()
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
USER_ID = "test"
quadrant_host="xx.gcp.cloud.qdrant.io"

# creating the config attributes
collection_name="memory" # this is the collection I created in QDRANT cloud
api_key=os.environ.get("QDRANT_API_KEY") # Getting the QDRANT api KEY
host=quadrant_host
port=6333 #Default port for QDRANT cloud

# Creating the config dict
config = {
"vector_store": {
"provider": "qdrant",
"config": {
"collection_name": collection_name,
"host": host,
"port": port,
"path": None,
"api_key":api_key
}
}
}

# this is the change, create the memory class using from config
memory = Memory().from_config(config)

USER_DATA = """
I am a strong believer in memory architecture.
"""

response = memory.add(USER_DATA, user_id=USER_ID)
print(response)
2 changes: 1 addition & 1 deletion mem0/memory/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 33,7 @@ def __init__(self, config: MemoryConfig = MemoryConfig()):
self.vector_store = VectorStoreFactory.create(self.config.vector_store.provider, self.config.vector_store.config)
self.llm = LlmFactory.create(self.config.llm.provider, self.config.llm.config)
self.db = SQLiteManager(self.config.history_db_path)
self.collection_name = self.config.vector_store.config.collection_name if "collection_name" in self.config.vector_store.config else "mem0"
self.collection_name = self.config.vector_store.config.collection_name

capture_event("mem0.init", self)

Expand Down
Loading