Skip to content

Commit 2397720

Browse files
committed
fix lint
1 parent e87e5f9 commit 2397720

File tree

3 files changed

+88
-50
lines changed

3 files changed

+88
-50
lines changed

src/codegate/pipeline/codegate_context_retriever/codegate.py

Lines changed: 8 additions & 4 deletions
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

@@ -79,17 +80,20 @@ async def process(
7980
PackageExtractor.extract_packages(snippet.code, snippet.language) # type: ignore
8081
)
8182

82-
logger.info(f"Found {len(snippet_packages)} packages "
83-
f"for language {snippet_language} in code snippets.")
83+
logger.info(
84+
f"Found {len(snippet_packages)} packages "
85+
f"for language {snippet_language} in code snippets."
86+
)
8487
# Find bad packages in the snippets
8588
bad_snippet_packages = await storage_engine.search(
86-
language=snippet_language, packages=snippet_packages) # type: ignore
89+
language=snippet_language, packages=snippet_packages
90+
) # type: ignore
8791
logger.info(f"Found {len(bad_snippet_packages)} bad packages in code snippets.")
8892

8993
# Remove code snippets from the user messages and search for bad packages
9094
# in the rest of the user query/messsages
9195
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
96+
user_messages = re.sub(r"⋮...*?⋮...\n\n", "", user_messages, flags=re.DOTALL)
9397

9498
# split messages into double newlines, to avoid passing so many content in the search
9599
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)