wlw/ai-agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor internal services to use agent-desk package structure

Browse Source 
- Updated import paths in multiple service files to reflect the new agent-desk module.
- Added a new configuration file for agent-desk in the Docker setup.
This commit is contained in:
mlogclub
2026-05-31 18:43:48 +08:00
parent 472b3053b8
commit 101577f163
381 changed files with 1401 additions and 1308 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/rag/answer.go
+8 -8
View File
@@ -8,14 +8,14 @@ import (
"github.com/mlogclub/simple/sqls"
"cs-ai-agent/internal/ai"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/dto"
"cs-ai-agent/internal/pkg/dto/request"
"cs-ai-agent/internal/pkg/dto/response"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/pkg/errorsx"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/ai"
"agent-desk/internal/models"
"agent-desk/internal/pkg/dto"
"agent-desk/internal/pkg/dto/request"
"agent-desk/internal/pkg/dto/response"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/pkg/errorsx"
"agent-desk/internal/repositories"
)
type answer struct {
internal/ai/rag/answer_test.go
+3 -3
View File
@@ -3,9 +3,9 @@ package rag
import (
"testing"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/dto/response"
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/models"
"agent-desk/internal/pkg/dto/response"
"agent-desk/internal/pkg/enums"
)
func TestBuildFallbackAnswer(t *testing.T) {
internal/ai/rag/chunk/fixed_provider.go
+1 -1
View File
@@ -1,8 +1,8 @@
package chunk
import (
"agent-desk/internal/pkg/enums"
"context"
"cs-ai-agent/internal/pkg/enums"
)
type fixedProvider struct{}
internal/ai/rag/chunk/provider.go
+1 -1
View File
@@ -1,8 +1,8 @@
package chunk
import (
"agent-desk/internal/pkg/enums"
"context"
"cs-ai-agent/internal/pkg/enums"
)
type Provider interface {
internal/ai/rag/chunk/registry.go
+1 -1
View File
@@ -1,8 +1,8 @@
package chunk
import (
"agent-desk/internal/pkg/enums"
"context"
"cs-ai-agent/internal/pkg/enums"
"fmt"
)
internal/ai/rag/chunk/structured_provider.go
+1 -1
View File
@@ -1,8 +1,8 @@
package chunk
import (
"agent-desk/internal/pkg/enums"
"context"
"cs-ai-agent/internal/pkg/enums"
"strings"
"github.com/gomarkdown/markdown"
internal/ai/rag/chunk/types.go
+1 -1
View File
@@ -1,6 +1,6 @@
package chunk
import "cs-ai-agent/internal/pkg/enums"
import "agent-desk/internal/pkg/enums"
type ChunkRequest struct {
KnowledgeBaseID int64
internal/ai/rag/chunk/utils.go
+1 -1
View File
@@ -1,8 +1,8 @@
package chunk
import (
"agent-desk/internal/pkg/enums"
"crypto/sha256"
"cs-ai-agent/internal/pkg/enums"
"encoding/hex"
"strings"
"unicode"
internal/ai/rag/index.go
+6 -6
View File
@@ -9,12 +9,12 @@ import (
"log/slog"
"time"
"cs-ai-agent/internal/ai"
ragchunk "cs-ai-agent/internal/ai/rag/chunk"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/ai"
ragchunk "agent-desk/internal/ai/rag/chunk"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/repositories"
"github.com/google/uuid"
"github.com/mlogclub/simple/sqls"
internal/ai/rag/index_cleanup_helpers.go
+2 -2
View File
@@ -5,8 +5,8 @@ import (
"fmt"
"log/slog"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"github.com/mlogclub/simple/common/strs"
"github.com/mlogclub/simple/sqls"
internal/ai/rag/index_document_helpers.go
+6 -5
View File
@@ -6,12 +6,13 @@ import (
"log/slog"
"time"
ragchunk "cs-ai-agent/internal/ai/rag/chunk"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
ragchunk "agent-desk/internal/ai/rag/chunk"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/ai"
"cs-ai-agent/internal/ai"
"github.com/mlogclub/simple/common/strs"
)
internal/ai/rag/index_faq_helpers.go
+4 -4
View File
@@ -5,10 +5,10 @@ import (
"fmt"
"time"
"cs-ai-agent/internal/ai"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/ai"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
)
func buildFAQChunkModel(knowledgeBase models.KnowledgeBase, faq models.KnowledgeFAQ, content string) (models.KnowledgeChunk, string) {
internal/ai/rag/index_flow.go
+3 -3
View File
@@ -3,9 +3,9 @@ package rag
import (
"fmt"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/repositories"
"github.com/mlogclub/simple/sqls"
)
internal/ai/rag/index_retrieve_helpers_test.go
+1 -1
View File
@@ -4,7 +4,7 @@ import (
"strings"
"testing"
"cs-ai-agent/internal/models"
"agent-desk/internal/models"
)
func TestBuildFAQChunkContent(t *testing.T) {
internal/ai/rag/index_run_helpers.go
+4 -4
View File
@@ -4,10 +4,10 @@ import (
"context"
"fmt"
"cs-ai-agent/internal/ai"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/ai"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/repositories"
"github.com/mlogclub/simple/common/strs"
"github.com/mlogclub/simple/sqls"
internal/ai/rag/index_status_helpers.go
+3 -3
View File
@@ -3,9 +3,9 @@ package rag
import (
"time"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/repositories"
"github.com/mlogclub/simple/sqls"
)
internal/ai/rag/index_storage_helpers.go
+2 -2
View File
@@ -5,8 +5,8 @@ import (
"fmt"
"log/slog"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"github.com/mlogclub/simple/sqls"
)
internal/ai/rag/rerank.go
+2 -2
View File
@@ -10,8 +10,8 @@ import (
"sort"
"time"
"cs-ai-agent/internal/ai"
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/ai"
"agent-desk/internal/pkg/enums"
)
type rerank struct{}
internal/ai/rag/retrieve.go
+4 -4
View File
@@ -6,10 +6,10 @@ import (
"log/slog"
"strings"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/repositories"
"github.com/mlogclub/simple/sqls"
)
internal/ai/rag/retrieve_flow.go
+1 -1
View File
@@ -4,7 +4,7 @@ import (
"fmt"
"log/slog"
"cs-ai-agent/internal/models"
"agent-desk/internal/models"
)
func newRetrieveTrace() *RetrieveTrace {
internal/ai/rag/retrieve_log.go
+3 -3
View File
@@ -5,9 +5,9 @@ import (
"fmt"
"time"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/dto"
"cs-ai-agent/internal/pkg/dto/response"
"agent-desk/internal/models"
"agent-desk/internal/pkg/dto"
"agent-desk/internal/pkg/dto/response"
"github.com/google/uuid"
"github.com/mlogclub/simple/sqls"
internal/ai/rag/retrieve_search.go
+5 -5
View File
@@ -8,11 +8,11 @@ import (
"strings"
"time"
"cs-ai-agent/internal/ai"
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/models"
"cs-ai-agent/internal/pkg/enums"
"cs-ai-agent/internal/repositories"
"agent-desk/internal/ai"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/models"
"agent-desk/internal/pkg/enums"
"agent-desk/internal/repositories"
"github.com/mlogclub/simple/sqls"
)
internal/ai/rag/retrieve_test.go
+1 -1
View File
@@ -3,7 +3,7 @@ package rag
import (
"testing"
"cs-ai-agent/internal/models"
"agent-desk/internal/models"
)
func TestResolveKnowledgeBaseSearchOptionsUsesKnowledgeBaseDefaults(t *testing.T) {
internal/ai/rag/utils.go
+1 -1
View File
@@ -1,7 +1,7 @@
package rag
import (
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/pkg/enums"
"strings"
"github.com/yuin/goldmark"
internal/ai/rag/utils_test.go
+2 -2
View File
@@ -1,8 +1,8 @@
package rag
import (
"cs-ai-agent/internal/ai/rag/vectordb"
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/ai/rag/vectordb"
"agent-desk/internal/pkg/enums"
"testing"
)
internal/ai/rag/vectordb/provider.go
+2 -2
View File
@@ -4,8 +4,8 @@ import (
"context"
"fmt"
"cs-ai-agent/internal/pkg/config"
"cs-ai-agent/internal/pkg/enums"
"agent-desk/internal/pkg/config"
"agent-desk/internal/pkg/enums"
)
var defaultProvider Provider
internal/ai/rag/vectordb/qdrant.go
+1 -1
View File
@@ -6,7 +6,7 @@ import (
"github.com/qdrant/go-client/qdrant"
"cs-ai-agent/internal/pkg/config"
"agent-desk/internal/pkg/config"
)
type Vector struct {