yv1ing/MollyAudit
1
0
Fork 0
mirror of https://github.com/yv1ing/MollyAudit.git synced 2025-09-16 14:55:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add context compression mechanism

Browse Source
This commit is contained in:
yv1ing
2025-02-09 12:05:41 +08:00
parent 32e61b8d24
commit 6a0cab65d1
4 changed files with 76 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -6,7 +6,7 @@ from audit import Audit
warnings.simplefilter('ignore', FutureWarning)
os.environ['OPENAI_API_BASE'] = 'https://yunwu.ai/v1'
os.environ['OPENAI_API_KEY'] = 'sk-SQhmr2wNQa2BpohUrxgJOFIDY9ODSxUkLQLWWlPD9qDNVsN1'
os.environ['OPENAI_API_KEY'] = 'sk-zpkHfWT0Zhvzc79lX11WS4dEyg5CkQ3RdZOSNDoLADaitfVM'
def result_callback(result):
@@ -14,8 +14,8 @@ def result_callback(result):
if __name__ == '__main__':
src_root = r'C:\Users\yvling\Desktop\PHP-Vuln'
language = 'php'
src_root = r'C:\Users\yvling\Desktop\JavaSecLab'
language = 'java'
audit = Audit()
audit.load_source_files(src_root, language)