118 lines
6.8 KiB
Python
118 lines
6.8 KiB
Python
# Generated by Django 5.2.15 on 2026-07-27 05:53
|
|
|
|
import django.db.models.deletion
|
|
import uuid
|
|
from django.conf import settings
|
|
from django.db import migrations, models
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
initial = True
|
|
|
|
dependencies = [
|
|
('agents', '0005_agentversion_allowed_actions'),
|
|
migrations.swappable_dependency(settings.AUTH_USER_MODEL),
|
|
]
|
|
|
|
operations = [
|
|
migrations.CreateModel(
|
|
name='ToolCall',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('public_id', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
|
|
('status', models.CharField(choices=[('pending_confirmation', '待确认'), ('running', '执行中'), ('succeeded', '成功'), ('failed', '失败'), ('rejected', '已拒绝')], max_length=32)),
|
|
('input_payload', models.JSONField(default=dict)),
|
|
('output_payload', models.JSONField(blank=True, null=True)),
|
|
('authorization_snapshot', models.JSONField(default=dict)),
|
|
('risk_snapshot', models.JSONField(default=dict)),
|
|
('idempotency_key', models.CharField(blank=True, max_length=160)),
|
|
('error_code', models.CharField(blank=True, max_length=80)),
|
|
('error_message', models.TextField(blank=True)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('started_at', models.DateTimeField(blank=True, null=True)),
|
|
('finished_at', models.DateTimeField(blank=True, null=True)),
|
|
('agent_run', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='tool_calls', to='agents.agentrun')),
|
|
('requested_by', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='requested_tool_calls', to=settings.AUTH_USER_MODEL)),
|
|
],
|
|
options={
|
|
'ordering': ['-created_at'],
|
|
},
|
|
),
|
|
migrations.CreateModel(
|
|
name='ToolConfirmation',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('status', models.CharField(default='pending', max_length=24)),
|
|
('reason', models.TextField(blank=True)),
|
|
('decided_at', models.DateTimeField(blank=True, null=True)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('call', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='confirmation', to='tools.toolcall')),
|
|
('decided_by', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='decided_tool_confirmations', to=settings.AUTH_USER_MODEL)),
|
|
],
|
|
),
|
|
migrations.CreateModel(
|
|
name='ToolProvider',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('name', models.CharField(max_length=120, unique=True)),
|
|
('provider_type', models.CharField(choices=[('native', '原生工具'), ('mcp_http', 'HTTP MCP')], default='mcp_http', max_length=24)),
|
|
('base_url', models.URLField(blank=True, max_length=500)),
|
|
('token_encrypted', models.TextField(blank=True)),
|
|
('enabled', models.BooleanField(default=True)),
|
|
('timeout_seconds', models.PositiveIntegerField(default=30)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('updated_at', models.DateTimeField(auto_now=True)),
|
|
('created_by', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='created_tool_providers', to=settings.AUTH_USER_MODEL)),
|
|
],
|
|
options={
|
|
'ordering': ['name'],
|
|
},
|
|
),
|
|
migrations.CreateModel(
|
|
name='UnifiedTool',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('code', models.CharField(max_length=160, unique=True)),
|
|
('external_name', models.CharField(blank=True, max_length=200)),
|
|
('name', models.CharField(max_length=160)),
|
|
('description', models.TextField(blank=True)),
|
|
('input_schema', models.JSONField(default=dict)),
|
|
('risk_level', models.CharField(choices=[('read_only', '只读'), ('low_write', '低风险写'), ('high_risk', '高风险')], default='read_only', max_length=24)),
|
|
('enabled', models.BooleanField(default=True)),
|
|
('synced_at', models.DateTimeField(blank=True, null=True)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('updated_at', models.DateTimeField(auto_now=True)),
|
|
('provider', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, related_name='tools', to='tools.toolprovider')),
|
|
],
|
|
options={
|
|
'ordering': ['code'],
|
|
},
|
|
),
|
|
migrations.AddField(
|
|
model_name='toolcall',
|
|
name='tool',
|
|
field=models.ForeignKey(on_delete=django.db.models.deletion.PROTECT, related_name='calls', to='tools.unifiedtool'),
|
|
),
|
|
migrations.CreateModel(
|
|
name='ToolAuthorization',
|
|
fields=[
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
|
('principal_type', models.CharField(max_length=24)),
|
|
('principal_id', models.PositiveBigIntegerField()),
|
|
('allowed', models.BooleanField(default=True)),
|
|
('created_at', models.DateTimeField(auto_now_add=True)),
|
|
('created_by', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='created_tool_authorizations', to=settings.AUTH_USER_MODEL)),
|
|
('tool', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='authorizations', to='tools.unifiedtool')),
|
|
],
|
|
),
|
|
migrations.AddConstraint(
|
|
model_name='toolcall',
|
|
constraint=models.UniqueConstraint(condition=models.Q(('idempotency_key__gt', '')), fields=('tool', 'idempotency_key'), name='unique_tool_call_idempotency_key'),
|
|
),
|
|
migrations.AddConstraint(
|
|
model_name='toolauthorization',
|
|
constraint=models.UniqueConstraint(fields=('tool', 'principal_type', 'principal_id'), name='unique_tool_authorization_principal'),
|
|
),
|
|
]
|