""" title: Kagi Search author: Woovie author_url: https://woovie.net git_url: https://github.com/woovie/open-webui-kagi-search description: Kagi search tool for Open WebUI required_open_webui_version: 0.5.18 requirements: version: 0.0.1 licence: MIT """ import requests from pydantic import BaseModel, Field class Tools: def __init__(self): """Initialize the Kagi Search Tool.""" self.api_url = "https://kagi.com/api/v0/search" self.valves = self.Valves() class Valves(BaseModel): API_KEY: str = Field(default="", description="Your API key here") def kagi_search(self, query: str) -> str: """ Searches Kagi for the query and returns the results in JSON format. """ auth_header = { "Authorization": f"Bot {self.valves.API_KEY}" } results = ( requests.get( # TODO clean up query, space to plus sign (URI encode???) self.api_url, params={"q": query}, headers=auth_header ).json() ) return f"Results from Kagi Search:\n{results}"