Do not @ giscus bot (#238)

ultralytics · Sep 16, 2024 · 385dc87 · 385dc87
1 parent 92ca1ac
commit 385dc87
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/utils/first_interaction.py b/utils/first_interaction.py
@@ -43,7 +43,11 @@ def get_completion(messages: list) -> str:
 
     r = requests.post(url, headers=headers, json=data)
     r.raise_for_status()
-    return r.json()["choices"][0]["message"]["content"].strip()
+    content = r.json()["choices"][0]["message"]["content"].strip()
+    remove = [" @giscus[bot]"]
+    for x in remove:
+        content = content.replace(x, "")
+    return content
 
 
 def get_pr_diff(pr_number):