From a48686bde9317ea88b43daeea17eee4d878cd015 Mon Sep 17 00:00:00 2001
From: Trung Dinh <xamarindevelopervietnam@outlook.com>
Date: Sun, 2 Feb 2025 18:01:16 +0700
Subject: [PATCH] Define USER_MESSAGE_ONLY_MODELS in the algo init file

---
 pr_agent/algo/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pr_agent/algo/__init__.py b/pr_agent/algo/__init__.py
index 40d4fc07..6026b67c 100644
--- a/pr_agent/algo/__init__.py
+++ b/pr_agent/algo/__init__.py
@@ -81,3 +81,11 @@ MAX_TOKENS = {
     "watsonx/ibm/granite-34b-code-instruct": 8191,
     "watsonx/mistralai/mistral-large": 32768,
 }
+
+USER_MESSAGE_ONLY_MODELS = [
+    "deepseek/deepseek-reasoner",
+    "o1-mini",
+    "o1-mini-2024-09-12",
+    "o1",
+    "o1-2024-12-17"
+]