style

Author: Pouyanpi

examples/configs/rag/custom_rag_output_rails/config.py

@@ -13,9 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from langchain_core.prompts import PromptTemplate
 from langchain_core.language_models.llms import BaseLLM
 from langchain_core.output_parsers import StrOutputParser
+from langchain_core.prompts import PromptTemplate
 
 from nemoguardrails import LLMRails
 from nemoguardrails.actions.actions import ActionResult

tests/llm_providers/test_langchain_integration.py

@@ -124,18 +124,18 @@ def test_langchain_provider_imports():
     for provider_name in llm_provider_names:
         try:
             provider_cls = _llm_providers[provider_name]
-            assert provider_cls is not None, (
-                f"Provider class for '{provider_name}' is None"
-            )
+            assert (
+                provider_cls is not None
+            ), f"Provider class for '{provider_name}' is None"
         except Exception as e:
             warnings.warn(f"Failed to import LLM provider '{provider_name}': {str(e)}")
 
     for provider_name in chat_provider_names:
         try:
             provider_cls = _chat_providers[provider_name]
-            assert provider_cls is not None, (
-                f"Provider class for '{provider_name}' is None"
-            )
+            assert (
+                provider_cls is not None
+            ), f"Provider class for '{provider_name}' is None"
         except Exception as e:
             warnings.warn(f"Failed to import chat provider '{provider_name}': {str(e)}")