summaryrefslogtreecommitdiff
path: root/src/ollama_web/tools/registry.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/ollama_web/tools/registry.py')
-rw-r--r--src/ollama_web/tools/registry.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/ollama_web/tools/registry.py b/src/ollama_web/tools/registry.py
index 92b321c..2b6c7ae 100644
--- a/src/ollama_web/tools/registry.py
+++ b/src/ollama_web/tools/registry.py
@@ -6,6 +6,8 @@ import json
from collections.abc import Callable, Sequence
from dataclasses import dataclass, field
+from .fetch import search_and_fetch
+from .pdf import pdf_to_text
from .scrape import scrape_url
from .search import web_search
@@ -72,6 +74,8 @@ def default_registry() -> ToolRegistry:
reg = ToolRegistry()
reg.register(web_search)
reg.register(scrape_url)
+ reg.register(search_and_fetch)
+ reg.register(pdf_to_text)
return reg