Spaces:
Running
Running
Commit ·
c119621
1
Parent(s): cdeed69
updated code, to allow continuing a retrieved/saved conversation, in api_server and conversation_manager
Browse files- api_server.py +2 -2
- core/memory_store.py +5 -4
api_server.py
CHANGED
|
@@ -175,8 +175,8 @@ def chat(req: ChatRequest, background_tasks: BackgroundTasks):
|
|
| 175 |
try:
|
| 176 |
state = from_dict(saved_data)
|
| 177 |
except Exception as e:
|
| 178 |
-
|
| 179 |
-
|
| 180 |
|
| 181 |
else:
|
| 182 |
#sessions[session_id] = ConversationState()
|
|
|
|
| 175 |
try:
|
| 176 |
state = from_dict(saved_data)
|
| 177 |
except Exception as e:
|
| 178 |
+
print("❌ RESTORE FAILED:", e)
|
| 179 |
+
state = ConversationState()
|
| 180 |
|
| 181 |
else:
|
| 182 |
#sessions[session_id] = ConversationState()
|
core/memory_store.py
CHANGED
|
@@ -1,14 +1,15 @@
|
|
| 1 |
-
|
| 2 |
# ./core/memory_store.py
|
| 3 |
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
from datetime import datetime
|
| 7 |
|
| 8 |
-
MEMORY_DIR = "memory"
|
| 9 |
|
| 10 |
if not os.path.exists(MEMORY_DIR):
|
| 11 |
-
os.makedirs(MEMORY_DIR)
|
|
|
|
| 12 |
|
| 13 |
|
| 14 |
def generate_label(history):
|
|
@@ -36,7 +37,7 @@ def save_state(session_id, state):
|
|
| 36 |
|
| 37 |
with open(filepath, "w", encoding="utf-8") as f:
|
| 38 |
json.dump(data, f, indent=2)
|
| 39 |
-
|
| 40 |
|
| 41 |
def load_state(session_id):
|
| 42 |
|
|
|
|
| 1 |
+
|
| 2 |
# ./core/memory_store.py
|
| 3 |
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
from datetime import datetime
|
| 7 |
|
| 8 |
+
MEMORY_DIR = "data/memory"
|
| 9 |
|
| 10 |
if not os.path.exists(MEMORY_DIR):
|
| 11 |
+
#os.makedirs(MEMORY_DIR)
|
| 12 |
+
os.makedirs(MEMORY_DIR, exist_ok=True)
|
| 13 |
|
| 14 |
|
| 15 |
def generate_label(history):
|
|
|
|
| 37 |
|
| 38 |
with open(filepath, "w", encoding="utf-8") as f:
|
| 39 |
json.dump(data, f, indent=2)
|
| 40 |
+
print(f"📁 MEMORY DIR: {os.path.abspath(MEMORY_DIR)}\n💾 SAVED SESSION: {session_id} at {filepath}")
|
| 41 |
|
| 42 |
def load_state(session_id):
|
| 43 |
|