From 8cc3d6d76c007a599095e3730395356a78fc9551 Mon Sep 17 00:00:00 2001
From: Dominik Peter <dominik.peter@bmsuisse.ch>
Date: Tue, 18 Mar 2025 19:18:57 +0100
Subject: [PATCH] Refactor async pipe execution for improved readability

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 94dd6a8..5b8a112 100644
--- a/main.py
+++ b/main.py
@@ -667,13 +667,13 @@ async def generate_openai_chat_completion(form_data: OpenAIChatCompletionForm):
             status_code=status.HTTP_404_NOT_FOUND,
             detail=f"Pipeline {form_data.model} not found",
         )
+    
     async def execute_pipe(pipe, *args, **kwargs):
         if inspect.isasyncgenfunction(pipe):
             async for res in pipe(*args, **kwargs):
                 yield res
         elif inspect.iscoroutinefunction(pipe):
-            ls = await pipe(*args, **kwargs)
-            for item in ls:
+            for item in await pipe(*args, **kwargs):
                 yield item
         else:
             for item in await run_in_threadpool(pipe, *args, **kwargs):