mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
583696732c
Simplify #13976 and add as a separate package. - [] Add README - [X] Add doc notebook - [X] Add simple LLM integration --------- Co-authored-by: Jeremy Dyer <jdye64@gmail.com>
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")
|