import llm
import json
from llm_tools_tricu_bridge import tricubridge

def test_tool():
    model = llm.get_model("echo")
    chain_response = model.chain(
        json.dumps(
            {
                "tool_calls": [
                    {"name": "tricubridge", "arguments": {"input": "x = t t"}}
                ]
            }
        ),
        tools=[tricubridge],
    )
    responses = list(chain_response.responses())
    tool_results = json.loads(responses[-1].text())["tool_results"]
    assert tool_results == [
        {"name": "tricubridge", "output": "t t", "tool_call_id": None}
    ]