PNArchives/Memoh
1
0
Fork 0
mirror of https://github.com/memohai/Memoh.git synced 2026-04-27 07:16:19 +09:00
Code Issues Packages Projects Releases Wiki Activity

feat: search provider

Browse Source
This commit is contained in:
Acbox
2026-02-16 01:22:15 +08:00
parent fc0ac96403
commit 961ecbb96d
47 changed files with 3189 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/db/sqlc/bots.sql.go
+12 -6
View File
@@ -14,7 +14,7 @@ import (
const createBot = `-- name: CreateBot :one
INSERT INTO bots (owner_user_id, type, display_name, avatar_url, is_active, metadata, status)
VALUES ($1, $2, $3, $4, $5, $6, $7)
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
`
type CreateBotParams struct {
@@ -52,6 +52,7 @@ func (q *Queries) CreateBot(ctx context.Context, arg CreateBotParams) (Bot, erro
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
@@ -83,7 +84,7 @@ func (q *Queries) DeleteBotMember(ctx context.Context, arg DeleteBotMemberParams
}
const getBotByID = `-- name: GetBotByID :one
SELECT id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
SELECT id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
FROM bots
WHERE id = $1
`
@@ -105,6 +106,7 @@ func (q *Queries) GetBotByID(ctx context.Context, id pgtype.UUID) (Bot, error) {
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
@@ -169,7 +171,7 @@ func (q *Queries) ListBotMembers(ctx context.Context, botID pgtype.UUID) ([]BotM
}
const listBotsByMember = `-- name: ListBotsByMember :many
SELECT b.id, b.owner_user_id, b.type, b.display_name, b.avatar_url, b.is_active, b.status, b.max_context_load_time, b.language, b.allow_guest, b.chat_model_id, b.memory_model_id, b.embedding_model_id, b.metadata, b.created_at, b.updated_at
SELECT b.id, b.owner_user_id, b.type, b.display_name, b.avatar_url, b.is_active, b.status, b.max_context_load_time, b.language, b.allow_guest, b.chat_model_id, b.memory_model_id, b.embedding_model_id, b.search_provider_id, b.metadata, b.created_at, b.updated_at
FROM bots b
JOIN bot_members m ON m.bot_id = b.id
WHERE m.user_id = $1
@@ -199,6 +201,7 @@ func (q *Queries) ListBotsByMember(ctx context.Context, userID pgtype.UUID) ([]B
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
@@ -214,7 +217,7 @@ func (q *Queries) ListBotsByMember(ctx context.Context, userID pgtype.UUID) ([]B
}
const listBotsByOwner = `-- name: ListBotsByOwner :many
SELECT id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
SELECT id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
FROM bots
WHERE owner_user_id = $1
ORDER BY created_at DESC
@@ -243,6 +246,7 @@ func (q *Queries) ListBotsByOwner(ctx context.Context, ownerUserID pgtype.UUID)
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
@@ -262,7 +266,7 @@ UPDATE bots
SET owner_user_id = $2,
updated_at = now()
WHERE id = $1
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
`
type UpdateBotOwnerParams struct {
@@ -287,6 +291,7 @@ func (q *Queries) UpdateBotOwner(ctx context.Context, arg UpdateBotOwnerParams)
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
@@ -302,7 +307,7 @@ SET display_name = $2,
metadata = $5,
updated_at = now()
WHERE id = $1
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
`
type UpdateBotProfileParams struct {
@@ -336,6 +341,7 @@ func (q *Queries) UpdateBotProfile(ctx context.Context, arg UpdateBotProfilePara
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
&i.Metadata,
&i.CreatedAt,
&i.UpdatedAt,
internal/db/sqlc/conversations.sql.go
+1 -1
View File
@@ -590,7 +590,7 @@ WITH updated AS (
SET display_name = $1,
updated_at = now()
WHERE bots.id = $2
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, metadata, created_at, updated_at
RETURNING id, owner_user_id, type, display_name, avatar_url, is_active, status, max_context_load_time, language, allow_guest, chat_model_id, memory_model_id, embedding_model_id, search_provider_id, metadata, created_at, updated_at
)
SELECT
updated.id AS id,
internal/db/sqlc/models.go
+10
View File
@@ -22,6 +22,7 @@ type Bot struct {
ChatModelID pgtype.UUID `json:"chat_model_id"`
MemoryModelID pgtype.UUID `json:"memory_model_id"`
EmbeddingModelID pgtype.UUID `json:"embedding_model_id"`
SearchProviderID pgtype.UUID `json:"search_provider_id"`
Metadata []byte `json:"metadata"`
CreatedAt pgtype.Timestamptz `json:"created_at"`
UpdatedAt pgtype.Timestamptz `json:"updated_at"`
@@ -202,6 +203,15 @@ type Schedule struct {
BotID pgtype.UUID `json:"bot_id"`
}
type SearchProvider struct {
ID pgtype.UUID `json:"id"`
Name string `json:"name"`
Provider string `json:"provider"`
Config []byte `json:"config"`
CreatedAt pgtype.Timestamptz `json:"created_at"`
UpdatedAt pgtype.Timestamptz `json:"updated_at"`
}
type Snapshot struct {
ID string `json:"id"`
ContainerID string `json:"container_id"`
internal/db/sqlc/search_providers.sql.go
+189
View File
@@ -0,0 +1,189 @@
// Code generated by sqlc. DO NOT EDIT.
// versions:
// sqlc v1.30.0
// source: search_providers.sql
package sqlc
import (
"context"
"github.com/jackc/pgx/v5/pgtype"
)
const createSearchProvider = `-- name: CreateSearchProvider :one
INSERT INTO search_providers (name, provider, config)
VALUES (
$1,
$2,
$3
)
RETURNING id, name, provider, config, created_at, updated_at
`
type CreateSearchProviderParams struct {
Name string `json:"name"`
Provider string `json:"provider"`
Config []byte `json:"config"`
}
func (q *Queries) CreateSearchProvider(ctx context.Context, arg CreateSearchProviderParams) (SearchProvider, error) {
row := q.db.QueryRow(ctx, createSearchProvider, arg.Name, arg.Provider, arg.Config)
var i SearchProvider
err := row.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
)
return i, err
}
const deleteSearchProvider = `-- name: DeleteSearchProvider :exec
DELETE FROM search_providers WHERE id = $1
`
func (q *Queries) DeleteSearchProvider(ctx context.Context, id pgtype.UUID) error {
_, err := q.db.Exec(ctx, deleteSearchProvider, id)
return err
}
const getSearchProviderByID = `-- name: GetSearchProviderByID :one
SELECT id, name, provider, config, created_at, updated_at FROM search_providers WHERE id = $1
`
func (q *Queries) GetSearchProviderByID(ctx context.Context, id pgtype.UUID) (SearchProvider, error) {
row := q.db.QueryRow(ctx, getSearchProviderByID, id)
var i SearchProvider
err := row.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
)
return i, err
}
const getSearchProviderByName = `-- name: GetSearchProviderByName :one
SELECT id, name, provider, config, created_at, updated_at FROM search_providers WHERE name = $1
`
func (q *Queries) GetSearchProviderByName(ctx context.Context, name string) (SearchProvider, error) {
row := q.db.QueryRow(ctx, getSearchProviderByName, name)
var i SearchProvider
err := row.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
)
return i, err
}
const listSearchProviders = `-- name: ListSearchProviders :many
SELECT id, name, provider, config, created_at, updated_at FROM search_providers
ORDER BY created_at DESC
`
func (q *Queries) ListSearchProviders(ctx context.Context) ([]SearchProvider, error) {
rows, err := q.db.Query(ctx, listSearchProviders)
if err != nil {
return nil, err
}
defer rows.Close()
var items []SearchProvider
for rows.Next() {
var i SearchProvider
if err := rows.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const listSearchProvidersByProvider = `-- name: ListSearchProvidersByProvider :many
SELECT id, name, provider, config, created_at, updated_at FROM search_providers
WHERE provider = $1
ORDER BY created_at DESC
`
func (q *Queries) ListSearchProvidersByProvider(ctx context.Context, provider string) ([]SearchProvider, error) {
rows, err := q.db.Query(ctx, listSearchProvidersByProvider, provider)
if err != nil {
return nil, err
}
defer rows.Close()
var items []SearchProvider
for rows.Next() {
var i SearchProvider
if err := rows.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
); err != nil {
return nil, err
}
items = append(items, i)
}
if err := rows.Err(); err != nil {
return nil, err
}
return items, nil
}
const updateSearchProvider = `-- name: UpdateSearchProvider :one
UPDATE search_providers
SET
name = $1,
provider = $2,
config = $3,
updated_at = now()
WHERE id = $4
RETURNING id, name, provider, config, created_at, updated_at
`
type UpdateSearchProviderParams struct {
Name string `json:"name"`
Provider string `json:"provider"`
Config []byte `json:"config"`
ID pgtype.UUID `json:"id"`
}
func (q *Queries) UpdateSearchProvider(ctx context.Context, arg UpdateSearchProviderParams) (SearchProvider, error) {
row := q.db.QueryRow(ctx, updateSearchProvider,
arg.Name,
arg.Provider,
arg.Config,
arg.ID,
)
var i SearchProvider
err := row.Scan(
&i.ID,
&i.Name,
&i.Provider,
&i.Config,
&i.CreatedAt,
&i.UpdatedAt,
)
return i, err
}
internal/db/sqlc/settings.sql.go
+16 -4
View File
@@ -19,6 +19,7 @@ SET max_context_load_time = 1440,
chat_model_id = NULL,
memory_model_id = NULL,
embedding_model_id = NULL,
search_provider_id = NULL,
updated_at = now()
WHERE id = $1
`
@@ -36,11 +37,13 @@ SELECT
bots.allow_guest,
chat_models.model_id AS chat_model_id,
memory_models.model_id AS memory_model_id,
embedding_models.model_id AS embedding_model_id
embedding_models.model_id AS embedding_model_id,
search_providers.id::text AS search_provider_id
FROM bots
LEFT JOIN models AS chat_models ON chat_models.id = bots.chat_model_id
LEFT JOIN models AS memory_models ON memory_models.id = bots.memory_model_id
LEFT JOIN models AS embedding_models ON embedding_models.id = bots.embedding_model_id
LEFT JOIN search_providers ON search_providers.id = bots.search_provider_id
WHERE bots.id = $1
`
@@ -52,6 +55,7 @@ type GetSettingsByBotIDRow struct {
ChatModelID pgtype.Text `json:"chat_model_id"`
MemoryModelID pgtype.Text `json:"memory_model_id"`
EmbeddingModelID pgtype.Text `json:"embedding_model_id"`
SearchProviderID string `json:"search_provider_id"`
}
func (q *Queries) GetSettingsByBotID(ctx context.Context, id pgtype.UUID) (GetSettingsByBotIDRow, error) {
@@ -65,6 +69,7 @@ func (q *Queries) GetSettingsByBotID(ctx context.Context, id pgtype.UUID) (GetSe
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
)
return i, err
}
@@ -78,9 +83,10 @@ WITH updated AS (
chat_model_id = COALESCE($4::uuid, bots.chat_model_id),
memory_model_id = COALESCE($5::uuid, bots.memory_model_id),
embedding_model_id = COALESCE($6::uuid, bots.embedding_model_id),
search_provider_id = COALESCE($7::uuid, bots.search_provider_id),
updated_at = now()
WHERE bots.id = $7
RETURNING bots.id, bots.max_context_load_time, bots.language, bots.allow_guest, bots.chat_model_id, bots.memory_model_id, bots.embedding_model_id
WHERE bots.id = $8
RETURNING bots.id, bots.max_context_load_time, bots.language, bots.allow_guest, bots.chat_model_id, bots.memory_model_id, bots.embedding_model_id, bots.search_provider_id
)
SELECT
updated.id AS bot_id,
@@ -89,11 +95,13 @@ SELECT
updated.allow_guest,
chat_models.model_id AS chat_model_id,
memory_models.model_id AS memory_model_id,
embedding_models.model_id AS embedding_model_id
embedding_models.model_id AS embedding_model_id,
search_providers.id::text AS search_provider_id
FROM updated
LEFT JOIN models AS chat_models ON chat_models.id = updated.chat_model_id
LEFT JOIN models AS memory_models ON memory_models.id = updated.memory_model_id
LEFT JOIN models AS embedding_models ON embedding_models.id = updated.embedding_model_id
LEFT JOIN search_providers ON search_providers.id = updated.search_provider_id
`
type UpsertBotSettingsParams struct {
@@ -103,6 +111,7 @@ type UpsertBotSettingsParams struct {
ChatModelID pgtype.UUID `json:"chat_model_id"`
MemoryModelID pgtype.UUID `json:"memory_model_id"`
EmbeddingModelID pgtype.UUID `json:"embedding_model_id"`
SearchProviderID pgtype.UUID `json:"search_provider_id"`
ID pgtype.UUID `json:"id"`
}
@@ -114,6 +123,7 @@ type UpsertBotSettingsRow struct {
ChatModelID pgtype.Text `json:"chat_model_id"`
MemoryModelID pgtype.Text `json:"memory_model_id"`
EmbeddingModelID pgtype.Text `json:"embedding_model_id"`
SearchProviderID string `json:"search_provider_id"`
}
func (q *Queries) UpsertBotSettings(ctx context.Context, arg UpsertBotSettingsParams) (UpsertBotSettingsRow, error) {
@@ -124,6 +134,7 @@ func (q *Queries) UpsertBotSettings(ctx context.Context, arg UpsertBotSettingsPa
arg.ChatModelID,
arg.MemoryModelID,
arg.EmbeddingModelID,
arg.SearchProviderID,
arg.ID,
)
var i UpsertBotSettingsRow
@@ -135,6 +146,7 @@ func (q *Queries) UpsertBotSettings(ctx context.Context, arg UpsertBotSettingsPa
&i.ChatModelID,
&i.MemoryModelID,
&i.EmbeddingModelID,
&i.SearchProviderID,
)
return i, err
}