LLM/pentestagent
1
0
Fork 0
mirror of https://github.com/GH05TCREW/pentestagent.git synced 2026-03-07 06:14:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #35 from selectqoma/fix/gemini-duplicate-assistant-message

Browse Source 
fix: remove duplicate assistant message when content+tool_calls present (fixes Gemini #8)
This commit is contained in:
Masic
2026-02-25 09:32:31 -07:00
committed by GitHub
parent 0ecda6bde4 a6cf68430f
commit 8aca9dd2c6
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pentestagent/agents/crew/orchestrator.py
View File

@@ -184,9 +184,6 @@ class CrewOrchestrator:
# If there are tool calls, the content is "thinking" (reasoning before action)
if response.content:
yield {"phase": "thinking", "content": response.content}
self._messages.append(
{"role": "assistant", "content": response.content}
)
def get_tc_name(tc):
if hasattr(tc, "function"):

4
pentestagent/llm/llm.py
View File

@@ -363,8 +363,8 @@ class LLM:
"claude-sonnet-4-20250514",
"claude-opus-4-20250514",
# Google
"gemini-2.5-pro",
"gemini-2.5-flash",
"gemini/gemini-2.5-pro",
"gemini/gemini-2.5-flash",
# Others via LiteLLM
"ollama/llama3",
"ollama/mixtral",