Codex
106 lines
2.2 KiB
Python
106 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import Literal, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class SearchTopic(BaseModel):
|
|
# 主题是模型每次运行时使用的长期情报目标。
|
|
name: str
|
|
description: str
|
|
|
|
|
|
class SignalItem(BaseModel):
|
|
# 模型产出的结构化情报会通过这个契约进入数据库和日报。
|
|
source_type: Literal["news", "github"]
|
|
topic: str
|
|
title: str
|
|
summary: str
|
|
source_url: str
|
|
source_name: str
|
|
published_at: Optional[datetime] = None
|
|
importance: int = Field(ge=1, le=5)
|
|
entities: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class AgentResult(BaseModel):
|
|
# Agent结果同时承载机器可读信号和人可读日报。
|
|
title: str
|
|
signals: list[SignalItem]
|
|
report_markdown: str
|
|
raw_response: dict = Field(default_factory=dict)
|
|
|
|
|
|
class RunResponse(BaseModel):
|
|
run_id: int
|
|
status: str
|
|
report_path: Optional[str] = None
|
|
|
|
|
|
class RunTriggerRead(BaseModel):
|
|
status: str
|
|
run_id: Optional[int] = None
|
|
|
|
|
|
class RunRead(BaseModel):
|
|
id: int
|
|
status: str
|
|
started_at: datetime
|
|
finished_at: Optional[datetime]
|
|
error: Optional[str]
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class SignalRead(BaseModel):
|
|
id: int
|
|
run_id: int
|
|
source_type: Literal["news", "github"]
|
|
topic: str
|
|
title: str
|
|
summary: str
|
|
source_url: str
|
|
source_name: str
|
|
published_at: Optional[datetime]
|
|
importance: int
|
|
entities: list[str]
|
|
created_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class ReportRead(BaseModel):
|
|
id: int
|
|
run_id: int
|
|
title: str
|
|
content_md: str
|
|
file_path: Optional[str]
|
|
created_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class DashboardRead(BaseModel):
|
|
runs: list[RunRead]
|
|
news_signals: list[SignalRead]
|
|
github_signals: list[SignalRead]
|
|
latest_report: Optional[ReportRead]
|
|
topics: list["TopicRead"]
|
|
|
|
|
|
class TopicCreate(BaseModel):
|
|
name: str = Field(min_length=2, max_length=120)
|
|
description: str = Field(min_length=5)
|
|
enabled: bool = True
|
|
|
|
|
|
class TopicRead(BaseModel):
|
|
id: int
|
|
name: str
|
|
description: str
|
|
enabled: bool
|
|
|
|
model_config = {"from_attributes": True}
|