remove llm_params

Author: Pouyanpi

nemoguardrails/actions/llm/generation.py

@@ -55,7 +55,6 @@
 )
 from nemoguardrails.embeddings.index import EmbeddingsIndex, IndexItem
 from nemoguardrails.kb.kb import KnowledgeBase
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.prompts import get_prompt
 from nemoguardrails.llm.taskmanager import LLMTaskManager, ParsedTaskOutput
 from nemoguardrails.llm.types import Task

nemoguardrails/actions/v2_x/generation.py

@@ -56,7 +56,6 @@
 )
 from nemoguardrails.embeddings.index import EmbeddingsIndex, IndexItem
 from nemoguardrails.llm.filters import colang
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.types import Task
 from nemoguardrails.logging import verbose
 from nemoguardrails.logging.explain import LLMCallInfo

nemoguardrails/eval/check.py

@@ -36,7 +36,6 @@
     InteractionSet,
 )
 from nemoguardrails.eval.ui.utils import EvalData
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.logging.explain import LLMCallInfo
 from nemoguardrails.rails.llm.config import Model

nemoguardrails/evaluate/evaluate_hallucination.py

@@ -23,7 +23,6 @@
 
 from nemoguardrails import LLMRails
 from nemoguardrails.evaluate.utils import load_dataset
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.prompts import Task
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.rails.llm.config import RailsConfig

nemoguardrails/evaluate/evaluate_moderation.py

@@ -22,7 +22,6 @@
 from nemoguardrails import LLMRails
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.evaluate.utils import load_dataset
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.prompts import Task
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.rails.llm.config import RailsConfig
@@ -235,29 +234,29 @@ def run(self):
 
         if self.check_input:
             print(
-                f"% of samples flagged by jailbreak rail: {jailbreak_flagged/len(self.dataset) * 100}"
+                f"% of samples flagged by jailbreak rail: {jailbreak_flagged / len(self.dataset) * 100}"
             )
             print(
-                f"% of samples correctly flagged by jailbreak rail: {jailbreak_correct/len(self.dataset) * 100}"
+                f"% of samples correctly flagged by jailbreak rail: {jailbreak_correct / len(self.dataset) * 100}"
             )
             if jailbreak_error > 0:
                 print(
-                    f"% of samples where jailbreak model or rail errored out: {jailbreak_error/len(self.dataset) * 100}"
+                    f"% of samples where jailbreak model or rail errored out: {jailbreak_error / len(self.dataset) * 100}"
                 )
             print("\n")
             print("*" * 50)
             print("\n")
 
         if self.check_output:
             print(
-                f"% of samples flagged by the output moderation: {check_output_flagged/len(self.dataset) * 100}"
+                f"% of samples flagged by the output moderation: {check_output_flagged / len(self.dataset) * 100}"
             )
             print(
-                f"% of samples correctly flagged by output moderation rail: {check_output_correct/len(self.dataset) * 100}"
+                f"% of samples correctly flagged by output moderation rail: {check_output_correct / len(self.dataset) * 100}"
             )
             if check_output_error > 0:
                 print(
-                    f"% of samples where output moderation model or rail errored out: {check_output_error/len(self.dataset) * 100}"
+                    f"% of samples where output moderation model or rail errored out: {check_output_error / len(self.dataset) * 100}"
                 )
             print("\n")
             print(

nemoguardrails/library/content_safety/actions.py

@@ -21,7 +21,6 @@
 from nemoguardrails.actions.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.context import llm_call_info_var
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.logging.explain import LLMCallInfo
 

nemoguardrails/library/llama_guard/actions.py

@@ -21,7 +21,6 @@
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.context import llm_call_info_var
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.llm.types import Task
 from nemoguardrails.logging.explain import LLMCallInfo

nemoguardrails/library/patronusai/actions.py

@@ -24,7 +24,6 @@
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.context import llm_call_info_var
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.llm.types import Task
 from nemoguardrails.logging.explain import LLMCallInfo

nemoguardrails/library/self_check/facts/actions.py

@@ -22,7 +22,6 @@
 from nemoguardrails.actions import action
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.context import llm_call_info_var
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.llm.types import Task
 from nemoguardrails.logging.explain import LLMCallInfo

nemoguardrails/library/self_check/input_check/actions.py

@@ -22,7 +22,6 @@
 from nemoguardrails.actions.actions import ActionResult, action
 from nemoguardrails.actions.llm.utils import llm_call
 from nemoguardrails.context import llm_call_info_var
-from nemoguardrails.llm.params import llm_params
 from nemoguardrails.llm.taskmanager import LLMTaskManager
 from nemoguardrails.llm.types import Task
 from nemoguardrails.logging.explain import LLMCallInfo