Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: refactor LLM model selection and attack surface analysis #233

Open
wants to merge 21 commits into
base: release/2.2.0
Choose a base branch
from
Open

feat: refactor LLM model selection and attack surface analysis #233

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
284d83b
refactor: replace GPT with LLM for vulnerability reporting
psyray Nov 10, 2024
3ad4ea6
feat: enhance LLM model selection and attack surface analysis
psyray Nov 11, 2024
82fccad
refactor(logging): improve vulnerability logging details
psyray Nov 11, 2024
b170a31
feat(ui): enhance LLM toolkit UI and refactor model management
psyray Nov 11, 2024
d6a1b4b
feat(llm): convert LLM markdown response to HTML sanitize it
psyray Nov 11, 2024
678e317
feat: enhance attack surface analysis with model selection and deleti…
psyray Nov 11, 2024
916ed10
feat: enhance markdown rendering and update UI settings
psyray Nov 11, 2024
7e87530
refactor: update LLM vulnerability report generation and storage
psyray Nov 12, 2024
e7e56c1
fix: remove unused imports
psyray Nov 12, 2024
5390b37
fix: enable section response generation
psyray Nov 12, 2024
20823c0
refactor: update fixtures and permissions, remove unused data
psyray Nov 13, 2024
0d3f6ba
feat: enhance reference handling
psyray Nov 13, 2024
f5f37f4
fix: task reference conversion
psyray Nov 13, 2024
ac29d7f
fix: update model selection logic
psyray Nov 14, 2024
c5d0496
feat: enhance model management and download functionality in LLM Toolkit
psyray Nov 14, 2024
a8bd492
feat: integrate WebSocket support for model operations
psyray Nov 17, 2024
638ea94
test: enhance OllamaManager and LLM tests with additional mock setups
psyray Nov 17, 2024
79b52e5
refactor: remove cancel download feature
psyray Nov 17, 2024
d9d7af4
feat: add model name to progress bar popup
psyray Nov 17, 2024
5d489f7
fix: update model selection API endpoint
psyray Nov 17, 2024
3809741
feat: enhance model URL handling
psyray Nov 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix: enable section response generation 
Uncommented the code responsible for generating technical, impact, and remediation sections in the LLMVulnerabilityReportGenerator class.
  • Loading branch information
@psyray
psyray committed Nov 12, 2024
commit 5390b374f3e6e775a04bbb6108133753e1cf11eb
6 changes: 3 additions & 3 deletions web/reNgine/llm/llm.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -101,9 +101,9 @@
context = vulnerability_prompt['context']

# Generate each section separately
#technical = self._get_section_response(validated_input, context + vulnerability_prompt['technical'])
#impact = self._get_section_response(validated_input, context + vulnerability_prompt['impact'])
#remediation = self._get_section_response(validated_input, context + vulnerability_prompt['remediation'])
technical = self._get_section_response(validated_input, context + vulnerability_prompt['technical'])
Fixed Show fixed Hide fixed
impact = self._get_section_response(validated_input, context + vulnerability_prompt['impact'])
Fixed Show fixed Hide fixed
remediation = self._get_section_response(validated_input, context + vulnerability_prompt['remediation'])
Fixed Show fixed Hide fixed
technical = ""
impact = ""
remediation = ""
Expand Down
Loading