From afd9ad0560d76c2a6d161dad52553b10ff428905 Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Thu, 27 Mar 2025 22:35:48 +0100 Subject: rework llm --- internal/db/models.go | 36 +++++++++++++++++++++++------------- 1 file changed, 23 insertions(+), 13 deletions(-) (limited to 'internal/db/models.go') diff --git a/internal/db/models.go b/internal/db/models.go index de9cd3cdb..fc3140a13 100644 --- a/internal/db/models.go +++ b/internal/db/models.go @@ -4,21 +4,31 @@ package db +import ( + "database/sql" +) + type Message struct { - ID string `json:"id"` - SessionID string `json:"session_id"` - MessageData string `json:"message_data"` - CreatedAt int64 `json:"created_at"` - UpdatedAt int64 `json:"updated_at"` + ID string `json:"id"` + SessionID string `json:"session_id"` + Role string `json:"role"` + Content string `json:"content"` + Thinking string `json:"thinking"` + Finished bool `json:"finished"` + ToolCalls sql.NullString `json:"tool_calls"` + ToolResults sql.NullString `json:"tool_results"` + CreatedAt int64 `json:"created_at"` + UpdatedAt int64 `json:"updated_at"` } type Session struct { - ID string `json:"id"` - Title string `json:"title"` - MessageCount int64 `json:"message_count"` - PromptTokens int64 `json:"prompt_tokens"` - CompletionTokens int64 `json:"completion_tokens"` - Cost float64 `json:"cost"` - UpdatedAt int64 `json:"updated_at"` - CreatedAt int64 `json:"created_at"` + ID string `json:"id"` + ParentSessionID sql.NullString `json:"parent_session_id"` + Title string `json:"title"` + MessageCount int64 `json:"message_count"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` + UpdatedAt int64 `json:"updated_at"` + CreatedAt int64 `json:"created_at"` } -- cgit v1.2.3