While in testing, only include aider if prompt specifically mentions it

aledanee · Oct 5, 2024 · efa8358 · efa8358
1 parent fe85ecf
commit efa8358
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/openai_server/agent_prompting.py b/openai_server/agent_prompting.py
@@ -662,7 +662,8 @@ def get_full_system_prompt(agent_code_writer_system_message, agent_system_site_p
     image_generation_helper = get_image_generation_helper()
     audio_transcription_helper = get_audio_transcription_helper()
     download_one_web_image_helper = get_download_one_web_image_helper()
-    aider_coder_helper = get_aider_coder_helper(base_url, api_key, model, autogen_timeout)
+    aider_coder_helper = get_aider_coder_helper(base_url, api_key, model,
+                                                autogen_timeout) if 'aider' in query or 'Aider' in query else ''
 
     chat_doc_query, internal_file_names = get_chat_doc_context(text_context_list, image_file,
                                                                temp_dir,
@@ -682,5 +683,14 @@ def get_full_system_prompt(agent_code_writer_system_message, agent_system_site_p
         "You have to prioritize these tools for the relevant tasks before using other tools or methods. \n"
     )
 
-    system_message = agent_code_writer_system_message + ask_question_about_image_helper + mermaid_renderer_helper + image_generation_helper + audio_transcription_helper + download_one_web_image_helper + agent_tools_note + chat_doc_query
+    system_message = agent_code_writer_system_message + \
+                     ask_question_about_image_helper + \
+                     mermaid_renderer_helper + \
+                     image_generation_helper + \
+                     audio_transcription_helper + \
+                     download_one_web_image_helper + \
+                     aider_coder_helper + \
+                     agent_tools_note + \
+                     chat_doc_query
+
     return system_message, internal_file_names, chat_doc_query, ask_question_about_image_helper, mermaid_renderer_helper
diff --git a/src/version.py b/src/version.py
@@ -1 +1 @@
-__version__ = "54be435dfe8b9b440bfd716d62e96b65719731a3"
+__version__ = "fe85ecf19049bcbdd96c85b7a9de266db64d2c88"
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		__version__ = "54be435dfe8b9b440bfd716d62e96b65719731a3"
		__version__ = "fe85ecf19049bcbdd96c85b7a9de266db64d2c88"