langchain/libs/community/tests/integration_tests/llms/test_titan_takeoff_pro.py

"""Test Titan Takeoff wrapper."""


import responses

from langchain_community.llms.titan_takeoff_pro import TitanTakeoffPro


@responses.activate
def test_titan_takeoff_pro_call() -> None:
    """Test valid call to Titan Takeoff."""
    url = "http://localhost:3000/generate"
    responses.add(responses.POST, url, json={"message": "2 + 2 is 4"}, status=200)

    # response = requests.post(url)
    llm = TitanTakeoffPro()
    output = llm("What is 2 + 2?")
    assert isinstance(output, str)