22 lines
602 B
Python
22 lines
602 B
Python
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}
|
|
]
|