from modaic import Retriever

from .config import AgentWRetreiverConfig
from .tools.jira.jira_api_tools import call_jira_api  # noqa: F401
from .utils.used import random_util  # noqa: F401


class ExampleRetriever(Retriever):
    config: AgentWRetreiverConfig

    def __init__(self, config: AgentWRetreiverConfig, needed_param: str, **kwargs):
        super().__init__(config, **kwargs)
        self.embedder_name = config.embedder
        self.needed_param = needed_param

    def retrieve(self, query: str) -> str:
        return f"Retrieved {self.config.num_fetch} results for {query}"