mblanke/StrikePackageGPT
1
0
Fork 0
mirror of https://github.com/mblanke/StrikePackageGPT.git synced 2026-03-01 14:20:21 -05:00
Code Issues Packages Projects Releases Wiki Activity

Final code quality improvements: fix error handling and memory management

Browse Source 
Co-authored-by: mblanke <9078342+mblanke@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2025-12-03 13:00:34 +00:00
parent d0aadefad9
commit 4e3cf99e04
4 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
services/hackgpt-api/app/llm_help.py
View File

@@ -11,7 +11,10 @@ import json
# Store conversation history per session
# Note: In production, use Redis or similar with TTL for scalability
# This simple in-memory dict will grow unbounded - implement cleanup as needed
conversation_contexts: Dict[str, List[Dict[str, str]]] = {}
MAX_SESSIONS = 1000 # Limit number of concurrent sessions
async def chat_completion(
@@ -87,6 +90,11 @@ Always emphasize ethical hacking practices and legal considerations."""
# Store in conversation history
if session_id:
if session_id not in conversation_contexts:
# Cleanup old sessions if limit reached
if len(conversation_contexts) >= MAX_SESSIONS:
# Remove oldest session (simple FIFO)
oldest_session = next(iter(conversation_contexts))
del conversation_contexts[oldest_session]
conversation_contexts[session_id] = []
conversation_contexts[session_id].append({"role": "user", "content": message})
conversation_contexts[session_id].append({"role": "assistant", "content": assistant_message})