import os
from langchain_anthropic import ChatAnthropic
from langchain_openai import ChatOpenAI
from langchain_google_genai import ChatGoogleGenerativeAI


def get_orchestrator():
    return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)


def get_implementer():
    return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)


def get_reviewer():
    return ChatAnthropic(model="claude-sonnet-4-20250514", temperature=0)


def get_researcher():
    return ChatAnthropic(model="claude-haiku-4-5-20251001", temperature=0)


def get_cross_reviewer():
    return ChatOpenAI(model="gpt-4.1", temperature=0.2)


def get_scanner():
    return ChatGoogleGenerativeAI(model="gemini-2.5-pro", temperature=0)


def get_fast_coder():
    return ChatOpenAI(
        model="deepseek-coder",
        base_url="https://api.deepseek.com/v1",
        api_key=os.getenv("DEEPSEEK_API_KEY"),
        temperature=0,
    )