Berack/upo-app-agents
0
0
Fork 0
Code Issues Pull Requests 3 Actions Packages Projects Releases Wiki Activity

Refactor team management #26

Merged
Berack96 merged 14 commits from 21-team-monitoring into main 2025-10-15 14:00:39 +02:00
Conversation 1 Commits 14 Files Changed 11 +5 -5
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit f8bed51a60 - Show all commits

4
src/app/agents/__init__.py
View File

@@ -1,5 +1,5 @@
from app.agents.predictor import PredictorInput, PredictorOutput from app.agents.predictor import PredictorInput, PredictorOutput
from app.agents.team import create_team_with
from app.agents.pipeline import Pipeline from app.agents.pipeline import Pipeline
from app.agents.team import AppTeam
__all__ = ["PredictorInput", "PredictorOutput", "create_team_with", "Pipeline"] __all__ = ["PredictorInput", "PredictorOutput", "Pipeline", "AppTeam"]

6
src/app/agents/team.py
View File

@@ -12,10 +12,10 @@ logging = logging.getLogger("AppTeam")
class AppTeam: class AppTeam:
def __init__(self, configs: AppConfig, team_models: AppModel, coordinator: AppModel | None = None): def __init__(self, configs: AppConfig, team_models: AppModel, team_leader: AppModel | None = None):
self.configs = configs self.configs = configs
self.team_models = team_models self.team_models = team_models
self.coordinator = coordinator or team_models self.team_leader = team_leader or team_models
self.listeners: dict[str, Callable[[RunOutputEvent | TeamRunOutputEvent], None]] = {} self.listeners: dict[str, Callable[[RunOutputEvent | TeamRunOutputEvent], None]] = {}
def add_listener(self, event: str, listener: Callable[[RunOutputEvent | TeamRunOutputEvent], None]) -> None: def add_listener(self, event: str, listener: Callable[[RunOutputEvent | TeamRunOutputEvent], None]) -> None:
@@ -26,7 +26,7 @@ class AppTeam:
async def run_team_async(self, query: str) -> str: async def run_team_async(self, query: str) -> str:
logging.info(f"Running team q='{query}'") logging.info(f"Running team q='{query}'")
team = AppTeam.create_team_with(self.configs, self.team_models, self.coordinator) team = AppTeam.create_team_with(self.configs, self.team_models, self.team_leader)
result = "No output from team" result = "No output from team"
async for run_event in team.arun(query, stream=True, stream_intermediate_steps=True): # type: ignore async for run_event in team.arun(query, stream=True, stream_intermediate_steps=True): # type: ignore