Skip to content
This repository was archived by the owner on Jun 5, 2025. It is now read-only.

Commit 9eca89f

Browse files
committed
fix lint
1 parent 5309697 commit 9eca89f

File tree

3 files changed

+15
-14
lines changed

3 files changed

+15
-14
lines changed

src/codegate/pipeline/codegate_context_retriever/codegate.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import json
22
import re
3+
34
import structlog
45
from litellm import ChatCompletionRequest
56

@@ -89,7 +90,7 @@ async def process(
8990
# Remove code snippets from the user messages and search for bad packages
9091
# in the rest of the user query/messsages
9192
user_messages = re.sub(r"```.*?```", "", user_message, flags=re.DOTALL)
92-
user_messages = re.sub(r"⋮...*?⋮...\n\n", "", user_messages, flags=re.DOTALL) # regex used in aider
93+
user_messages = re.sub(r"⋮...*?⋮...\n\n", "", user_messages, flags=re.DOTALL)
9394

9495
# split messages into double newlines, to avoid passing so many content in the search
9596
split_messages = user_messages.split("\n\n")

src/codegate/pipeline/extract_snippets/extract_snippets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import os
22
import re
3-
from pygments.lexers import guess_lexer
43
from typing import List, Optional
54

65
import structlog
76
from litellm.types.llms.openai import ChatCompletionRequest
7+
from pygments.lexers import guess_lexer
88

99
from codegate.pipeline.base import CodeSnippet, PipelineContext, PipelineResult, PipelineStep
1010

0 commit comments

Comments
 (0)