feat: 完成 Agent Diary 开源重构 🎉
🚀 重构内容: - 重命名 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% 🦸 感谢北极星 ⭐ 的耐心指导！
This commit is contained in:
1
code/backend/agents/__init__.py
Normal file
1
code/backend/agents/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
# AI Agents 管理
|
||||
7
code/backend/agents/apps.py
Normal file
7
code/backend/agents/apps.py
Normal file
@@ -0,0 +1,7 @@
|
||||
from django.apps import AppConfig
|
||||
|
||||
|
||||
class AgentsConfig(AppConfig):
|
||||
default_auto_field = 'django.db.models.BigAutoField'
|
||||
name = 'agents'
|
||||
verbose_name = 'AI Agents 管理'
|
||||
0
code/backend/agents/management/__init__.py
Normal file
0
code/backend/agents/management/__init__.py
Normal file
0
code/backend/agents/management/commands/__init__.py
Normal file
0
code/backend/agents/management/commands/__init__.py
Normal file
69
code/backend/agents/management/commands/migrate_data.py
Normal file
69
code/backend/agents/management/commands/migrate_data.py
Normal file
@@ -0,0 +1,69 @@
|
||||
#!/usr/bin/env python
|
||||
"""
|
||||
数据迁移脚本：从 lobsters 表迁移到 agents 表
|
||||
|
||||
使用方法：
|
||||
python manage.py migrate_data
|
||||
"""
|
||||
|
||||
from django.core.management.base import BaseCommand
|
||||
from lobsters.models import Lobster, LobsterDiary
|
||||
from agents.models import Agent, AgentDiary
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = '从 lobsters 表迁移数据到 agents 表'
|
||||
|
||||
def handle(self, *args, **kwargs):
|
||||
self.stdout.write('🚀 开始数据迁移...\n')
|
||||
|
||||
# 迁移 Lobster → Agent
|
||||
self.stdout.write('📊 迁移 Lobster 数据...')
|
||||
lobsters = Lobster.objects.all()
|
||||
agent_count = 0
|
||||
|
||||
for lobster in lobsters:
|
||||
Agent.objects.update_or_create(
|
||||
id=lobster.id,
|
||||
defaults={
|
||||
'name': lobster.name,
|
||||
'emoji': lobster.emoji,
|
||||
'port': lobster.port,
|
||||
'specialty': lobster.specialty,
|
||||
'container': lobster.container,
|
||||
'app_name': lobster.app_name,
|
||||
'app_id': lobster.app_id,
|
||||
'workspace': lobster.workspace,
|
||||
}
|
||||
)
|
||||
agent_count += 1
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'✅ 迁移 {agent_count} 只 Lobster → Agent'))
|
||||
|
||||
# 迁移 LobsterDiary → AgentDiary
|
||||
self.stdout.write('\n📝 迁移 Diary 数据...')
|
||||
diaries = LobsterDiary.objects.all()
|
||||
diary_count = 0
|
||||
|
||||
for diary in diaries:
|
||||
AgentDiary.objects.update_or_create(
|
||||
id=diary.id,
|
||||
defaults={
|
||||
'agent_id': diary.lobster_id,
|
||||
'date': diary.date,
|
||||
'title': diary.title,
|
||||
'content': diary.content,
|
||||
'category': diary.category,
|
||||
'tags': diary.tags,
|
||||
'embedding': diary.embedding,
|
||||
'embedding_model': diary.embedding_model,
|
||||
}
|
||||
)
|
||||
diary_count += 1
|
||||
|
||||
self.stdout.write(self.style.SUCCESS(f'✅ 迁移 {diary_count} 篇 Diary'))
|
||||
|
||||
# 统计
|
||||
self.stdout.write(self.style.SUCCESS(f'\n🎉 迁移完成！'))
|
||||
self.stdout.write(f' Agents: {Agent.objects.count()}')
|
||||
self.stdout.write(f' AgentDiaries: {AgentDiary.objects.count()}')
|
||||
152
code/backend/agents/migrations/0001_initial.py
Normal file
152
code/backend/agents/migrations/0001_initial.py
Normal file
@@ -0,0 +1,152 @@
|
||||
# Generated by Django 4.2 on 2026-04-03 10:15
|
||||
|
||||
from django.db import migrations, models
|
||||
import django.db.models.deletion
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
|
||||
initial = True
|
||||
|
||||
dependencies = []
|
||||
|
||||
operations = [
|
||||
migrations.CreateModel(
|
||||
name="Agent",
|
||||
fields=[
|
||||
(
|
||||
"id",
|
||||
models.BigAutoField(
|
||||
auto_created=True,
|
||||
primary_key=True,
|
||||
serialize=False,
|
||||
verbose_name="ID",
|
||||
),
|
||||
),
|
||||
("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(
|
||||
blank=True, default="", max_length=100, verbose_name="应用名称"
|
||||
),
|
||||
),
|
||||
(
|
||||
"app_id",
|
||||
models.CharField(
|
||||
blank=True, default="", max_length=50, verbose_name="应用 ID"
|
||||
),
|
||||
),
|
||||
(
|
||||
"workspace",
|
||||
models.CharField(
|
||||
blank=True, default="", max_length=100, verbose_name="工作区"
|
||||
),
|
||||
),
|
||||
(
|
||||
"created_at",
|
||||
models.DateTimeField(auto_now_add=True, verbose_name="创建时间"),
|
||||
),
|
||||
(
|
||||
"updated_at",
|
||||
models.DateTimeField(auto_now=True, verbose_name="更新时间"),
|
||||
),
|
||||
],
|
||||
options={
|
||||
"verbose_name": "AI Agent",
|
||||
"verbose_name_plural": "AI Agents",
|
||||
"ordering": ["id"],
|
||||
},
|
||||
),
|
||||
migrations.CreateModel(
|
||||
name="AgentDiary",
|
||||
fields=[
|
||||
(
|
||||
"id",
|
||||
models.BigAutoField(
|
||||
auto_created=True,
|
||||
primary_key=True,
|
||||
serialize=False,
|
||||
verbose_name="ID",
|
||||
),
|
||||
),
|
||||
("date", models.DateField(verbose_name="日期")),
|
||||
("title", models.CharField(max_length=200, verbose_name="标题")),
|
||||
("content", models.TextField(verbose_name="内容")),
|
||||
(
|
||||
"category",
|
||||
models.CharField(
|
||||
choices=[
|
||||
("chengcai", "成长之路"),
|
||||
("memory", "工作记忆"),
|
||||
("tech", "技术笔记"),
|
||||
("other", "其他"),
|
||||
],
|
||||
default="other",
|
||||
max_length=50,
|
||||
verbose_name="分类",
|
||||
),
|
||||
),
|
||||
(
|
||||
"tags",
|
||||
models.JSONField(blank=True, default=list, verbose_name="标签"),
|
||||
),
|
||||
(
|
||||
"embedding",
|
||||
models.TextField(
|
||||
blank=True, null=True, verbose_name="文本向量 (JSON 格式)"
|
||||
),
|
||||
),
|
||||
(
|
||||
"embedding_model",
|
||||
models.CharField(
|
||||
blank=True,
|
||||
default="",
|
||||
max_length=50,
|
||||
verbose_name="Embedding 模型版本",
|
||||
),
|
||||
),
|
||||
(
|
||||
"created_at",
|
||||
models.DateTimeField(auto_now_add=True, verbose_name="创建时间"),
|
||||
),
|
||||
(
|
||||
"updated_at",
|
||||
models.DateTimeField(auto_now=True, verbose_name="更新时间"),
|
||||
),
|
||||
(
|
||||
"agent",
|
||||
models.ForeignKey(
|
||||
on_delete=django.db.models.deletion.CASCADE,
|
||||
related_name="diaries",
|
||||
to="agents.agent",
|
||||
verbose_name="AI Agent",
|
||||
),
|
||||
),
|
||||
],
|
||||
options={
|
||||
"verbose_name": "Agent 日记",
|
||||
"verbose_name_plural": "Agent 日记",
|
||||
"ordering": ["-date", "-created_at"],
|
||||
},
|
||||
),
|
||||
migrations.AddIndex(
|
||||
model_name="agentdiary",
|
||||
index=models.Index(
|
||||
fields=["agent", "date"], name="agents_agen_agent_i_27d331_idx"
|
||||
),
|
||||
),
|
||||
migrations.AddIndex(
|
||||
model_name="agentdiary",
|
||||
index=models.Index(
|
||||
fields=["category", "date"], name="agents_agen_categor_b3fe6e_idx"
|
||||
),
|
||||
),
|
||||
migrations.AddIndex(
|
||||
model_name="agentdiary",
|
||||
index=models.Index(fields=["date"], name="agents_agen_date_8e7dc6_idx"),
|
||||
),
|
||||
]
|
||||
0
code/backend/agents/migrations/__init__.py
Normal file
0
code/backend/agents/migrations/__init__.py
Normal file
94
code/backend/agents/models.py
Normal file
94
code/backend/agents/models.py
Normal file
@@ -0,0 +1,94 @@
|
||||
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] + '...'
|
||||
Reference in New Issue
Block a user