| """service to handle & process the browser""" | |
| from Brain.src.model.requests.request_model import BrowserItem | |
| from Brain.src.rising_plugin.risingplugin import getCompletionOnly | |
| class BrowserService: | |
| """query to get the link of the item from the list""" | |
| def query_item(self, items: list[BrowserItem.ItemReq], query: str) -> str: | |
| prompt_template = f""" | |
| User is trying to '{query}' and it includes the title of the item. | |
| Please return the link of best relatedness of the item with the title from the below data.\n {items}""" | |
| return getCompletionOnly(query=prompt_template) | |
| def query_ask(self, items: list[str], query: str) -> str: | |
| prompt_template = f""" | |
| User is asking question related to website that users visit. The following is the user's question. | |
| '{query}'.\n | |
| Also the below data is list of sentence in the website. | |
| '{items}' \n | |
| Please provide me the proper answer.\n """ | |
| return getCompletionOnly(query=prompt_template) | |