Add ignore comment to chatbot index errors

Author: cyab05

python/rubin/rag/chat_prompt_test.py

@@ -170,12 +170,12 @@ def handle_user_input(
                 },
                 {"callbacks": [stream_handler]},
             )
-            msgs.add_ai_message(result["answer"])
+            msgs.add_ai_message(result["answer"]) # type: ignore[index]
 
             # Display source documents in an expander
             with st.expander("See sources"):
                 scores = [
-                    chunk.metadata["score"] for chunk in result["context"]
+                    chunk.metadata["score"] for chunk in result["context"] # type: ignore[index]
                 ]
 
                 if scores:
@@ -184,7 +184,7 @@ def handle_user_input(
                         max_score * 0.9
                     )  # Set threshold to 90% of the highest score
 
-                    for chunk in result["context"]:
+                    for chunk in result["context"]: # type: ignore[index]
                         score = chunk.metadata["score"]
 
                         # Only show sources with scores