FilesExpand file tree

main

/

inference_request.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

27 lines (25 loc) · 930 Bytes

main

/

inference_request.py

Copy path

Top

File metadata and controls

• Code
• Blame

27 lines (25 loc) · 930 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

# Copyright (c) Microsoft Corporation. All rights reserved.

def user_message_text(body: dict) -> list[str]:

if isinstance(body.get("input"), str):

return [body["input"]]

messages = body.get("messages")

if messages is None:

messages = body.get("input", [])

texts: list[str] = []

if not isinstance(messages, list):

return texts

for message in messages:

if not isinstance(message, dict) or message.get("role") != "user":

continue

content = message.get("content")

if isinstance(content, str):

texts.append(content)

elif isinstance(content, list):

texts.extend(

part["text"]

for part in content

if isinstance(part, dict)

and part.get("type") in ("text", "input_text")

and isinstance(part.get("text"), str)

)

return texts