resolved issue with factcheck tests

Author: abhijitpal1247

nemoguardrails/library/autoguard/actions.py

@@ -24,7 +24,6 @@
 from nemoguardrails.actions.actions import ActionResult
 from nemoguardrails.kb.kb import KnowledgeBase
 from nemoguardrails.llm.taskmanager import LLMTaskManager
-from nemoguardrails.logging.verbose import Styles
 
 log = logging.getLogger(__name__)
 
@@ -226,13 +225,11 @@ async def autoguard_input_api(
     )
     if autoguard_response["guardrails_triggered"] and show_autoguard_message:
         print(
-            Styles.YELLOW,
             f"AutoGuard on Input: {autoguard_response['combined_response']}",
         )
     else:
         if autoguard_response["pii_fast"]["guarded"] and show_autoguard_message:
             print(
-                Styles.YELLOW,
                 f"AutoGuard on Input: {autoguard_response['pii_fast']['response']}",
             )
 
@@ -262,7 +259,6 @@ async def autoguard_output_api(
     )
     if autoguard_response["guardrails_triggered"] and show_autoguard_message:
         print(
-            Styles.YELLOW,
             f"AutoGuard on LLM Response: {autoguard_response['combined_response']}",
         )
 

tests/test_autoguard_factcheck.py

@@ -47,6 +47,19 @@ async def retrieve_relevant_chunks():
     )
 
 
+@action(is_system_action=True)
+async def autoguard_retrieve_relevant_chunks_input():
+    """Retrieve relevant chunks from the knowledge base and add them to the context."""
+    context_updates = {}
+    relevant_chunks = "\n".join(build_kb())
+    context_updates["relevant_chunks"] = relevant_chunks
+
+    return ActionResult(
+        return_value=context_updates["relevant_chunks"],
+        context_updates=context_updates,
+    )
+
+
 @pytest.mark.asyncio
 async def test_fact_checking_correct(httpx_mock):
     config = RailsConfig.from_path(os.path.join(CONFIGS_FOLDER, "autoguard_factcheck"))