zhaoxi
c0e4fd34ae
* refactor: overhaul workflow and chat architecture - Separate Chat and Workflow API endpoints and database models - Use JSONB to store workflow execution context in Postgres - Convert workflow engine to use pydantic-ai execution graphs inside a Ray task - Update frontend React components to support standalone workflow creation - Remove obsolete and broken workflow runner tests Co-authored-by: zhaoxi826 <198742034+zhaoxi826@users.noreply.github.com> * refactor: overhaul workflow and chat architecture - Separate Chat and Workflow API endpoints and database models - Use JSONB to store workflow execution context in Postgres - Convert workflow engine to use pydantic-ai execution graphs inside a Ray task - Update frontend React components to support standalone workflow creation - Remove obsolete and broken workflow runner tests Co-authored-by: zhaoxi826 <198742034+zhaoxi826@users.noreply.github.com> * refactor: overhaul workflow and chat architecture - Separate Chat and Workflow API endpoints and database models - Use JSONB to store workflow execution context in Postgres - Convert workflow engine to use pydantic-ai execution graphs inside a Ray task - Update frontend React components to support standalone workflow creation - Move workflow_engine inside workflow package to keep core root clean - Remove obsolete and broken workflow runner tests Co-authored-by: zhaoxi826 <198742034+zhaoxi826@users.noreply.github.com> * refactor: overhaul workflow and chat architecture - Separate Chat and Workflow API endpoints and database models - Use JSONB to store workflow execution context in Postgres - Convert workflow engine to use pydantic-ai execution graphs inside a Ray task - Update frontend React components to support standalone workflow creation - Move workflow_engine inside workflow package to keep core root clean - Replace sqlmodel with pure sqlalchemy mappings globally - Remove obsolete and broken workflow runner tests Co-authored-by: zhaoxi826 <198742034+zhaoxi826@users.noreply.github.com> --------- Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com> Co-authored-by: zhaoxi826 <198742034+zhaoxi826@users.noreply.github.com>
47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from sqlalchemy import select
|
|
from typing import List, Optional
|
|
from kilostar.core.postgres_database.model.workflow import EventRecord
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, AsyncSession
|
|
|
|
|
|
class EventDatabase:
|
|
def __init__(self, async_session_maker: async_sessionmaker[AsyncSession]):
|
|
self.async_session_maker = async_session_maker
|
|
|
|
async def upsert_event(self, trace_id: str, event_data_json: str) -> EventRecord:
|
|
async with self.async_session_maker() as session:
|
|
statement = select(EventRecord).where(EventRecord.trace_id == trace_id)
|
|
results = await session.execute(statement)
|
|
record = results.scalar_one_or_none()
|
|
if record:
|
|
record.event_data_json = event_data_json
|
|
else:
|
|
record = EventRecord(trace_id=trace_id, event_data_json=event_data_json)
|
|
session.add(record)
|
|
await session.commit()
|
|
await session.refresh(record)
|
|
return record
|
|
|
|
async def get_event(self, trace_id: str) -> Optional[EventRecord]:
|
|
async with self.async_session_maker() as session:
|
|
statement = select(EventRecord).where(EventRecord.trace_id == trace_id)
|
|
results = await session.execute(statement)
|
|
return results.scalar_one_or_none()
|
|
|
|
async def get_all_events(self) -> List[EventRecord]:
|
|
async with self.async_session_maker() as session:
|
|
statement = select(EventRecord)
|
|
results = await session.execute(statement)
|
|
return results.scalars().all()
|
|
|
|
async def delete_event(self, trace_id: str) -> bool:
|
|
async with self.async_session_maker() as session:
|
|
statement = select(EventRecord).where(EventRecord.trace_id == trace_id)
|
|
results = await session.execute(statement)
|
|
record = results.scalar_one_or_none()
|
|
if record:
|
|
await session.delete(record)
|
|
await session.commit()
|
|
return True
|
|
return False
|