Julia Ostheimer commited on
Commit
67f186b
·
1 Parent(s): 768cf00

Log user input prompt as session state variable

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -180,15 +180,17 @@ def bot(message, history, session_data) -> list[Any]:
180
  # create text response
181
  # TODO: see how state can be set in chat interface
182
  config = {"configurable": {"thread_id": "abc123"}, "callbacks": [langfuse_handler]}
 
 
183
  response = graph.invoke(
184
- {"messages": [{"role": "user", "content": message.get("text")}]},
185
  config=config,
186
  )
187
 
188
  logger.info("Generated a response", response=response)
189
 
190
- # Example: append dict to session state
191
- session_data.append("test")
192
 
193
  return [response["messages"][-1].content], session_data
194
 
 
180
  # create text response
181
  # TODO: see how state can be set in chat interface
182
  config = {"configurable": {"thread_id": "abc123"}, "callbacks": [langfuse_handler]}
183
+
184
+ user_input_prompt = message.get("text")
185
  response = graph.invoke(
186
+ {"messages": [{"role": "user", "content": user_input_prompt}]},
187
  config=config,
188
  )
189
 
190
  logger.info("Generated a response", response=response)
191
 
192
+ # Log user input prompt to session state
193
+ session_data = user_input_prompt
194
 
195
  return [response["messages"][-1].content], session_data
196