firts agent concept
This commit is contained in:
16
tests/conftest.py
Normal file
16
tests/conftest.py
Normal file
@@ -0,0 +1,16 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user