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

Commit cd526a1

Browse files
committed
fix linting
1 parent c1981f8 commit cd526a1

File tree

1 file changed

+4
-5
lines changed
  • src/codegate/pipeline/codegate_context_retriever

1 file changed

+4
-5
lines changed

src/codegate/pipeline/codegate_context_retriever/codegate.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,5 @@
11
import json
22
import re
3-
import malicious_pypi_dummy
4-
5-
63
import structlog
74
from litellm import ChatCompletionRequest
85

@@ -82,9 +79,11 @@ async def process(
8279
PackageExtractor.extract_packages(snippet.code, snippet.language) # type: ignore
8380
)
8481

85-
logger.info(f"Found {len(snippet_packages)} packages for language {snippet_language} in code snippets.")
82+
logger.info(f"Found {len(snippet_packages)} packages "
83+
"for language {snippet_language} in code snippets.")
8684
# Find bad packages in the snippets
87-
bad_snippet_packages = await storage_engine.search(language=snippet_language, packages=snippet_packages) # type: ignore
85+
bad_snippet_packages = await storage_engine.search(
86+
language=snippet_language, packages=snippet_packages) # type: ignore
8887
logger.info(f"Found {len(bad_snippet_packages)} bad packages in code snippets.")
8988

9089
# Remove code snippets from the user messages and search for bad packages

0 commit comments

Comments
 (0)