Codex
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from app.db.models import Topic
|
|
from app.db.session import SessionLocal
|
|
|
|
|
|
DEFAULT_TOPICS = [
|
|
{
|
|
"name": "Large models and agents",
|
|
"description": "New model releases, agent frameworks, tooling, benchmarks, and product launches.",
|
|
},
|
|
{
|
|
"name": "Open-source AI projects",
|
|
"description": "Important GitHub projects, open model releases, framework updates, and adoption signals.",
|
|
},
|
|
{
|
|
"name": "AI products and funding",
|
|
"description": "AI product launches, startup funding, acquisitions, and strategic market movement.",
|
|
},
|
|
]
|
|
|
|
def main() -> None:
|
|
# 种子数据只创建模型研判时使用的长期主题，采集来源由Agent运行时动态完成。
|
|
db = SessionLocal()
|
|
try:
|
|
for item in DEFAULT_TOPICS:
|
|
exists = db.query(Topic).filter(Topic.name == item["name"]).first()
|
|
if exists:
|
|
continue
|
|
db.add(Topic(**item, enabled=True))
|
|
db.commit()
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|