import pytest
from src.naliiabot.bot.agent.agent import Agent
from src.naliiabot.bot.factories.llm_factory import LLMFactory


@pytest.fixture
def anthropic_model():
    llmFactory = LLMFactory("anthropic")
    anthropic_model = llmFactory.get_model()

    return anthropic_model


@pytest.fixture
def agent(anthropic_model):
    return Agent(model=anthropic_model)