GreyDGL / PentestGPT

A GPT-empowered penetration testing tool
MIT License
7.12k stars 864 forks source link

[Feature] Consider to add Google Search #7

Open GreyDGL opened 1 year ago

GreyDGL commented 1 year ago

Create a new branch to work on this.

GreyDGL commented 1 year ago

General design:

  1. Allow user to google search in the process of testing.
  2. Google search embedded reasoning: use google search to improve reasoning process.
    • This part is very tricky, because the keywords are not likely to be found.
    • Some functions that may help in practice:
      1. search based on the test result: check if the machine walkthrough is available online (need a toggle).
      2. check what can do for a specific service.
  3. Google search embedded test generation: check the ways to scan a specific item (images, files, etc.)
GreyDGL commented 1 year ago

Basic google search function added. Consider to add parsers for detailed usage.

SATUNIX commented 1 year ago

Would it be beneficial to add a vector DB search to gather relevant and readily available data to feed into the LLM of choice? This would save unique Google APIs and web scraping. Already working on the vectorDB (ChromaDB)with someone if that helps:) Could put the implementation into a PR. OR There could be additional discuss options incuding browser output and the prompt for the LLM to search and expand for itself.... idk though...

GreyDGL commented 1 year ago

@SATUNIX Yes the vector DB implementation is on the dev plan. Will integrate into the current implementation, but need some further tests to ensure its performance.

GreyDGL commented 7 months ago

Adding the basics of Google search feature. Will do further implementations to complete the logic.