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)