17 lines
366 B
Python
17 lines
366 B
Python
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)
|