Skip to content

Commit 494bbcb

Browse files
committed
[Fix] Moderate multiple images in one prompt
1 parent cd7dbf9 commit 494bbcb

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

edenai_apis/llmengine/utils/moderation.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,11 @@ async def standard_moderation(*args, **kwargs):
6161

6262
for message in messages:
6363
if "content" in message:
64-
tasks.append(moderate_content(headers, message["content"]))
64+
if isinstance(message["content"], list):
65+
for content in message["content"]:
66+
tasks.append(moderate_content(headers, [content]))
67+
else:
68+
tasks.append(moderate_content(headers, message["content"]))
6569

6670
async with aiohttp.ClientSession() as session:
6771
await asyncio.gather(*tasks)

0 commit comments

Comments
 (0)