File size: 392 Bytes
88e3f4a
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from omniff.integrations.langchain import OmniFFTool


def test_omniff_tool_init():
    tool = OmniFFTool()
    assert tool.name == "omniff"
    assert tool.base_url == "http://localhost:8000"
    assert "multimodal" in tool.description.lower()


def test_omniff_tool_custom_url():
    tool = OmniFFTool(base_url="http://gpu-server:9000")
    assert tool.base_url == "http://gpu-server:9000"