feat: mcp and backfill fixes
This commit is contained in:
@@ -12,6 +12,7 @@ class JobStatus(StrEnum):
|
||||
RUNNING = "running"
|
||||
DONE = "done"
|
||||
FAILED = "failed"
|
||||
CANCELED = "canceled"
|
||||
|
||||
|
||||
class Account(SQLModel, table=True):
|
||||
|
||||
@@ -25,7 +25,7 @@ async def _media_map(
|
||||
return {}
|
||||
media_rows = await pool.fetch(
|
||||
"SELECT id, chat_id, message_id, kind, downloaded, mime, file_size, "
|
||||
"ttl_seconds FROM media "
|
||||
"ttl_seconds, extracted_text FROM media "
|
||||
"WHERE account_id = $1 AND message_id = ANY($2::bigint[])",
|
||||
account_id,
|
||||
message_ids,
|
||||
|
||||
@@ -356,6 +356,7 @@ def media_ref_from(
|
||||
file_size=(media_row["file_size"] if media_row else None)
|
||||
or obj.get("file_size"),
|
||||
ttl_seconds=media_row["ttl_seconds"] if media_row else None,
|
||||
extracted_text=media_row["extracted_text"] if media_row else None,
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -77,6 +77,7 @@ class MediaRef(BaseModel):
|
||||
mime: str | None = None
|
||||
file_size: int | None = None
|
||||
ttl_seconds: int | None = None
|
||||
extracted_text: str | None = None
|
||||
|
||||
|
||||
class ReactionCount(BaseModel):
|
||||
|
||||
Reference in New Issue
Block a user