mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
8 lines
256 B
Python
8 lines
256 B
Python
|
"""Test TritonTensorRT Chat API wrapper."""
|
||
|
|
from langchain_nvidia_trt import TritonTensorRTLLM
|
||
|
|
|
||
|
|
|
||
|
|
def test_initialization() -> None:
|
||
|
|
"""Test integration initialization."""
|
||
|
|
TritonTensorRTLLM(model_name="ensemble", server_url="http://localhost:8001")
|