File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
1115def _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
You can’t perform that action at this time.
0 commit comments