ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-26 23:44:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Use logs instead of warnings

Browse source
This commit is contained in:
Jeremiah Lowin 2024-11-29 17:55:41 -05:00
commit b3ca9e2df1
4 changed files with 18 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_tool_manager.py
View file

@ -1,5 +1,4 @@
import warnings
import logging
import pytest
from pydantic import BaseModel
@ -83,7 +82,7 @@ class TestAddTools:
):
manager.add_tool(lambda x: x)
def test_warn_on_duplicate_tools(self):
def test_warn_on_duplicate_tools(self, caplog):
"""Test warning on duplicate tools."""
def f(x: int) -> int:
@ -91,10 +90,11 @@ class TestAddTools:
manager = ToolManager()
manager.add_tool(f)
with pytest.warns(ResourceWarning):
with caplog.at_level(logging.WARNING):
manager.add_tool(f)
assert "Tool already exists: f" in caplog.text
def test_disable_warn_on_duplicate_tools(self):
def test_disable_warn_on_duplicate_tools(self, caplog):
"""Test disabling warning on duplicate tools."""
def f(x: int) -> int:
@ -103,9 +103,9 @@ class TestAddTools:
manager = ToolManager()
manager.add_tool(f)
manager.warn_on_duplicate_tools = False
with warnings.catch_warnings():
warnings.simplefilter("error")
with caplog.at_level(logging.WARNING):
manager.add_tool(f)
assert "Tool already exists: f" not in caplog.text
class TestCallTools: