Implement configurable API providers from configs.yaml #43
@@ -89,10 +89,9 @@ class PipelineInputs:
|
|||||||
# Agent getters
|
# Agent getters
|
||||||
# ======================
|
# ======================
|
||||||
def get_agent_team(self) -> Team:
|
def get_agent_team(self) -> Team:
|
||||||
market, news, social = self.get_tools()
|
market_agent = self.team_model.get_agent(MARKET_INSTRUCTIONS, "Market Agent", tools=[MarketAPIsTool()])
|
||||||
market_agent = self.team_model.get_agent(MARKET_INSTRUCTIONS, "Market Agent", tools=[market])
|
news_agent = self.team_model.get_agent(NEWS_INSTRUCTIONS, "News Agent", tools=[NewsAPIsTool()])
|
||||||
news_agent = self.team_model.get_agent(NEWS_INSTRUCTIONS, "News Agent", tools=[news])
|
social_agent = self.team_model.get_agent(SOCIAL_INSTRUCTIONS, "Socials Agent", tools=[SocialAPIsTool()])
|
||||||
social_agent = self.team_model.get_agent(SOCIAL_INSTRUCTIONS, "Socials Agent", tools=[social])
|
|
||||||
return Team(
|
return Team(
|
||||||
model=self.team_leader_model.get_model(TEAM_LEADER_INSTRUCTIONS),
|
model=self.team_leader_model.get_model(TEAM_LEADER_INSTRUCTIONS),
|
||||||
name="CryptoAnalysisTeam",
|
name="CryptoAnalysisTeam",
|
||||||
@@ -105,17 +104,3 @@ class PipelineInputs:
|
|||||||
|
|
||||||
def get_agent_report_generator(self) -> Agent:
|
def get_agent_report_generator(self) -> Agent:
|
||||||
return self.report_generation_model.get_agent(REPORT_GENERATION_INSTRUCTIONS, "Report Generator Agent")
|
return self.report_generation_model.get_agent(REPORT_GENERATION_INSTRUCTIONS, "Report Generator Agent")
|
||||||
|
|
||||||
def get_tools(self) -> tuple[MarketAPIsTool, NewsAPIsTool, SocialAPIsTool]:
|
|
||||||
"""
|
|
||||||
Restituisce la lista di tools disponibili per gli agenti.
|
|
||||||
"""
|
|
||||||
api = self.configs.api
|
|
||||||
|
|
||||||
market_tool = MarketAPIsTool()
|
|
||||||
market_tool.handler.set_retries(api.retry_attempts, api.retry_delay_seconds)
|
|
||||||
news_tool = NewsAPIsTool()
|
|
||||||
news_tool.handler.set_retries(api.retry_attempts, api.retry_delay_seconds)
|
|
||||||
social_tool = SocialAPIsTool()
|
|
||||||
social_tool.handler.set_retries(api.retry_attempts, api.retry_delay_seconds)
|
|
||||||
return market_tool, news_tool, social_tool
|
|
||||||
|
|||||||
Reference in New Issue
Block a user