observer summary (#1532)

This commit is contained in:
Shuchang Zheng
2025-01-10 14:59:53 -08:00
committed by GitHub
parent 558b9befc4
commit 374b2326c4
6 changed files with 133 additions and 27 deletions

View File

@@ -2093,6 +2093,8 @@ class AgentDB:
workflow_permanent_id: str | None = None,
url: str | None = None,
prompt: str | None = None,
summary: str | None = None,
output: dict[str, Any] | None = None,
organization_id: str | None = None,
) -> ObserverCruise:
async with self.Session() as session:
@@ -2116,6 +2118,10 @@ class AgentDB:
observer_cruise.url = url
if prompt:
observer_cruise.prompt = prompt
if summary:
observer_cruise.summary = summary
if output:
observer_cruise.output = output
await session.commit()
await session.refresh(observer_cruise)
return ObserverCruise.model_validate(observer_cruise)

View File

@@ -548,6 +548,8 @@ class ObserverCruiseModel(Base):
workflow_permanent_id = Column(String, nullable=True)
prompt = Column(UnicodeText, nullable=True)
url = Column(String, nullable=True)
summary = Column(String, nullable=True)
output = Column(JSON, nullable=True)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)