启动demo
按照教程配置好环境即可进行聊天。
实现Agent
按照教程复制代码即可。
自定义插件
自定义的API,注释千万要符合规范。
自己实现一个简单的问好插件如下:
class HelloSpeaker(BaseAction):
language_options = {
'english' : 'Hello',
'chinese' : 'Nihao',
'japanese' : 'konnijiwa'
}
def __init__(self, language = 'english'):
super().__init__()
if language not in self.language_options:
raise ValueError(f"{language} not supported")
else:
self.l = language
@tool_api
def say_hello(self, language: str) -> str:
""" say hello in different languages
Args:
language (:class:`str`): required language in English
Returns:
:class:`str`: result
"""
r = language.lower()
return self.language_options.get(r, "hello")
完整代码见 Lagent