From 8aee4d25ed6de2fb6c624edaf8f5c9674e511532 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 12 Sep 2024 14:19:53 -0700 Subject: [PATCH] add ModelSettings.use_system_prompt --- aider/coders/base_coder.py | 14 ++++++++++---- aider/models.py | 17 +++++++++++++++++ 2 files changed, 27 insertions(+), 4 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 823b418a0..b380f08c4 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -966,10 +966,16 @@ class Coder: chunks = ChatChunks() - chunks.system = [ - dict(role="user", content=main_sys), - dict(role="assistant", content="Ok."), - ] + if self.main_model.use_system_prompt: + chunks.system = [ + dict(role="system", content=main_sys), + ] + else: + chunks.system = [ + dict(role="user", content=main_sys), + dict(role="assistant", content="Ok."), + ] + chunks.examples = example_messages self.summarize_end() diff --git a/aider/models.py b/aider/models.py index d513e9ecf..08faf966b 100644 --- a/aider/models.py +++ b/aider/models.py @@ -77,6 +77,7 @@ class ModelSettings: max_tokens: Optional[int] = None cache_control: bool = False caches_by_default: bool = False + use_system_prompt: bool = True # https://platform.openai.com/docs/models/gpt-4-and-gpt-4-turbo @@ -425,6 +426,22 @@ MODEL_SETTINGS = [ lazy=True, reminder="sys", ), + ModelSettings( + "openai/o1-mini", + "whole", + weak_model_name="openai/gpt-4o-mini", + use_repo_map=True, + reminder="user", + use_system_prompt=False, + ), + ModelSettings( + "o1-mini", + "whole", + weak_model_name="gpt-4o-mini", + use_repo_map=True, + reminder="user", + use_system_prompt=False, + ), ]