Spaces:
Build error
Build error
| import time | |
| from typing import Any | |
| from util.chatbot import SearchBot | |
| from util.vector_store import VectorStore | |
| from util.logger import logger | |
| class API(): | |
| HF_SPACE="JR818/Jarvis-Backend" | |
| PROJECT='filecoin' | |
| def __init__(self,mode=None) -> None: | |
| self.mode = mode | |
| self.vector_store = VectorStore() | |
| self.search_bot = SearchBot() | |
| logger("Created API") | |
| def auto_complete(self,query): | |
| if not query: | |
| return [] | |
| if self.mode == 'debug': | |
| suggestion = [ | |
| {"name":"Filecoin "+query,"url":"wwww"}, | |
| {"name":"Filecoin "+query,"url":"wwww"}, | |
| {"name":"Filecoin "+query,"url":"wwww"}, | |
| {"name":"Filecoin "+query,"url":"wwww"}, | |
| {"name":"Filecoin "+query,"url":"wwww"} | |
| ] | |
| return suggestion | |
| docs = self.vector_store.marginal_search(query) | |
| res = [] | |
| for doc in docs: | |
| name = doc.metadata.get('title_llm','') | |
| url = doc.metadata.get('source','') | |
| if name == '' or url == '': | |
| continue | |
| res.append({"name":name,"url":url}) | |
| logger.write_log({"query":query,"api":'auto_complete',"result":res}) | |
| return res | |
| def search(self,query): | |
| if query == '': | |
| yield [] | |
| yield "Please input a query to search" | |
| return | |
| if self.mode == 'debug': | |
| source = [ | |
| {"name":"Filecoin","url":"wwww","description":"xxxx"}, | |
| {"name":"Filecoin Doc","url":"wwww","description":"xxxx"}, | |
| {"name":"Filecoin Doc2","url":"wwww","description":"xxxx"}, | |
| ] | |
| yield source | |
| text =f"this is a sample answer to {query} " | |
| for i in text: | |
| yield i | |
| time.sleep(0.1) | |
| return | |
| docs = self.vector_store.search(query) | |
| source = [] | |
| context = [] | |
| for doc in docs: | |
| if doc.metadata['score']<0.2: | |
| continue | |
| if 'title_llm' in doc.metadata: | |
| source.append({ | |
| "name":doc.metadata.get('title_llm',''), | |
| "url":doc.metadata.get('source',''), | |
| "description":doc.metadata.get('description_llm','') | |
| }) | |
| context.append(doc.metadata.get('description','')) | |
| yield source | |
| context = "\n".join(context) | |
| # print(f"context {context}") | |
| answer = self.search_bot(query,context,stream=True) | |
| text = "" | |
| for i in answer: | |
| text += i | |
| yield i | |
| logger.write_log({ | |
| "query":query, | |
| "api":'search', | |
| "source":source, | |
| 'answer':text}) | |