fix merge

owenzhw · Mar 9, 2024 · 8eeafd3 · 8eeafd3
1 parent 5d2309c
commit 8eeafd3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/multiagents/llms/feedback_openai.py b/multiagents/llms/feedback_openai.py
@@ -204,6 +204,11 @@ def extract_rules_from_feedback(self, messages, output, preferred_output):
 
         return valid_rules
 
+    def judge_feedback(self, feedback):
+        judge_messages = [{'role': 'system', 'content': JUDGE_FEEDBACK_PROMPT}, {'role': 'user', 'content': feedback}]
+        reply = call_openai(judge_messages)
+        return "yes" in reply.lower()
+
     def user_input(self, placeholder):
         if USERINPUTREADFROMFILE:
             print('='*10 + 'USER FEEDBACK: ' + placeholder, flush=True)  # 参与前端逻辑，不能修改