Merge pull request #286 from stacklok/link-dashboard-secret

Link dashboard in message returned to the chat

Author: aponcedeleonch

src/codegate/pipeline/codegate_context_retriever/codegate.py

@@ -34,7 +34,8 @@ async def get_objects_from_search(
     ) -> list[object]:
         storage_engine = StorageEngine()
         objects = await storage_engine.search(
-            search, distance=0.8, ecosystem=ecosystem, packages=packages)
+            search, distance=0.8, ecosystem=ecosystem, packages=packages
+        )
         return objects
 
     def generate_context_str(self, objects: list[object], context: PipelineContext) -> str:
@@ -109,7 +110,8 @@ async def process(
 
         # Look for matches in vector DB using list of packages as filter
         searched_objects = await self.get_objects_from_search(
-            last_user_message_str, ecosystem, packages)
+            last_user_message_str, ecosystem, packages
+        )
 
         logger.info(
             f"Found {len(searched_objects)} matches in the database",

src/codegate/pipeline/secrets/secrets.py

@@ -349,7 +349,9 @@ async def process_chunk(
             # Create notification chunk
             notification_chunk = self._create_chunk(
                 chunk,
-                f"\n🛡️ Codegate prevented {redacted_count} {secret_text} from being leaked by redacting them.\n\n",  # noqa
+                f"\n🛡️ [Codegate prevented {redacted_count} {secret_text}]"
+                f"(http://localhost:8990/?search=codegate-secrets) from being leaked "
+                f"by redacting them.\n\n",
             )
 
             # Reset the counter

src/codegate/storage/storage_engine.py

@@ -144,8 +144,9 @@ async def search_by_property(self, name: str, properties: List[str]) -> list[obj
             logger.error(f"An error occurred: {str(e)}")
             return []
 
-    async def search(self, query: str, limit=5, distance=0.3,
-                     ecosystem=None, packages=None) -> list[object]:
+    async def search(
+        self, query: str, limit=5, distance=0.3, ecosystem=None, packages=None
+    ) -> list[object]:
         """
         Search the 'Package' collection based on a query string.
 

src/codegate/utils/utils.py

@@ -18,7 +18,7 @@ def generate_vector_string(package) -> str:
         "malicious": "However, this package is found to be malicious and must not be used.",
     }
     vector_str += f" is a {type_map.get(package['type'], 'package of unknown type')}. "
-    package_name = quote(package["name"], safe='')
+    package_name = quote(package["name"], safe="")
     package_url = f"https://www.insight.stacklok.com/report/{package['type']}/{package_name}"
 
     # Add extra status