Update chat.py

chat.py

@@ -16,6 +16,14 @@ def langchainConversation(conversation):
     chat_template = ChatPromptTemplate.from_messages(prompts)
     return chat_template.format_messages()
 def segmind_input_parser(input):
+    toreturn = []
+    for thisdict in input:
+        if(thisdidt['role'] == 'ai'):
+            toreturn.append({'role':'assistant','content':thisdict['context']})
+        else:
+            toreturn.append({'role':thisdict['role'],'content':thisdict['context']})
+    return toreturn
+def workers_input_parser(input):
     toreturn = []
     for thisdict in input:
         toreturn.append({'role':thisdict['role'],'content':thisdict['context']})