Skip to content

Commit ef9fe6f

Browse files
.
1 parent ac3ce00 commit ef9fe6f

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

sentry_sdk/ai/_openai_completions_api.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,20 @@
55
ChatCompletionMessageParam,
66
ChatCompletionSystemMessageParam,
77
)
8-
from typing import Iterable, Union
8+
from typing import Iterable
9+
10+
11+
def _is_system_instruction(message: "ChatCompletionMessageParam"):
12+
return isinstance(message, dict) and message.get("role") == "system"
913

1014

1115
def _get_system_instructions(
12-
messages: "Iterable[Union[ChatCompletionMessageParam, str]]",
16+
messages: "Iterable[ChatCompletionMessageParam]",
1317
) -> "list[ChatCompletionSystemMessageParam]":
14-
system_messages = []
18+
system_instructions = []
1519

1620
for message in messages:
17-
if isinstance(message, dict) and message.get("role") == "system":
18-
system_messages.append(message)
21+
if _is_system_instruction(message):
22+
system_instructions.append(message)
1923

20-
return system_messages
24+
return system_instructions

0 commit comments

Comments
 (0)