from __future__ import annotations
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from ..environment.environment import Environment


class ThinkTool:
    name = "think"
    schema = {
        "type": "function",
        "function": {
            "name": "think",
            "description": "Internal monologue. Use this to reason about your strategy without any external effect.",
            "parameters": {
                "type": "object",
                "properties": {
                    "content": {
                        "type": "string",
                        "description": "Your internal thoughts and reasoning",
                    },
                },
                "required": ["content"],
            },
        },
    }

    @staticmethod
    def run(env: Environment, aid: str, args: dict) -> str:
        content = args.get("content", "")
        return "Thought Acknowledged"