update prompt

app.py

@@ -127,8 +127,8 @@ When responding to queries, follow these guidelines:
 
 3. Proper Source References:
    - ALWAYS USE INLINE CITATIONS and References with embed source URLs where users can verify information or explore further. 
-   - The inline citations should be in the format [1], [2] etc. where on clicking on it the user should be redirected to the source url in a new tab.
-   - Do not inlcude references at the end of response.
+   - The inline citations should be in the format '[Source]' where on clicking on 'Source' the user should be redirected to the source url in a new tab.
+   - Include References at the end if needed.
 
 FOLLOW ALL THE GIVEN INSTRUCTIONS, FAILURE TO DO SO WILL RESULT IN TERMINATION OF THE CHAT.
 
@@ -195,22 +195,11 @@ async def websocket_endpoint(websocket: WebSocket):
                         {"input": question, 'language': language},
                         config={"configurable": {"session_id": session_id}}
                     ):
-                        if "context" in chunk:
-                            context = chunk['context']
                         # Send each chunk to the client
                         if "answer" in chunk:
                             complete_response += chunk['answer']
                             await websocket.send_json({'response': chunk['answer']})
 
-                    if context:
-                        citations = re.findall(r'\[(\d+)\]', complete_response)
-                        citation_numbers = list(map(int, citations))
-                        sources = dict()
-                        for index, doc in enumerate(context):
-                            if (index+1) in citation_numbers:
-                                sources[f"[{index+1}]"] = doc.metadata["source"]
-                        await websocket.send_json({'sources': sources})
-
                 await stream_response()
             except Exception as e:
                 print(f"Error during message handling: {e}")