Skip to content

Commit f86569d

Browse files
committed
fix: fixed bug checking for browser tool when not available
1 parent 780b3c8 commit f86569d

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

gptme/cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from .message import Message
2828
from .models import get_model
2929
from .prompts import get_prompt
30-
from .tools import execute_msg, get_tool
30+
from .tools import execute_msg, has_tool
3131
from .tools.browser import read_url
3232
from .util import epoch_to_age, generate_name
3333

@@ -578,7 +578,7 @@ def _parse_prompt(prompt: str) -> str | None:
578578
for path in paths:
579579
result += _parse_prompt(path) or ""
580580

581-
if get_tool("browser") is None:
581+
if not has_tool("browser"):
582582
logger.warning("Browser tool not available, skipping URL read")
583583
else:
584584
for url in urls:

gptme/tools/__init__.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import logging
2-
from collections.abc import Generator
2+
from collections.abc import Callable, Generator
33
from dataclasses import dataclass
4-
from collections.abc import Callable
54
from xml.etree import ElementTree
65

76
from ..message import Message
@@ -210,3 +209,10 @@ def get_tool(tool_name: str) -> ToolSpec:
210209
if tool.name == tool_name:
211210
return tool
212211
raise ValueError(f"Tool '{tool_name}' not found")
212+
213+
214+
def has_tool(tool_name: str) -> bool:
215+
for tool in loaded_tools:
216+
if tool.name == tool_name:
217+
return True
218+
return False

0 commit comments

Comments
 (0)