diff --git a/src/codegate/pipeline/codegate_context_retriever/codegate.py b/src/codegate/pipeline/codegate_context_retriever/codegate.py index 208f5daf..ac33b700 100644 --- a/src/codegate/pipeline/codegate_context_retriever/codegate.py +++ b/src/codegate/pipeline/codegate_context_retriever/codegate.py @@ -140,8 +140,10 @@ async def process( task_start, task_content, task_end, rest_of_message = match.groups() # Embed the context into the task block - updated_task_content = f"{task_start}Context: {context_str}\n" + \ - f"Query: {task_content.strip()}{task_end}" + updated_task_content = ( + f"{task_start}Context: {context_str}\n" + + f"Query: {task_content.strip()}{task_end}" + ) # Combine the updated task block with the rest of the message context_msg = updated_task_content + rest_of_message