def search(server_name, url_suff, kbase_id, req_payload, org_id, token, search_size=1): full_addr = server_name + url_suff["search"] full_addr = full_addr.format(knowledgebaseId=kbase_id) req = HTTPRequest(full_addr, "POST") for key in req_payload.keys(): req.payload_append(key, req_payload[key]) req.payload_append("pageSize", search_size) req.payload_append("pageNumber", 1) req.payload_append("languageCode", "en-US") req.payload_append("documentType", "Faq") req.add_header("Content-Type", "application/json") req.add_header("organizationid", org_id) req.add_header("token", token) req.add_header("cache-control", "no-cache") response = (req.post()) print("Response status: ", response.status_code) resp_json = response.json() if resp_json.get("results"): results = resp_json["results"] else: results = [] return results
def create_kbase(server_name, url_suff, payload_info, org_id, token, kbase_responses={}): # Check if # of databases exceeds limit (5) if len(kbase_responses.items()) == 5: return {}, full_addr = server_name + url_suff["create_kbase"] req = HTTPRequest(full_addr, "POST") for key in payload_info.keys(): req.payload_append(key, payload_info[key]) #'Content-Type': "application/json", #'organizationid': "3ae6bd8b-23b6-47c7-a9a0-8dc56833ca18", #'token': "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJvcmdJZCI6IjNhZTZiZDhiLTIzYjYtNDdjNy1hOWEwLThkYzU2ODMzY2ExOCIsImV4cCI6MTU3MTUxNjYyOCwiaWF0IjoxNTcxNTEzMDI4fQ.S1mQ4sdeSaPhTCwUodVOb6QW3yOzTnQKOunSU732SSI", #'cache-control': "no-cache", req.add_header("Content-Type", "application/json") req.add_header("organizationid", org_id) req.add_header("token", token) req.add_header("cache-control", "no-cache") response = req.post() print(response) print("response: ", response.json())
def create_kbase(server_name, url_suff, payload_info, kbase_responses={}): # Check if # of databases exceeds limit (5) if len(kbase_responses.items()) == 5: return {}, full_addr = server_name + url_suff["create_kbase"] req = HTTPRequest(full_addr, "POST") for key in payload_info.keys(): req.payload_append(key, payload_info[key]) response = req.post() resp_id = response.json().get("id") kbase_responses['status_code'] = response.status_code if resp_id: kbase_responses[resp_id] = (response["dateCreated"], response["dateModified"], response["selfUri"]) else: return kbase_responses return kbase_responses
def upload_doc(server_name, url_suff, kbase_id, lang_code, payload, org_id, token, categories={}): full_addr = server_name + url_suff["upload_doc"] full_addr = full_addr.format(knowledgebaseId=kbase_id, languageCode=lang_code) req = HTTPRequest(full_addr, "POST") req.payload_append("type", "faq") key = "faq" val = {"question": payload["question"], "answer": payload["answer"], "alternatives": []} req.payload_append(key, val) key = "categories" val = [] for elem1, elem2 in categories.items(): val.append({elem1: elem2}) req.payload_append(key, val) req.payload_append("externalUrl", "") req.add_header("Content-Type", "application/json") req.add_header("organizationid", org_id) req.add_header("token", token) req.add_header("cache-control", "no-cache") response = req.post() print("RESPONSE CODE: ",response.status_code) print("RESPONSE JSON: ",response.json()) return response.status_code
class PopulateProject(): # URL of API API_URL = 'http://*****:*****@%s.com" % (self.randomword(randint(5,15)), self.randomword(randint(4,8))) } self.req.post('PersonsServlet', person) def randomword(self, length): return ''.join(random.choice(string.lowercase) for i in range(length))
def train_model(server_name, url_suff, kbase_id, lang_code, org_id, token): full_addr = server_name + url_suff["train_model"] full_addr = full_addr.format(knowledgebaseId=kbase_id, languageCode=lang_code) req = HTTPRequest(full_addr, "POST") req.add_header("Content-Type", "application/json") req.add_header("organizationid", org_id) req.add_header("token", token) req.add_header("cache-control", "no-cache") response = req.post() response_dict = response.json() print(response.status_code) training_id = response_dict["id"] return (training_id)
def upload_doc(server_name, url_suff, kbase_id, lang_code, payload, categories={}): full_addr = server_name + url_suff["upload_doc"] full_addr = full_addr.format(knowledgebaseID=kbase_id, languageCode=lang_code) req = HTTPRequest(full_addr, "POST") req.payload_append("type", "Faq") key = "faq" val = {"question": payload["question"], "answer": payload["answer"], "alternatives": payload["alternatives"]} req.payload_append(key, val) key = "categories" val = [] for elem1, elem2 in categories.items(): val.append({elem1: elem2}) req.payload_append(key, val) req.payload_append("externalUrl", "") response = req.post() return response.status_code
def create_ctg(server_name, url_suff, payload_info_ctg, kbase_id, lang_code): full_addr = server_name + url_suff["create_ctg"] full_addr = full_addr.format(knowledgebaseId=kbase_id, languageCode=lang_code) req = HTTPRequest(full_addr, "POST") for key in payload_info_ctg.keys(): req.payload_append(key, payload_info_ctg[key]) response = req.post() response_dict = response.json() resp_kbase_id = response_dict.get("knowledgeBase", {}).get("id") ctg = response_dict.get("id") if resp_kbase_id in kbase_categories.keys(): kbase_responses[resp_kbase_id].append(ctg) else: kbase_responses[resp_kbase_id] = [ctg] return response.status_code