Fixed indentation issue (#382)

Zipstack · Jun 3, 2024 · dabda23 · dabda23
1 parent 828162c
commit dabda23
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/prompt-service/src/unstract/prompt_service/main.py b/prompt-service/src/unstract/prompt_service/main.py
@@ -378,19 +378,21 @@ def prompt_processor() -> Any:
                 "Retrieving context from adapter",
             )
 
-        retrieval_strategy = output.get(PSKeys.RETRIEVAL_STRATEGY)
+            retrieval_strategy = output.get(PSKeys.RETRIEVAL_STRATEGY)
 
-        if retrieval_strategy in {PSKeys.SIMPLE, PSKeys.SUBQUESTION}:
-            answer, context = run_retrieval(
-                tool_settings=tool_settings,
-                output=output,
-                doc_id=doc_id,
-                llm=llm,
-                vector_index=vector_index,
-                retrieval_type=retrieval_strategy,
-            )
-        else:
-            app.logger.info("Invalid retrieval strategy passed: %s", retrieval_strategy)
+            if retrieval_strategy in {PSKeys.SIMPLE, PSKeys.SUBQUESTION}:
+                answer, context = run_retrieval(
+                    tool_settings=tool_settings,
+                    output=output,
+                    doc_id=doc_id,
+                    llm=llm,
+                    vector_index=vector_index,
+                    retrieval_type=retrieval_strategy,
+                )
+            else:
+                app.logger.info(
+                    "Invalid retrieval strategy passed: %s", retrieval_strategy
+                )
 
             _publish_log(
                 log_events_id,