feat: 新增文心一言接口

This commit is contained in:
bmy
2024-06-28 21:34:49 +08:00
parent 066ef112ad
commit e82c254dc4
5 changed files with 368 additions and 112 deletions

View File

@@ -256,6 +256,8 @@ class LLM:
原地左转 38 度 [{'func': 'turn','angle': -38}],
蜂鸣器发声 5 秒 [{'func': 'beep', 'time': 5}]
发光或者照亮 5 秒 [{'func': 'light', 'time': 5}]
向右走 30cm照亮 2s [{'func': 'move', 'x': 0, 'y': -0.3}, {'func': 'light', 'time': 2}],
向左移 0.2m, 向后 0.1m [{'func': 'move', 'x': 0, 'y': 0.2},{'func': 'move', 'x': -0.1, 'y': 0}],
'''
self.prompt += '''你无需回复我'''
self.messages = []
@@ -277,7 +279,8 @@ class LLM:
messages=self.messages,
)
self.messages.append(self.resp.to_message())
return self.resp.get_result()
resp = self.resp.get_result().replace(' ', '').replace('\n', '').replace('\t', '')
return resp[7:-3]
class CountRecord:
def __init__(self, stop_count=2) -> None: