Method _query

codeclash/agents/mini_anthropic_model.py:167–182

(self, messages: list[dict], **kwargs)

        ]

    def _query(self, messages: list[dict], **kwargs):
        assert messages[0]["role"] == "system"
        api_messages = self._set_cache_control_on_last_message(
            [{"role": m["role"], "content": self._strip_display_text(m["content"])} for m in messages[1:]]
        )
        extra_model_kwargs = self.config.model_kwargs | kwargs
        if self.config.drop_none_model_kwargs:
            extra_model_kwargs = {k: v for k, v in extra_model_kwargs.items() if v is not None}
        return self.client.messages.create(
            model=self.config.model_name,
            max_tokens=self.config.max_tokens,
            system=messages[0]["content"],
            messages=api_messages,
            tools=[ANTHROPIC_BASH_TOOL],
            **extra_model_kwargs,
        )

    def query(self, messages: list[dict[str, str]], **kwargs) -> dict:
        for attempt in retry(logger=logger, abort_exceptions=self.abort_exceptions):