Abbot – «qq v1.0.0» by Alumstabase

Code

<br /> # Change the code below with the code for your skill!<br /> import re<br /> import requests<br /> from typing import Tuple, List</p> <p>class MODELS:<br /> QA = «st_msmarco_distilbert_base_v3»<br /> SIMILARITY = «st_paraphrase_distilroberta_base_v1»</p> <p>class NLUDB:<br /> pass</p> <p>class NLUDBEmbeddingIndex:<br /> «»»<br /> An Embedding Index is a persistent index over embeddings.<br /> «»»</p> <p> def __init__(self, nludb: NLUDB, id: str, name: str):<br /> self.nludb = nludb<br /> self.name = name<br /> self.id = id</p> <p> def insert(self, value: str):<br /> return self.nludb.post(‘embedding-index/insert’, {<br /> «indexId»: self.id,<br /> «value»: value<br /> })</p> <p> def embed(self):<br /> return self.nludb.post(‘embedding-index/embed’, {<br /> «indexId»: self.id<br /> })</p> <p> def search(self, query: str):<br /> return self.nludb.post(‘embedding-index/search’, {<br /> «indexId»: self.id,<br /> «query»: query<br /> })</p> <p>class NLUDB:<br /> def __init__(self, api_key: str=None, version: int=1):<br /> self.api_key = api_key<br /> self.endpoint = «https://api.nludb.com/api/v{}».format(version)</p> <p> def post(self, operation: str, payload):<br /> url = «{}/{}».format(self.endpoint, operation)<br /> resp = requests.post(<br /> url,<br /> json=payload,<br /> headers = {«Authorization»: «Bearer {}».format(self.api_key)})<br /> j = resp.json()<br /> if ‘data’ in j:<br /> return j[‘data’]<br /> else:<br /> raise Exception(j[‘reason’])</p> <p> def create_index(self, name: str, model: str, upsert: bool = False):<br /> res = self.post(‘embedding-index/create’, {<br /> «name»: name,<br /> «model»: model,<br /> «upsert»: upsert<br /> })<br /> return NLUDBEmbeddingIndex(nludb=self, name=name, id=res[«id»])</p> <p>class NludbChatbotBase:<br /> def __init__(self, api_key: str, server_name: str, min_confidence: float = 0.4):<br /> self.nludb = NLUDB(api_key)<br /> self.server_name = server_name<br /> self.min_confidence = min_confidence<br /> self.index = self.nludb.create_index(<br /> server_name,<br /> MODELS.QA,<br /> upsert=True<br /> )</p> <p> def get_fact(self, message: str) -> str:<br /> words = re.sub(«\s\s+» , » «, message).split(‘ ‘)<br /> if len(words) == 0:<br /> return None<br /> first = words[0].lower()<br /> if first in [‘fact’, ‘fact:’, ‘learn’, ‘learn:’]:<br /> return » «.join(words[1:])<br /> return None</p> <p> def dispatch(self, message: str, username: str = None, roomname: str = None) -> str:<br /> «»»<br /> Todo: add a «forget» feature<br /> «»»<br /> maybe_fact = self.get_fact(message)<br /> if maybe_fact is not None:<br /> return self.add_fact(maybe_fact, username, roomname)<br /> return self.answer_question(message, username, roomname)</p> <p> def add_fact(self, fact: str, username: str = None, roomname: str = None) -> str:<br /> self.index.insert(fact)<br /> self.index.embed()<br /> return None</p> <p> def answer_question(self, question: str, username: str = None, roomname: str = None) -> Tuple[float, str]:<br /> resp = self.index.search(question)<br /> if ‘hits’ not in resp or len(resp[‘hits’]) == 0:<br /> return None</p> <p> confidence = resp[‘hits’][0][‘score’]<br /> text = resp[‘hits’][0][‘text’]</p> <p> if confidence < self.min_confidence:<br /> return None<br /> return (confidence, text) </p> <p>class NludbChatbot(NludbChatbotBase):<br /> «»»<br /> This class is intended to wrap the core functionality with<br /> any extra personality, error messages, etc.</p> <p> Also can provide a threshold here below which either no fact is<br /> returned or the bot humorously tries to hedge its answer..<br /> «»»<br /> def __init__(self, api_key: str, server_name: str):<br /> super().__init__(api_key, server_name)</p> <p> def dispatch(self, message: str, username: str = None, roomname: str = None) -> str:<br /> message = message.strip()<br /> words = re.split(‘\s+’, message)<br /> if len(message) == 0 or len(words) == 0:<br /> return «»»QQ can learn to answer questions.</p> <p> Learn something:<br /> `qq fact: Bob’s birthday is in January.`<br /> `qq fact: See Jamie about new account setups.`</p> <p> Ask something:<br /> `qq when is Bob’s birthday?`<br /> `qq who do I see about creating a new account?»»»<br /> elif bot.arguments == ‘forget everything you know’:<br /> return «I don’t yet know how to forget»</p> <p> maybe_fact = self.get_fact(message)<br /> if maybe_fact is not None:<br /> if len(maybe_fact.strip()) == 0:<br /> return «Say `qq fact Some fact: goes here` to learn something»<br /> super().dispatch(message, username, roomname)<br /> return «{}… got it!».format(maybe_fact)</p> <p> # It’s a question<br /> resp = super().dispatch(message, username, roomname)<br /> if resp is None:<br /> return «I’m not really sure how to answer that..\n\nAdd new facts with `qq fact: Some fact goes here`.»</p> <p> score, text = resp<br /> return text</p> <p>secret = bot.secrets.read(«nludbapikey»)<br /> if secret is None:<br /> bot.reply(«Please set the `nludbapikey` secret in https://Ab.Bot»)<br /> else:<br /> # if docs is None:<br /> # bot.reply(«I haven’t learned anything yet! Say `qq fact Some fact goes here` to teach me»)<br /> user = bot.from_user[‘UserName’]<br /> #bot.reply(«{}».format(bot.conversation_reference))<br /> chatbot = NludbChatbot(secret, «AbbotDemo1»)<br /> reply = chatbot.dispatch(bot.arguments.strip())<br /> bot.reply(reply)