class HanlpClient(object):
def __init__(self, host='localhost', port=10052):
logger.info("connect to hanlp provider %s %d ..."%(host, port))
self.client=ServiceClient(nlp_service, 'NlpProcsStub', host, port)
def get_deps(self, text):
request = nlp_messages.NlTexts(texts=[nlp_messages.NlText(text=text)])
response = self.client.GetDependencyGraph(request)
return response
def disp_deps(self, text):
"""
$ python -m sagas.bots.hanlp_client disp_deps '苹果电脑可以运行开源阿尔法狗代码吗'
:param text:
:return:
"""
response=self.get_deps(text)
print('✔', response.summary)
if 'zh_SBV|text' in response.coreGraph:
print('♟ sbv', response.coreGraph['zh_SBV|text'])
if 'zh_VOB|head' in response.coreGraph:
print('\t► vob.head', response.coreGraph['zh_VOB|head'])
if 'zh_VOB|text' in response.coreGraph:
print('\t\t✜ vob', response.coreGraph['zh_VOB|text'])
for k,v in response.coreGraph.items():
print(k,v)
def put_deps(self, sents, props):
response = self.get_deps(sents)
for k, v in response.coreGraph.items():
props[k]=v
def extract(self, text):
request = nlp_messages.NlTokenizerRequest(text=nlp_messages.NlText(text=text))
response = self.client.EntityExtractor(request)
return response
class HanlpDelegator(object):
def __init__(self):
self.client = ServiceClient(nlp_service, 'NlpProcsStub', 'localhost',
10052)
async def handle(self, req: BlueInteract):
cnt = BotMessage()
cnt.ParseFromString(req.body)
result = self.extract(cnt.message)
response = result.SerializeToString()
return response
def tokenize(self, text):
request = nlp_messages.NlTokenizerRequest(text=nlp_messages.NlText(
text=text))
response = self.client.Tokenizer(request)
return response
def extract(self, text):
request = nlp_messages.NlTokenizerRequest(text=nlp_messages.NlText(
text=text))
response = self.client.EntityExtractor(request)
return response