diff --git a/src/backend/app/services/ai_agent/__init__.py b/src/backend/app/services/ai_agent/__init__.py index 5eb34f2..4060524 100644 --- a/src/backend/app/services/ai_agent/__init__.py +++ b/src/backend/app/services/ai_agent/__init__.py @@ -1,4 +1,4 @@ -from .crew import CrewAgent -from .llm_connector import LLMConnector +from .crew import TradingCrew, get_trading_crew +from .llm_connector import MiniMaxLLM, MiniMaxConnector -__all__ = ["CrewAgent", "LLMConnector"] +__all__ = ["TradingCrew", "get_trading_crew", "MiniMaxLLM", "MiniMaxConnector"] diff --git a/src/backend/app/services/ai_agent/crew.py b/src/backend/app/services/ai_agent/crew.py index 2cf3533..6e546e8 100644 --- a/src/backend/app/services/ai_agent/crew.py +++ b/src/backend/app/services/ai_agent/crew.py @@ -1,7 +1,7 @@ from typing import List, Optional, Dict, Any from crewai import Agent, Task, Crew from .llm_connector import MiniMaxConnector, MiniMaxLLM -from ..core.config import get_settings +from ...core.config import get_settings class StrategyValidator: