119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
package turso
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
|
|
"trankilou.fr/lassistanoque/backend/internal/adapter/database/orm"
|
|
"trankilou.fr/lassistanoque/backend/internal/domain"
|
|
)
|
|
|
|
type TursoChatRepository struct {
|
|
db *sql.DB
|
|
chatTable orm.Table[domain.Chat]
|
|
messageTable orm.Table[domain.Message]
|
|
}
|
|
|
|
func NewTursoChatRepository(db *sql.DB) *TursoChatRepository {
|
|
dialect := orm.NewSqliteDialect()
|
|
chatTable, err := orm.NewTable[domain.Chat](db, dialect, "chats")
|
|
messageTable, err := orm.NewTable[domain.Message](db, dialect, "chat_messages")
|
|
if err != nil {
|
|
log.Fatalf("error creating lasebuche agent table")
|
|
}
|
|
|
|
return &TursoChatRepository{
|
|
db: db,
|
|
chatTable: chatTable,
|
|
messageTable: messageTable,
|
|
}
|
|
}
|
|
|
|
func (r *TursoChatRepository) ListChats(userID string, teamID string, page int) ([]*domain.Chat, error) {
|
|
return r.chatTable.Select(
|
|
orm.WithWhere(
|
|
"team_id=$1 and team_id in (select team_id from user_teams where user_id=$2)",
|
|
teamID,
|
|
userID,
|
|
),
|
|
orm.WithOrder("_date_created desc"),
|
|
orm.WithPagination(page, 50),
|
|
)
|
|
}
|
|
|
|
func (r *TursoChatRepository) ListChatsWithUnfreshTitles(teamID string) ([]*domain.Chat, error) {
|
|
return r.chatTable.Select(
|
|
orm.WithWhere(
|
|
"team_id=$1 and fresh_title=0",
|
|
teamID,
|
|
),
|
|
)
|
|
}
|
|
|
|
func (r *TursoChatRepository) GetChat(userID string, teamID string, id string) (*domain.Chat, error) {
|
|
return r.chatTable.SelectOne(
|
|
"id=$1 and team_id=$2 and team_id in (select team_id from user_teams where user_id=$3)",
|
|
id,
|
|
teamID,
|
|
userID,
|
|
)
|
|
}
|
|
|
|
func (r *TursoChatRepository) CreateChat(userID string, chat *domain.Chat) (*domain.Chat, error) {
|
|
return r.chatTable.Insert(chat)
|
|
}
|
|
|
|
func (r *TursoChatRepository) UpdateChat(userID string, chat *domain.Chat) (*domain.Chat, error) {
|
|
return r.chatTable.Update(chat)
|
|
}
|
|
|
|
func (r *TursoChatRepository) DeleteChat(userID string, teamID string, id string) error {
|
|
return r.chatTable.DeleteWhere(
|
|
"id=$1 and team_id=$2 and team_id in (select team_id from user_teams where user_id=$3)",
|
|
id,
|
|
teamID,
|
|
userID,
|
|
)
|
|
}
|
|
|
|
func (r *TursoChatRepository) GetChatMessages(userID string, teamID string, chatID string, withToolCallResponses bool) ([]*domain.Message, error) {
|
|
var messages []*domain.Message
|
|
var err error
|
|
if !withToolCallResponses {
|
|
messages, err = r.messageTable.Select(
|
|
orm.WithWhere(
|
|
`chat_id=$1
|
|
and team_id=$2
|
|
and team_id in (select team_id from user_teams where user_id=$3)
|
|
and (tool_call_id='' or tool_call_id is null)`,
|
|
chatID,
|
|
teamID,
|
|
userID,
|
|
),
|
|
orm.WithOrder("_date_created asc"),
|
|
)
|
|
} else {
|
|
messages, err = r.messageTable.Select(
|
|
orm.WithWhere(
|
|
"chat_id=$1 and team_id=$2 and team_id in (select team_id from user_teams where user_id=$3)",
|
|
chatID,
|
|
teamID,
|
|
userID,
|
|
),
|
|
orm.WithOrder("_date_created asc"),
|
|
)
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, m := range messages {
|
|
m.Json2toolCall()
|
|
}
|
|
return messages, nil
|
|
}
|
|
|
|
func (r *TursoChatRepository) CreateChatMessage(userID string, message *domain.Message) (*domain.Message, error) {
|
|
message.ToolCall2Json()
|
|
return r.messageTable.Insert(message)
|
|
}
|