From a3706c79f80a95514d3e8500d48525bcc5ee1986 Mon Sep 17 00:00:00 2001
From: bobo <bobo.yang@merico.dev>
Date: Tue, 28 May 2024 18:10:31 +0800
Subject: [PATCH] Refactor code to remove model parameter from @chat_json
 decorator

---
 merico/github/code_task_summary/command.py | 2 +-
 merico/github/new_branch/command.py        | 2 +-
 merico/github/new_pr/command.py            | 2 +-
 merico/github/update_pr/command.py         | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/merico/github/code_task_summary/command.py b/merico/github/code_task_summary/command.py
index 895a888..37906ca 100644
--- a/merico/github/code_task_summary/command.py
+++ b/merico/github/code_task_summary/command.py
@@ -31,7 +31,7 @@ PROMPT = (
 )
 
 
-@chat_json(prompt=PROMPT, model="gpt-4-1106-preview")
+@chat_json(prompt=PROMPT)
 def generate_code_task_summary(issue_body):
     pass
 
diff --git a/merico/github/new_branch/command.py b/merico/github/new_branch/command.py
index ea9f084..c456589 100644
--- a/merico/github/new_branch/command.py
+++ b/merico/github/new_branch/command.py
@@ -26,7 +26,7 @@ PROMPT = (
 )
 
 
-@chat_json(prompt=PROMPT, model="gpt-4-1106-preview")
+@chat_json(prompt=PROMPT)
 def generate_branch_name(task):
     pass
 
diff --git a/merico/github/new_pr/command.py b/merico/github/new_pr/command.py
index 42b2f62..c66a55c 100644
--- a/merico/github/new_pr/command.py
+++ b/merico/github/new_pr/command.py
@@ -39,7 +39,7 @@ PROMPT = (
 )
 
 
-@chat_json(prompt=PROMPT, model="gpt-4-1106-preview")
+@chat_json(prompt=PROMPT)
 def generate_pr_content_llm(issue, commit_message, user_input):
     pass
 
diff --git a/merico/github/update_pr/command.py b/merico/github/update_pr/command.py
index f339f58..51ea4ad 100644
--- a/merico/github/update_pr/command.py
+++ b/merico/github/update_pr/command.py
@@ -41,7 +41,7 @@ PROMPT = (
 )
 
 
-@chat_json(prompt=PROMPT, model="gpt-4-1106-preview")
+@chat_json(prompt=PROMPT)
 def generate_pr_content_llm(issue, commit_messages):
     pass