add
This commit is contained in:
12
main.py
12
main.py
@@ -18,8 +18,9 @@ load_dotenv()
|
||||
CHROMA_DB_PATH = os.environ.get("CHROMA_DB_PATH", "/app/chroma_db")
|
||||
CHROMA_COLLECTION_NAME = os.environ.get("CHROMA_COLLECTION_NAME", "memories")
|
||||
|
||||
# SQLite configuration for graph store
|
||||
SQLITE_DB_PATH = os.environ.get("SQLITE_DB_PATH", "/app/graph_store.db")
|
||||
NEO4J_URI = os.environ.get("NEO4J_URI", "bolt://neo4j:7687")
|
||||
NEO4J_USERNAME = os.environ.get("NEO4J_USERNAME", "neo4j")
|
||||
NEO4J_PASSWORD = os.environ.get("NEO4J_PASSWORD", "mem0graph")
|
||||
|
||||
# Ollama configuration
|
||||
OLLAMA_HOST = os.environ.get("OLLAMA_HOST", "http://localhost:11434")
|
||||
@@ -38,11 +39,8 @@ DEFAULT_CONFIG = {
|
||||
},
|
||||
},
|
||||
"graph_store": {
|
||||
"provider": "chromadb",
|
||||
"config": {
|
||||
"path": CHROMA_DB_PATH,
|
||||
"collection_name": "graph",
|
||||
},
|
||||
"provider": "neo4j",
|
||||
"config": {"url": NEO4J_URI, "username": NEO4J_USERNAME, "password": NEO4J_PASSWORD},
|
||||
},
|
||||
"llm": {
|
||||
"provider": "ollama",
|
||||
|
||||
Reference in New Issue
Block a user