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
