41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import requests
|
|
|
|
class Similar():
|
|
def __init__(self):
|
|
self.url = 'http://127.0.0.1:6007'
|
|
|
|
def post(self,url,json_p):
|
|
headers = {
|
|
"Authorization": "Bearer ACCESS_TOKEN"
|
|
}
|
|
response = requests.post(url,json=json_p,headers=headers)
|
|
return response.text
|
|
|
|
def similar(self,texts1,texts2):
|
|
url = self.url + '/v1/embedding'
|
|
json_p = {
|
|
"sentences1":texts1,
|
|
"sentences2":texts2,
|
|
}
|
|
return self.post(url,json_p)
|
|
|
|
def load_know(self,know_id,contents,drop_dup,is_cover = False):
|
|
url = self.url + '/v1/load_know'
|
|
json_p = {
|
|
"know_id": know_id,
|
|
"contents": contents,
|
|
"drop_dup": drop_dup,
|
|
"is_cover": is_cover,
|
|
}
|
|
return self.post(url,json_p)
|
|
|
|
def know_sim(self,query,know_id,top_k = 10):
|
|
url = self.url + '/v1/know_sim'
|
|
json_p = {
|
|
"query": query,
|
|
"know_id": know_id,
|
|
"top_k": top_k
|
|
}
|
|
return self.post(url,json_p)
|
|
|