Method run

openosint/agent.py:501–559

Execute one agent turn. Parameters ---------- prompt: User message or OSINT target description. on_tool_call: Optional async callback invoked before each tool execution. Signature: ``async def on_tool_call(name: str, input

(
        self,
        prompt: str,
        on_tool_call: Any = None,
    )

        self.history = []

    async def run(
        self,
        prompt: str,
        on_tool_call: Any = None,
    ) -> AgentResponse:
        """
        Execute one agent turn.

        Parameters
        ----------
        prompt:
            User message or OSINT target description.
        on_tool_call:
            Optional async callback invoked before each tool execution.
            Signature: ``async def on_tool_call(name: str, input: dict) -> None``

        Returns
        -------
        AgentResponse
            Final text response and list of tool calls made.
        """
        self.history.append({"role": "user", "content": prompt})
        ctx = _AgentRunContext(
            messages=list(self.history),
            tool_calls=[],
            on_tool_call=on_tool_call,
        )
        try:
            while True:
                response = self.client.messages.create(
                    model=self.model,
                    max_tokens=_MAX_TOKENS,
                    system=SYSTEM_PROMPT,
                    tools=TOOL_DEFINITIONS,  # type: ignore[arg-type]
                    messages=ctx.messages,  # type: ignore[arg-type]
                )
                if response.stop_reason == "end_turn":
                    text = _extract_first_text(response.content)
                    self.history.append({"role": "assistant", "content": response.content})
                    return AgentResponse(content=text, tool_calls=ctx.tool_calls)
                if response.stop_reason == "tool_use":
                    ctx.messages.append({"role": "assistant", "content": response.content})
                    await _process_tool_turn(ctx, response.content)
                else:
                    break
        except anthropic.AuthenticationError:
            return AgentResponse(
                content="",
                error="Invalid API key. Run 'openosint config' to update it.",
            )
        except anthropic.APIConnectionError:
            return AgentResponse(
                content="",
                error="Cannot reach the Anthropic API. Check your internet connection.",
            )
        except Exception as exc:
            logger.exception("Unexpected error in Anthropic agent loop.")
            return AgentResponse(content="", error=str(exc))