fix naming of tool call const

Author: shellmayr

sentry_sdk/ai/utils.py

@@ -14,14 +14,14 @@ class GEN_AI_ALLOWED_MESSAGE_ROLES:
     SYSTEM = "system"
     USER = "user"
     ASSISTANT = "assistant"
-    TOOL_CALL = "tool"
+    TOOL = "tool"
 
 
 GEN_AI_MESSAGE_ROLE_REVERSE_MAPPING = {
     GEN_AI_ALLOWED_MESSAGE_ROLES.SYSTEM: ["system"],
     GEN_AI_ALLOWED_MESSAGE_ROLES.USER: ["user", "human"],
     GEN_AI_ALLOWED_MESSAGE_ROLES.ASSISTANT: ["assistant", "ai"],
-    GEN_AI_ALLOWED_MESSAGE_ROLES.TOOL_CALL: ["tool", "tool_call"],
+    GEN_AI_ALLOWED_MESSAGE_ROLES.TOOL: ["tool", "tool_call"],
 }
 
 GEN_AI_MESSAGE_ROLE_MAPPING = {}

sentry_sdk/integrations/openai_agents/utils.py

@@ -130,7 +130,7 @@ def _set_input_data(span, get_response_kwargs):
             elif message.get("type") == "function_call_output":
                 request_messages.append(
                     {
-                        "role": GEN_AI_ALLOWED_MESSAGE_ROLES.TOOL_CALL,
+                        "role": GEN_AI_ALLOWED_MESSAGE_ROLES.TOOL,
                         "content": [message],
                     }
                 )