fcking black

ls1intum · Nov 27, 2024 · 75fbaf2 · 75fbaf2
1 parent 57d0262
commit 75fbaf2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/app/pipeline/chat/lecture_chat_pipeline.py b/app/pipeline/chat/lecture_chat_pipeline.py
@@ -60,7 +60,12 @@ class LectureChatPipeline(Pipeline):
     prompt: ChatPromptTemplate
     callback: LectureChatCallback
 
-    def __init__(self, callback: LectureChatCallback, dto: LectureChatPipelineExecutionDTO, variant: str):
+    def __init__(
+        self,
+        callback: LectureChatCallback,
+        dto: LectureChatPipelineExecutionDTO,
+        variant: str,
+    ):
         super().__init__(implementation_id="lecture_chat_pipeline")
         # Set the langchain chat model
         request_handler = CapabilityRequestHandler(

diff --git a/app/web/routers/pipelines.py b/app/web/routers/pipelines.py
@@ -153,7 +153,9 @@ def run_lecture_chat_pipeline_worker(dto, variant):
         )
         match variant:
             case "default" | "lecture_chat_pipeline_reference_impl":
-                pipeline = LectureChatPipeline(callback=callback, dto=dto, variant=variant)
+                pipeline = LectureChatPipeline(
+                    callback=callback, dto=dto, variant=variant
+                )
             case _:
                 raise ValueError(f"Unknown variant: {variant}")
     except Exception as e: