gerasdf commited on
Commit ·
0594bb7
1
Parent(s): 551b27a
sessions only saved for real users, names of sesssions are better, etc
Browse files
query.py
CHANGED
|
@@ -104,6 +104,9 @@ def new_state():
|
|
| 104 |
"history" : None,
|
| 105 |
})
|
| 106 |
|
|
|
|
|
|
|
|
|
|
| 107 |
store = None
|
| 108 |
def auth(token, state, request: gr.Request):
|
| 109 |
global store
|
|
@@ -122,7 +125,7 @@ def auth(token, state, request: gr.Request):
|
|
| 122 |
token=os.environ.get("ASTRA_DB_APPLICATION_TOKEN"),
|
| 123 |
api_endpoint=os.environ.get("ASTRA_DB_API_ENDPOINT"),
|
| 124 |
)
|
| 125 |
-
user_session =
|
| 126 |
session_data = {
|
| 127 |
'user' : state["user"],
|
| 128 |
'session' : request.session_hash,
|
|
@@ -149,7 +152,7 @@ def not_authenticated(state):
|
|
| 149 |
|
| 150 |
def add_history(state, request, type, message):
|
| 151 |
if not state["history"]:
|
| 152 |
-
session = request
|
| 153 |
state["history"] = AstraDBChatMessageHistory(
|
| 154 |
session_id=session,
|
| 155 |
collection_name=f'{os.environ.get("ASTRA_DB_COLLECTION")}_chat_history',
|
|
|
|
| 104 |
"history" : None,
|
| 105 |
})
|
| 106 |
|
| 107 |
+
def session_id(state: dict, request: gr.Request) -> str:
|
| 108 |
+
return f'{state["user"]}_{request.session_hash}'
|
| 109 |
+
|
| 110 |
store = None
|
| 111 |
def auth(token, state, request: gr.Request):
|
| 112 |
global store
|
|
|
|
| 125 |
token=os.environ.get("ASTRA_DB_APPLICATION_TOKEN"),
|
| 126 |
api_endpoint=os.environ.get("ASTRA_DB_API_ENDPOINT"),
|
| 127 |
)
|
| 128 |
+
user_session = session_id(state, request)
|
| 129 |
session_data = {
|
| 130 |
'user' : state["user"],
|
| 131 |
'session' : request.session_hash,
|
|
|
|
| 152 |
|
| 153 |
def add_history(state, request, type, message):
|
| 154 |
if not state["history"]:
|
| 155 |
+
session = session_id(state, request)
|
| 156 |
state["history"] = AstraDBChatMessageHistory(
|
| 157 |
session_id=session,
|
| 158 |
collection_name=f'{os.environ.get("ASTRA_DB_COLLECTION")}_chat_history',
|