Skip to content

Commit 9c120c2

Browse files
committed
Fix imports in tests
1 parent 815fc34 commit 9c120c2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/test_embeddings_only_user_messages.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
import pytest
1919

2020
from nemoguardrails import LLMRails, RailsConfig
21-
from nemoguardrails.actions.llm.utils import LLMCallException
21+
from nemoguardrails.exceptions import LLMCallException
2222
from nemoguardrails.llm.filters import colang
2323
from tests.utils import TestChat
2424

tests/test_tool_calling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@
1919
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage, ToolMessage
2020

2121
from nemoguardrails.actions.llm.utils import (
22-
LLMCallException,
2322
_convert_messages_to_langchain_format,
2423
_extract_content,
2524
_store_tool_calls,
2625
get_and_clear_tool_calls_contextvar,
2726
llm_call,
2827
)
2928
from nemoguardrails.context import tool_calls_var
29+
from nemoguardrails.exceptions import LLMCallException
3030
from nemoguardrails.rails.llm.llmrails import GenerationResponse
3131

3232

0 commit comments

Comments
 (0)