agentpool 2.1.9__py3-none-any.whl → 2.5.0__py3-none-any.whl

agentpool/delegation/base_team.py

@@ -22,7 +22,7 @@ from agentpool_config.teams import TeamConfig
 if TYPE_CHECKING:
     from collections.abc import AsyncIterator, Iterator, Sequence
 
-    from evented.configs import EventConfig
+    from evented_config import EventConfig
     from psygnal.containers._evented_list import ListEvents
     from toprompt import AnyPromptType
 
@@ -31,7 +31,6 @@ if TYPE_CHECKING:
         ModelType,
         ProcessorCallback,
         PromptCompatible,
-        SupportsStructuredOutput,
         ToolType,
     )
     from agentpool.delegation.teamrun import ExtendedTeamTalk, TeamRun
@@ -66,9 +65,6 @@ class BaseTeam[TDeps, TResult](MessageNode[TDeps, TResult]):
         display_name: str | None = None,
         shared_prompt: str | None = None,
         mcp_servers: list[str | MCPServerConfig] | None = None,
-        picker: SupportsStructuredOutput | None = None,
-        num_picks: int | None = None,
-        pick_prompt: str | None = None,
         event_configs: Sequence[EventConfig] | None = None,
         agent_pool: AgentPool | None = None,
     ) -> None:
@@ -97,28 +93,6 @@ class BaseTeam[TDeps, TResult](MessageNode[TDeps, TResult]):
         self.shared_prompt = shared_prompt
         self._main_task: asyncio.Task[ChatMessage[Any] | None] | None = None
         self._infinite = False
-        self.picker = picker
-        self.num_picks = num_picks
-        self.pick_prompt = pick_prompt
-
-    async def pick_agents(self, task: str) -> Sequence[MessageNode[Any, Any]]:
-        """Pick agents to run."""
-        from agentpool.agents.interactions import Interactions
-
-        if self.picker:
-            interactions = Interactions(self.picker)
-            if self.num_picks == 1:
-                result = await interactions.pick(self, task, self.pick_prompt)
-                return [result.selection]
-            multi_result = await interactions.pick_multiple(
-                self,
-                task,
-                min_picks=self.num_picks or 1,
-                max_picks=self.num_picks,
-                prompt=self.pick_prompt,
-            )
-            return multi_result.selections
-        return list(self.nodes)
 
     def _on_node_changed(
         self, index: int, old: MessageNode[Any, Any], new: MessageNode[Any, Any]
@@ -382,7 +356,7 @@ class BaseTeam[TDeps, TResult](MessageNode[TDeps, TResult]):
                         team_config = shared_pool.manifest.teams.get(self.name)
         if not team_config:
             mode = "parallel" if isinstance(self, Team) else "sequential"
-            team_config = TeamConfig(name=self.name, mode=mode, members=[])
+            team_config = TeamConfig(name=self.name, mode=mode, members=[])  # type: ignore[call-arg]
         if not pool_ids:
             logger.debug("No pool found for team.", team=self.name)
             return TeamContext(
@@ -495,8 +469,8 @@ if __name__ == "__main__":
     async def main() -> None:
         from agentpool import Agent, Team
 
-        agent = Agent("My Agent")
-        agent_2 = Agent("My Agent")
+        agent = Agent("My Agent", model="openai:gpt-5-nano")
+        agent_2 = Agent("My Agent", model="openai:gpt-5-nano")
         team = Team([agent, agent_2], mcp_servers=["uvx mcp-server-git"])
         async with team:
             print(await agent.tools.get_tools())