style fix

Cinnamon · Apr 3, 2024 · 6d05a7e · 6d05a7e
1 parent c1f480e
commit 6d05a7e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/libs/ktem/ktem/reasoning/simple.py b/libs/ktem/ktem/reasoning/simple.py
@@ -7,7 +7,6 @@
 
 import tiktoken
 from ktem.components import llms
-from ktem.reasoning.base import BaseReasoning
 from theflow.settings import settings as flowsettings
 
 from kotaemon.base import (
@@ -296,12 +295,14 @@ async def run(  # type: ignore
 
         return answer
 
+
 def extract_evidence_images(self, evidence: str):
-        """Util function to extract and isolate images from context/evidence"""
-        image_pattern = r"src='(data:image\/[^;]+;base64[^']+)'"
-        matches = re.findall(image_pattern, evidence)
-        context = re.sub(image_pattern, "", evidence)
-        return context, matches
+    """Util function to extract and isolate images from context/evidence"""
+    image_pattern = r"src='(data:image\/[^;]+;base64[^']+)'"
+    matches = re.findall(image_pattern, evidence)
+    context = re.sub(image_pattern, "", evidence)
+    return context, matches
+
 
 class RewriteQuestionPipeline(BaseComponent):
     """Rewrite user question