PentestGPT
PentestGPT copied to clipboard
[Feature] Consider to add Google Search
Create a new branch to work on this.
General design:
- Allow user to google search in the process of testing.
- 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:
- search based on the test result: check if the machine walkthrough is available online (need a toggle).
- check what can do for a specific service.
- Google search embedded test generation: check the ways to scan a specific item (images, files, etc.)
Basic google search function added. Consider to add parsers for detailed usage.
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...
@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.
Adding the basics of Google search feature. Will do further implementations to complete the logic.