diff --git a/pentestagent/agents/crew/orchestrator.py b/pentestagent/agents/crew/orchestrator.py index c7d9733..c8edd9d 100644 --- a/pentestagent/agents/crew/orchestrator.py +++ b/pentestagent/agents/crew/orchestrator.py @@ -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"): diff --git a/pentestagent/llm/llm.py b/pentestagent/llm/llm.py index e5a79f5..b2a9c9f 100644 --- a/pentestagent/llm/llm.py +++ b/pentestagent/llm/llm.py @@ -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",