feat: mentionable flag for agents + labels in create
Some checks failed
Deploy Tracker / deploy (push) Failing after 3s
Some checks failed
Deploy Tracker / deploy (push) Failing after 3s
This commit is contained in:
parent
64a1223c1d
commit
274294c183
@ -40,6 +40,7 @@ def member_out(m: Member) -> MemberOut:
|
||||
task_listen=m.agent_config.task_listen,
|
||||
prompt=m.agent_config.prompt,
|
||||
model=m.agent_config.model,
|
||||
mentionable=m.agent_config.mentionable if m.agent_config.mentionable is not None else True,
|
||||
)
|
||||
return MemberOut(
|
||||
id=str(m.id),
|
||||
|
||||
@ -28,6 +28,7 @@ class AgentConfigSchema(BaseModel):
|
||||
task_listen: str = ListenMode.MENTIONS
|
||||
prompt: str | None = None
|
||||
model: str | None = None
|
||||
mentionable: bool = True
|
||||
|
||||
|
||||
class MemberCreate(BaseModel):
|
||||
@ -119,10 +120,12 @@ async def create_member(req: MemberCreate, db: AsyncSession = Depends(get_db)):
|
||||
config = AgentConfig(
|
||||
member_id=member.id,
|
||||
capabilities=ac.capabilities if ac else [],
|
||||
labels=ac.labels if ac else [],
|
||||
chat_listen=ac.chat_listen if ac else ListenMode.ALL,
|
||||
task_listen=ac.task_listen if ac else ListenMode.ALL,
|
||||
prompt=ac.prompt if ac else None,
|
||||
model=ac.model if ac else None,
|
||||
mentionable=ac.mentionable if ac else True,
|
||||
)
|
||||
db.add(config)
|
||||
|
||||
|
||||
@ -87,6 +87,7 @@ class AgentConfigOut(BaseModel):
|
||||
task_listen: str
|
||||
prompt: str | None = None
|
||||
model: str | None = None
|
||||
mentionable: bool = True
|
||||
|
||||
|
||||
class MemberOut(BaseModel):
|
||||
|
||||
@ -51,5 +51,6 @@ class AgentConfig(Base):
|
||||
model: Mapped[str | None] = mapped_column(String(100))
|
||||
provider: Mapped[str | None] = mapped_column(String(50))
|
||||
max_concurrent_tasks: Mapped[int] = mapped_column(default=2)
|
||||
mentionable: Mapped[bool] = mapped_column(default=True)
|
||||
|
||||
member: Mapped["Member"] = relationship(back_populates="agent_config")
|
||||
|
||||
Loading…
Reference in New Issue
Block a user