Sathvika-Alla commited on
Commit
456993f
·
verified ·
1 Parent(s): 34060d3

Update chatbot-gradio.py

Browse files
Files changed (1) hide show
  1. chatbot-gradio.py +15 -2
chatbot-gradio.py CHANGED
@@ -245,9 +245,19 @@ async def handle_query(user_input: str, session_state:str):
245
  prompt=prompt,
246
  settings=settings
247
  )
 
 
 
 
 
 
 
 
 
248
 
249
- func_name = result.model_dump()["metadata"]["messages"]["messages"][2]["items"][0]["name"] if result.model_dump()["metadata"]["messages"]["messages"][2]["items"][0]["name"] else None
250
- print(func_name)
 
251
  log_func = kernel.get_function("ChatMemoryPlugin", "log_interaction")
252
  await log_func.invoke(
253
  kernel=kernel,
@@ -258,6 +268,9 @@ async def handle_query(user_input: str, session_state:str):
258
  )
259
 
260
  return str(result)
 
 
 
261
 
262
  except Exception as e:
263
  # Handle errors properly
 
245
  prompt=prompt,
246
  settings=settings
247
  )
248
+
249
+ metadata = result.model_dump().get("metadata", {})
250
+ messages = metadata.get("messages", {})
251
+ message_list = messages.get("messages", [])
252
+
253
+ if len(message_list) > 2 and message_list[2].get("items") and len(message_list[2]["items"]) > 0:
254
+ func_name = message_list[2]["items"][0].get("name")
255
+ else:
256
+ func_name = None
257
 
258
+ # func_name = result.model_dump()["metadata"]["messages"]["messages"][2]["items"][0]["name"] if result.model_dump()["metadata"]["messages"]["messages"][2]["items"][0]["name"] else None
259
+ # print(func_name)
260
+
261
  log_func = kernel.get_function("ChatMemoryPlugin", "log_interaction")
262
  await log_func.invoke(
263
  kernel=kernel,
 
268
  )
269
 
270
  return str(result)
271
+
272
+ except (KeyError, IndexError, AttributeError):
273
+ func_name = None
274
 
275
  except Exception as e:
276
  # Handle errors properly