refactor: Remove unused O1 model translation function from prompts and client

- Deleted the o1_system_prompt_translate function from prompts.py.
- Removed references to the O1 model translation in client.py to streamline the codebase.

Author: dragos199993

notdiamond/llms/client.py

@@ -42,7 +42,6 @@
 from notdiamond.prompts import (
     _curly_escape,
     inject_system_prompt,
-    o1_system_prompt_translate,
 )
 from notdiamond.types import NDApiKeyValidator
 
@@ -906,8 +905,6 @@ def invoke(
                     messages, best_llm.system_prompt
                 )
 
-            messages = o1_system_prompt_translate(messages, best_llm)
-
             self.call_callbacks("on_model_select", best_llm, best_llm.model)
 
             llm = self._llm_from_config(best_llm, callbacks=self.callbacks)
@@ -1109,8 +1106,6 @@ async def ainvoke(
                     messages, best_llm.system_prompt
                 )
 
-            messages = o1_system_prompt_translate(messages, best_llm)
-
             self.call_callbacks("on_model_select", best_llm, best_llm.model)
 
             llm = self._llm_from_config(best_llm, callbacks=self.callbacks)

notdiamond/prompts.py

@@ -2,9 +2,6 @@
 import re
 from typing import Dict, List
 
-from notdiamond.llms.config import LLMConfig
-from notdiamond.llms.providers import is_o1_model
-
 LOGGER = logging.getLogger(__name__)
 LOGGER.setLevel(logging.INFO)
 
@@ -37,17 +34,3 @@ def _curly_escape(text: str) -> str:
     return re.sub(r"(?<!{){([a-zA-Z])}(?!})", r"{{\1}}", text)
 
 
-def o1_system_prompt_translate(
-    messages: List[Dict[str, str]], llm: LLMConfig
-) -> List[Dict[str, str]]:
-    if is_o1_model(llm):
-        translated_messages = []
-        for msg in messages:
-            if msg["role"] == "system":
-                translated_messages.append(
-                    {"role": "user", "content": msg["content"]}
-                )
-            else:
-                translated_messages.append(msg)
-        return translated_messages
-    return messages