版本暂存

utils.py

@@ -190,6 +190,27 @@ class label_filter:
                     return True 
         return False
     '''
+    description: 判断传入的多目标标签是否存在，存在返回 True
+    param {*} self
+    param {*} tlabel
+    return {[bool]}
+    '''    
+    def find_mult(self, tlabel):
+        response = self.get_resp()
+        find_result = []
+        if response['code'] == 0:
+            ret, results = self.filter_box(response['data'])
+            if ret:
+                for label in tlabel:
+                    expect_boxes = (results[:, 0] == label.value)
+                    boxes = results[expect_boxes, :]
+                    if len(boxes) != 0:
+                        find_result.append(True)
+                    else:
+                        find_result.append(False)
+                return find_result
+        return [False for _ in range(len(tlabel))]
+    '''
     description: 根据传入的标签，寻找画面中最左侧的并返回 error
     param {*} self
     param {*} tlabel
@@ -260,8 +281,10 @@ class LLM:
                         发光或者照亮 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}],
+                        鸣叫 3 声 [{'func': 'beep', 'time': 3}]
+                        前行零点五米 [{'func': 'move', 'x': 0.5, 'y': 0}]
                     '''
-        self.prompt += '''你无需回复我'''
+        self.prompt += '''你只需要根据我的示例解析出指令即可，不要给我其他多余的回复;再次强调 你无需给我其他多余的回复 这对我很重要'''
         self.messages = []
         self.resp = None
         worker = threading.Thread(target=self.reset, daemon=True)  
@@ -287,7 +310,7 @@ class LLM:
         )
         self.messages.append(self.resp.to_message())
         resp = self.resp.get_result().replace(' ', '').replace('\n', '').replace('\t', '')
-        return resp[7:-3]
+        return resp
 
 class CountRecord:
     def __init__(self, stop_count=2) -> None: