[DERCBOT-1037] TU WIP

gen-ai/orchestrator-server/src/main/python/server/tests/services/test_rag_chain.py

@@ -141,11 +141,10 @@ async def test_rag_chain(
     # Call function
     await execute_rag_chain(query, debug=True)
 
+    # Assert that the given observability_setting is used
     mocked_get_callback_handler_factory.assert_called_once_with(
         setting=query.observability_setting
     )
-    # Assert LangChain qa chain is created using the expected settings from query
-
     # Assert qa chain is ainvoke()d with the expected settings from query
     mocked_chain.ainvoke.assert_called_once_with(
         input=inputs,
@@ -159,6 +158,7 @@ async def test_rag_chain(
         ),
         debug=mocked_rag_debug_data(query, mocked_rag_answer, mocked_callback, 1),
     )
+    # Assert the rag guard is called
     mocked_rag_guard.assert_called_once_with(
         inputs, response
     )