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

Commit 8f3f08b

Browse files
committed
fix linting
1 parent 4aa0947 commit 8f3f08b

File tree

1 file changed

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

1 file changed

+3
-5
lines changed

src/codegate/pipeline/codegate_context_retriever/codegate.py

Lines changed: 3 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,10 @@ 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 for {snippet_language} in code snippets.")
8683
# Find bad packages in the snippets
87-
bad_snippet_packages = await storage_engine.search(language=snippet_language, packages=snippet_packages) # type: ignore
84+
bad_snippet_packages = await storage_engine.search(
85+
language=snippet_language, packages=snippet_packages) # type: ignore
8886
logger.info(f"Found {len(bad_snippet_packages)} bad packages in code snippets.")
8987

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

0 commit comments

Comments
 (0)