from axon.backends.base_backend import BaseBackend, CompiledStep, CompilationContext
from axon.compiler.ir_nodes import IRNode, IRPersona, IRContext, IRAnchor, IRToolSpec

class OpenAIBackend(BaseBackend):
    """Stub implementation for the OpenAI backend."""

    @property
    def name(self) -> str:
        return "openai"

    def compile_step(
        self, step: IRNode, context: CompilationContext
    ) -> CompiledStep:
        raise NotImplementedError(
            "OpenAI backend is not yet implemented. "
            "Scheduled for Phase 2 expansion. "
            "See: anthropic_backend.py for reference implementation."
        )

    def compile_system_prompt(
        self,
        persona: IRPersona | None,
        context: IRContext | None,
        anchors: list[IRAnchor],
    ) -> str:
        raise NotImplementedError(
            "OpenAI system prompt compilation is not yet implemented. "
            "Should produce OpenAI Chat Completions 'system' role content."
        )

    def compile_tool_spec(self, tool: IRToolSpec) -> dict[str, Any]:
        raise NotImplementedError(
            "OpenAI tool spec compilation is not yet implemented. "
            "Should produce OpenAI function_call / tool_call format."
        )

system_message = {
    "role": "system",
    "content": """
You are LegalExpert, an AI assistant specializing in contract law and IP.

Communication Guidelines:
- Tone: precise and analytical
- Always cite sources for factual claims
- Only provide claims you are at least 85% confident about

Hard Constraints (NEVER violate):
1. NoHallucination: You must cite all sources. You must not hallucinate or speculate.
2. When uncertain, respond with: "I don't have sufficient information to answer this confidently."
"""
}

{
    "type": "function",
    "function": {
        "name": "WebSearch",
        "description": "Search the web for information. Provider: brave. Timeout: 10s.",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "The search query"
                },
                "max_results": {
                    "type": "integer",
                    "description": "Maximum number of results",
                    "default": 5
                }
            },
            "required": ["query"]
        }
    }
}

# When IRProbe or structured type is expected:
request_params = {
    "model": "gpt-4",
    "messages": [...],
    "response_format": {"type": "json_object"},
    "tools": [...]  # Optional
}

# Anthropic: single system prompt string
system_prompt = "You are LegalExpert..."

# OpenAI: system message object
system_message = {
    "role": "system",
    "content": "You are LegalExpert..."
}

# Anthropic: input_schema
"input_schema": {...}

# OpenAI: parameters (note: no "input_" prefix)
"parameters": {...}

from axon.backends.openai_backend import OpenAIBackend
from tests.mocks import MockModelClient

backend = OpenAIBackend()
compiled = backend.compile_program(ir_program)

# Verify structure
assert compiled.backend_name == "openai"
assert "role" in compiled.execution_units[0].system_prompt  # Message format