feat: add Session, Message, ToolCall models with migrations
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
99
src/openclaw/models.py
Normal file
99
src/openclaw/models.py
Normal file
@@ -0,0 +1,99 @@
|
||||
from django.db import models
|
||||
|
||||
|
||||
class Session(models.Model):
|
||||
session_id = models.CharField(max_length=64)
|
||||
agent_name = models.CharField(max_length=128)
|
||||
source_node = models.CharField(max_length=64)
|
||||
session_version = models.IntegerField(default=0)
|
||||
model_provider = models.CharField(max_length=64, blank=True, default="")
|
||||
model_id = models.CharField(max_length=128, blank=True, default="")
|
||||
thinking_level = models.CharField(max_length=64, blank=True, default="")
|
||||
start_time = models.DateTimeField(null=True, blank=True)
|
||||
end_time = models.DateTimeField(null=True, blank=True)
|
||||
cwd = models.CharField(max_length=512, blank=True, default="")
|
||||
total_tokens = models.IntegerField(default=0)
|
||||
total_cost = models.FloatField(default=0.0)
|
||||
message_count = models.IntegerField(default=0)
|
||||
tool_call_count = models.IntegerField(default=0)
|
||||
error_count = models.IntegerField(default=0)
|
||||
raw_file_path = models.TextField(blank=True, default="")
|
||||
pushed_at = models.DateTimeField(null=True, blank=True)
|
||||
status = models.CharField(max_length=16, default="active")
|
||||
metadata = models.JSONField(default=dict, blank=True)
|
||||
|
||||
created_at = models.DateTimeField(auto_now_add=True)
|
||||
updated_at = models.DateTimeField(auto_now=True)
|
||||
|
||||
class Meta:
|
||||
db_table = "sessions"
|
||||
unique_together = ("session_id", "agent_name")
|
||||
ordering = ["-start_time"]
|
||||
|
||||
def __str__(self):
|
||||
return f"Session({self.session_id} {self.agent_name})"
|
||||
|
||||
|
||||
class Message(models.Model):
|
||||
session = models.ForeignKey(
|
||||
Session, on_delete=models.CASCADE, related_name="messages"
|
||||
)
|
||||
message_id = models.CharField(max_length=128)
|
||||
parent_id = models.CharField(max_length=128, blank=True, default="")
|
||||
seq = models.IntegerField(default=0)
|
||||
role = models.CharField(max_length=32)
|
||||
content_text = models.TextField(blank=True, default="")
|
||||
raw_content = models.JSONField(default=list, blank=True)
|
||||
raw_message = models.JSONField(default=dict, blank=True)
|
||||
timestamp = models.DateTimeField()
|
||||
# assistant 专用
|
||||
model = models.CharField(max_length=128, blank=True, default="")
|
||||
provider = models.CharField(max_length=64, blank=True, default="")
|
||||
stop_reason = models.CharField(max_length=64, blank=True, default="")
|
||||
tokens_input = models.IntegerField(default=0)
|
||||
tokens_output = models.IntegerField(default=0)
|
||||
tokens_cache_read = models.IntegerField(default=0)
|
||||
tokens_cache_write = models.IntegerField(default=0)
|
||||
tokens_total = models.IntegerField(default=0)
|
||||
cost_total = models.FloatField(default=0.0)
|
||||
# toolResult 专用
|
||||
tool_call_id = models.CharField(max_length=128, blank=True, default="")
|
||||
tool_name = models.CharField(max_length=128, blank=True, default="")
|
||||
is_error = models.BooleanField(default=False)
|
||||
exit_code = models.IntegerField(null=True, blank=True)
|
||||
duration_ms = models.IntegerField(null=True, blank=True)
|
||||
|
||||
created_at = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
class Meta:
|
||||
db_table = "messages"
|
||||
ordering = ["seq"]
|
||||
|
||||
def __str__(self):
|
||||
return f"Message({self.message_id} {self.role})"
|
||||
|
||||
|
||||
class ToolCall(models.Model):
|
||||
session = models.ForeignKey(
|
||||
Session, on_delete=models.CASCADE, related_name="tool_calls"
|
||||
)
|
||||
message = models.ForeignKey(
|
||||
Message, on_delete=models.CASCADE, related_name="tool_calls"
|
||||
)
|
||||
tool_call_id = models.CharField(max_length=128)
|
||||
tool_name = models.CharField(max_length=128)
|
||||
arguments = models.JSONField(default=dict, blank=True)
|
||||
result_text = models.TextField(blank=True, default="")
|
||||
is_error = models.BooleanField(default=False)
|
||||
exit_code = models.IntegerField(null=True, blank=True)
|
||||
duration_ms = models.IntegerField(null=True, blank=True)
|
||||
seq = models.IntegerField(default=0)
|
||||
|
||||
created_at = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
class Meta:
|
||||
db_table = "tool_calls"
|
||||
ordering = ["seq"]
|
||||
|
||||
def __str__(self):
|
||||
return f"ToolCall({self.tool_name} {self.tool_call_id})"
|
||||
Reference in New Issue
Block a user