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,
|
task_listen=m.agent_config.task_listen,
|
||||||
prompt=m.agent_config.prompt,
|
prompt=m.agent_config.prompt,
|
||||||
model=m.agent_config.model,
|
model=m.agent_config.model,
|
||||||
|
mentionable=m.agent_config.mentionable if m.agent_config.mentionable is not None else True,
|
||||||
)
|
)
|
||||||
return MemberOut(
|
return MemberOut(
|
||||||
id=str(m.id),
|
id=str(m.id),
|
||||||
|
|||||||
@ -28,6 +28,7 @@ class AgentConfigSchema(BaseModel):
|
|||||||
task_listen: str = ListenMode.MENTIONS
|
task_listen: str = ListenMode.MENTIONS
|
||||||
prompt: str | None = None
|
prompt: str | None = None
|
||||||
model: str | None = None
|
model: str | None = None
|
||||||
|
mentionable: bool = True
|
||||||
|
|
||||||
|
|
||||||
class MemberCreate(BaseModel):
|
class MemberCreate(BaseModel):
|
||||||
@ -119,10 +120,12 @@ async def create_member(req: MemberCreate, db: AsyncSession = Depends(get_db)):
|
|||||||
config = AgentConfig(
|
config = AgentConfig(
|
||||||
member_id=member.id,
|
member_id=member.id,
|
||||||
capabilities=ac.capabilities if ac else [],
|
capabilities=ac.capabilities if ac else [],
|
||||||
|
labels=ac.labels if ac else [],
|
||||||
chat_listen=ac.chat_listen if ac else ListenMode.ALL,
|
chat_listen=ac.chat_listen if ac else ListenMode.ALL,
|
||||||
task_listen=ac.task_listen if ac else ListenMode.ALL,
|
task_listen=ac.task_listen if ac else ListenMode.ALL,
|
||||||
prompt=ac.prompt if ac else None,
|
prompt=ac.prompt if ac else None,
|
||||||
model=ac.model if ac else None,
|
model=ac.model if ac else None,
|
||||||
|
mentionable=ac.mentionable if ac else True,
|
||||||
)
|
)
|
||||||
db.add(config)
|
db.add(config)
|
||||||
|
|
||||||
|
|||||||
@ -87,6 +87,7 @@ class AgentConfigOut(BaseModel):
|
|||||||
task_listen: str
|
task_listen: str
|
||||||
prompt: str | None = None
|
prompt: str | None = None
|
||||||
model: str | None = None
|
model: str | None = None
|
||||||
|
mentionable: bool = True
|
||||||
|
|
||||||
|
|
||||||
class MemberOut(BaseModel):
|
class MemberOut(BaseModel):
|
||||||
|
|||||||
@ -51,5 +51,6 @@ class AgentConfig(Base):
|
|||||||
model: Mapped[str | None] = mapped_column(String(100))
|
model: Mapped[str | None] = mapped_column(String(100))
|
||||||
provider: Mapped[str | None] = mapped_column(String(50))
|
provider: Mapped[str | None] = mapped_column(String(50))
|
||||||
max_concurrent_tasks: Mapped[int] = mapped_column(default=2)
|
max_concurrent_tasks: Mapped[int] = mapped_column(default=2)
|
||||||
|
mentionable: Mapped[bool] = mapped_column(default=True)
|
||||||
|
|
||||||
member: Mapped["Member"] = relationship(back_populates="agent_config")
|
member: Mapped["Member"] = relationship(back_populates="agent_config")
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user