(no commit message)
This commit is contained in:
17
program/retriever.py
Normal file
17
program/retriever.py
Normal file
@@ -0,0 +1,17 @@
|
||||
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}"
|
||||
Reference in New Issue
Block a user