From 2b92a501fd940441bb91c9a2ba6a8355b9555244 Mon Sep 17 00:00:00 2001 From: Fraser Date: Mon, 12 Jun 2023 21:36:53 -0400 Subject: [PATCH] include letter key in citation form --- api/chat.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/api/chat.py b/api/chat.py index bb82b9c..03a2f53 100644 --- a/api/chat.py +++ b/api/chat.py @@ -192,9 +192,14 @@ def talk_to_robot_simple(index, query: str): for block in talk_to_robot_internal(index, query, []): if block['state'] == 'loading' and block['phase'] == 'semantic' and 'citations' in block: - res['citations'] = block['citations'] + citations = {} + for i, c in enumerate(block['citations']): + citations[chr(ord('a') + i)] = c + res['citations'] = citations + elif block['state'] == 'streaming': res['response'] += block['content'] + elif block['state'] == 'error': res['response'] = block['error']