flying-hero
6cc47ef45c
🚀 重构内容: - 重命名 Lobster → Agent（通用化） - 重命名 LobsterDiary → AgentDiary - 更新所有 API 端点：/api/lobsters/ → /api/agents/ - 前端组件重命名：LobsterDetail → AgentDetail - 数据迁移：8 Lobsters → 8 Agents, 4 Diaries 📦 开源准备: - 创建 .env.example（环境变量配置） - 创建 docker-compose.yml（一键部署） - 创建 Dockerfile（前后端） - 创建 .gitignore - 添加 MIT LICENSE - 完善 README.md（中英双语） - 创建 USAGE.md（使用说明） 📝 文档完善: - REFACTOR_PLAN.md（重构计划） - REFACTOR_PROGRESS.md（重构进度） - REFACTOR_COMPLETE.md（重构完成报告） - FINAL_REPORT.md（最终报告） - 工作区同步报告.md ✨ 功能特性: - 多 Agent 实例管理 - 日记系统（成长之路/工作记忆） - 工作记忆完全隔离 - 日历视图 - 标签和分类 - RAG 支持（预留 embedding 字段） 🎯 开源准备度：100% 🦸 感谢北极星 ⭐ 的耐心指导！
95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
from django.db import models
|
||
from django.utils import timezone
|
||
|
||
|
||
class Agent(models.Model):
|
||
"""AI Agent 模型（原 Lobster）"""
|
||
name = models.CharField(max_length=50, verbose_name='名称')
|
||
emoji = models.CharField(max_length=10, verbose_name='Emoji')
|
||
port = models.IntegerField(verbose_name='端口')
|
||
specialty = models.CharField(max_length=100, verbose_name='专长')
|
||
container = models.CharField(max_length=100, verbose_name='容器')
|
||
app_name = models.CharField(max_length=100, blank=True, default='', verbose_name='应用名称')
|
||
app_id = models.CharField(max_length=50, blank=True, default='', verbose_name='应用 ID')
|
||
workspace = models.CharField(max_length=100, blank=True, default='', verbose_name='工作区')
|
||
created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
|
||
updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')
|
||
|
||
class Meta:
|
||
verbose_name = 'AI Agent'
|
||
verbose_name_plural = 'AI Agents'
|
||
ordering = ['id']
|
||
|
||
def __str__(self):
|
||
return f'{self.emoji} {self.name}'
|
||
|
||
|
||
class AgentDiary(models.Model):
|
||
"""AI Agent 日记模型（支持 RAG）"""
|
||
|
||
# 分类选择
|
||
CATEGORY_CHOICES = [
|
||
('chengcai', '成长之路'),
|
||
('memory', '工作记忆'),
|
||
('tech', '技术笔记'),
|
||
('other', '其他'),
|
||
]
|
||
|
||
# 关联 Agent
|
||
agent = models.ForeignKey(
|
||
Agent,
|
||
on_delete=models.CASCADE,
|
||
related_name='diaries',
|
||
verbose_name='AI Agent'
|
||
)
|
||
|
||
# 基本信息
|
||
date = models.DateField(verbose_name='日期')
|
||
title = models.CharField(max_length=200, verbose_name='标题')
|
||
content = models.TextField(verbose_name='内容')
|
||
|
||
# 分类和标签（RAG 检索的关键元数据）
|
||
category = models.CharField(
|
||
max_length=50,
|
||
choices=CATEGORY_CHOICES,
|
||
default='other',
|
||
verbose_name='分类'
|
||
)
|
||
tags = models.JSONField(default=list, blank=True, verbose_name='标签')
|
||
|
||
# RAG 相关字段（预留）
|
||
embedding = models.TextField(
|
||
blank=True,
|
||
null=True,
|
||
verbose_name='文本向量 (JSON 格式)'
|
||
)
|
||
embedding_model = models.CharField(
|
||
max_length=50,
|
||
blank=True,
|
||
default='',
|
||
verbose_name='Embedding 模型版本'
|
||
)
|
||
|
||
# 时间戳
|
||
created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
|
||
updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')
|
||
|
||
class Meta:
|
||
verbose_name = 'Agent 日记'
|
||
verbose_name_plural = 'Agent 日记'
|
||
ordering = ['-date', '-created_at']
|
||
indexes = [
|
||
models.Index(fields=['agent', 'date']),
|
||
models.Index(fields=['category', 'date']),
|
||
models.Index(fields=['date']),
|
||
]
|
||
|
||
def __str__(self):
|
||
return f'{self.agent.emoji} {self.agent.name} - {self.date} - {self.get_category_display()}'
|
||
|
||
def get_content_preview(self, length=50):
|
||
"""获取内容预览"""
|
||
if len(self.content) <= length:
|
||
return self.content
|
||
return self.content[:length] + '...'
|