Skip to content

create_team

fastagency.studio.io.ionats.create_team async #

create_team(
    team_id: UUID, user_id: UUID
) -> Callable[[str], list[dict[str, Any]]]
Source code in fastagency/studio/io/ionats.py 
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
async def create_team(
    team_id: UUID, user_id: UUID
) -> Callable[[str], list[dict[str, Any]]]:
    team_dict = await DefaultDB.backend().find_model(team_id)

    team_model: Union[TwoAgentTeam, MultiAgentTeam]
    if "initial_agent" in team_dict["json_str"]:
        team_model = TwoAgentTeam(**team_dict["json_str"])
    elif "agent_1" in team_dict["json_str"]:
        team_model = MultiAgentTeam(**team_dict["json_str"])
    else:
        raise ValueError(f"Unknown team model {team_dict['json_str']}")

    autogen_team = await team_model.create_autogen(team_id, user_id)

    return autogen_team.initiate_chat  # type: ignore[no-any-return]