File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
nemoguardrails/llm/providers Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 3232from langchain_nvidia_ai_endpoints import ChatNVIDIA as ChatNVIDIAOriginal
3333from pydantic import Field
3434
35- log = logging .getLogger (__name__ ) # pragma: no cover
35+ log = logging .getLogger (__name__ )
3636
3737
38- def stream_decorator (func ): # pragma: no cover
38+ def stream_decorator (func ):
3939 @wraps (func )
4040 def wrapper (
4141 self ,
@@ -81,7 +81,7 @@ async def wrapper(
8181
8282# NOTE: this needs to have the same name as the original class,
8383# otherwise, there's a check inside `langchain-nvidia-ai-endpoints` that will fail.
84- class ChatNVIDIA (ChatNVIDIAOriginal ): # pragma: no cover
84+ class ChatNVIDIA (ChatNVIDIAOriginal ):
8585 streaming : bool = Field (
8686 default = False , description = "Whether to use streaming or not"
8787 )
You can’t perform that action at this time.
0 commit comments