Remove get_tool_name_from_messages

This was superseded by using the client enum.

Related: #830

Author: jhrozek

src/codegate/pipeline/codegate_context_retriever/codegate.py

@@ -4,6 +4,7 @@
 import structlog
 from litellm import ChatCompletionRequest
 
+from codegate.clients.clients import ClientType
 from codegate.pipeline.base import (
     AlertSeverity,
     PipelineContext,
@@ -13,7 +14,7 @@
 from codegate.pipeline.extract_snippets.extract_snippets import extract_snippets
 from codegate.storage.storage_engine import StorageEngine
 from codegate.utils.package_extractor import PackageExtractor
-from codegate.utils.utils import generate_vector_string, get_tool_name_from_messages
+from codegate.utils.utils import generate_vector_string
 
 logger = structlog.get_logger("codegate")
 
@@ -128,14 +129,13 @@ async def process(  # noqa: C901
             new_request = request.copy()
 
             # perform replacement in all the messages starting from this index
-            base_tool = get_tool_name_from_messages(request)
-            if base_tool != "open interpreter":
+            if context.client != ClientType.OPEN_INTERPRETER:
                 for i in range(last_user_idx, len(new_request["messages"])):
                     message = new_request["messages"][i]
                     message_str = str(message["content"])  # type: ignore
                     context_msg = message_str
                     # Add the context to the last user message
-                    if base_tool in ["cline", "kodu"]:
+                    if context.client in [ClientType.CLINE, ClientType.KODU]:
                         match = re.search(r"<task>\s*(.*?)\s*</task>(.*)", message_str, re.DOTALL)
                         if match:
                             task_content = match.group(1)  # Content within <task>...</task>

src/codegate/utils/utils.py

@@ -32,23 +32,3 @@ def generate_vector_string(package) -> str:
     # add description
     vector_str += f" - Package offers this functionality: {package['description']}"
     return vector_str
-
-
-def get_tool_name_from_messages(data):
-    """
-    Identifies the tool name based on the content of the messages.
-
-    Args:
-        request (dict): The request object containing messages.
-        tools (list): A list of tool names to search for.
-
-    Returns:
-        str: The name of the tool found in the messages, or None if no match is found.
-    """
-    tools = ["Cline", "Kodu", "Open Interpreter", "Aider"]
-    for message in data.get("messages", []):
-        message_content = str(message.get("content", ""))
-        for tool in tools:
-            if tool in message_content:
-                return tool.lower()
-    return None