Gaykar commited on
Commit
4ed95ae
·
1 Parent(s): c6421b9

memory store part

Browse files
Files changed (1) hide show
  1. app/memory_store/memory_store.py +5 -1
app/memory_store/memory_store.py CHANGED
@@ -3,9 +3,13 @@ from langgraph.store.postgres import PostgresStore
3
  from langgraph.checkpoint.postgres import PostgresSaver
4
  from app.memory_store.embeddings import remote_embeddings
5
 
 
 
 
 
6
 
7
  pool = ConnectionPool(
8
- conninfo=DB_URI,
9
  min_size=1,
10
  max_size=10,
11
  kwargs={"autocommit": True}
 
3
  from langgraph.checkpoint.postgres import PostgresSaver
4
  from app.memory_store.embeddings import remote_embeddings
5
 
6
+ from app.core.config import settings
7
+
8
+
9
+ DB_URL_FOR_CHECKPOINTER_STORE=settings.DB_URL_FOR_CHECKPOINTER_STORE
10
 
11
  pool = ConnectionPool(
12
+ conninfo=DB_URL_FOR_CHECKPOINTER_STORE,
13
  min_size=1,
14
  max_size=10,
15
  kwargs={"autocommit": True}