mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-25 04:40:06 +00:00
multiple: pydantic 2 compatibility, v0.3 (#26443)
Signed-off-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> Co-authored-by: Dan O'Donovan <dan.odonovan@gmail.com> Co-authored-by: Tom Daniel Grande <tomdgrande@gmail.com> Co-authored-by: Grande <Tom.Daniel.Grande@statsbygg.no> Co-authored-by: Bagatur <baskaryan@gmail.com> Co-authored-by: ccurme <chester.curme@gmail.com> Co-authored-by: Harrison Chase <hw.chase.17@gmail.com> Co-authored-by: Tomaz Bratanic <bratanic.tomaz@gmail.com> Co-authored-by: ZhangShenao <15201440436@163.com> Co-authored-by: Friso H. Kingma <fhkingma@gmail.com> Co-authored-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Nuno Campos <nuno@langchain.dev> Co-authored-by: Morgante Pell <morgantep@google.com>
This commit is contained in:
@@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Any, Generator
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
from pydantic import AnyHttpUrl
|
||||
|
||||
from langchain_community.utilities.nvidia_riva import (
|
||||
AudioStream,
|
||||
@@ -126,7 +127,10 @@ def stream() -> AudioStream:
|
||||
def test_init(asr: RivaASR) -> None:
|
||||
"""Test that ASR accepts valid arguments."""
|
||||
for key, expected_val in CONFIG.items():
|
||||
assert getattr(asr, key, None) == expected_val
|
||||
if key == "url":
|
||||
assert asr.url == AnyHttpUrl(expected_val) # type: ignore
|
||||
else:
|
||||
assert getattr(asr, key, None) == expected_val
|
||||
|
||||
|
||||
@pytest.mark.requires("riva.client")
|
||||
@@ -162,7 +166,7 @@ def test_get_service(asr: RivaASR) -> None:
|
||||
svc = asr._get_service()
|
||||
assert str(svc.auth.ssl_cert) == CONFIG["ssl_cert"]
|
||||
assert svc.auth.use_ssl == SVC_USE_SSL
|
||||
assert svc.auth.uri == SVC_URI
|
||||
assert str(svc.auth.uri) == SVC_URI
|
||||
|
||||
|
||||
@pytest.mark.requires("riva.client")
|
||||
|
@@ -57,7 +57,10 @@ def tts() -> RivaTTS:
|
||||
def test_init(tts: RivaTTS) -> None:
|
||||
"""Test that ASR accepts valid arguments."""
|
||||
for key, expected_val in CONFIG.items():
|
||||
assert getattr(tts, key, None) == expected_val
|
||||
if key == "url":
|
||||
assert str(tts.url) == expected_val + "/" # type: ignore
|
||||
else:
|
||||
assert getattr(tts, key, None) == expected_val
|
||||
|
||||
|
||||
@pytest.mark.requires("riva.client")
|
||||
|
Reference in New Issue
Block a user