refactor: 重构advice模块及添加MongoDB支持

This commit is contained in:
renzhiyuan 2026-02-05 13:37:42 +08:00
parent 6fedb76631
commit ec5ff4a0a9
41 changed files with 2237 additions and 603 deletions

View File

@ -10,6 +10,7 @@ import (
"ai_scheduler/internal/biz/tools_regis"
"ai_scheduler/internal/config"
"ai_scheduler/internal/data/impl"
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/domain/component"
"ai_scheduler/internal/domain/repo"
"ai_scheduler/internal/domain/workflow"
@ -27,7 +28,7 @@ import (
)
// InitializeApp 初始化应用程序
func InitializeApp(ctx context.Context, *config.Config, log.AllLogger) (*server.Servers, func(), error) {
func InitializeApp(context.Context, *config.Config, log.AllLogger) (*server.Servers, func(), error) {
panic(wire.Build(
server.ProviderSetServer,
workflow.ProviderSetWorkflow,
@ -43,6 +44,7 @@ func InitializeApp(ctx context.Context, *config.Config, log.AllLogger) (*server.
// tool_callback.ProviderSetCallBackTools,
component.ProviderSet,
repo.ProviderSet,
mongo_model.ProviderSetMongo,
))
}

View File

@ -56,6 +56,14 @@ redis:
db:
driver: mysql
source: root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
mongo:
source: mongodb://root:lsxd2026123@192.168.6.115:27017
dataBase: ai_scheduler
maxPoolSize: 100
minPoolSize: 10
maxConnIdleTime: 30
connectTimeout: 10
socketTimeout: 30
oss:
access_key: "LTAI5tGGZzjf3tvqWk8SQj2G"
secret_key: "S0NKOAUaYWoK4EGSxrMFmYDzllhvpq"

View File

@ -50,6 +50,14 @@ redis:
db:
driver: mysql
source: root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai_test?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
mongo:
source: mongodb://root:lsxd2026123@192.168.6.115:27017
dataBase: ai_scheduler_test
maxPoolSize: 100
minPoolSize: 10
maxConnIdleTime: 30
connectTimeout: 10
socketTimeout: 30
oss:
access_key: "LTAI5tGGZzjf3tvqWk8SQj2G"
secret_key: "S0NKOAUaYWoK4EGSxrMFmYDzllhvpq"

View File

@ -53,7 +53,8 @@ db:
driver: mysql
source: root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai_test?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
mongo:
source: root:SD###sdf323r343@tcp(121.199.38.107:3306)/sys_ai_test?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
source: mongodb://root:lsxd2026123@192.168.6.115:27017
dataBase: ai_scheduler_test
maxPoolSize: 100
minPoolSize: 10
maxConnIdleTime: 30

View File

@ -7,7 +7,7 @@ services:
container_name: mysql_db
restart: unless-stopped
environment:
MYSQL_ROOT_PASSWORD: ${MYSQL_ROOT_PASSWORD:-rootpassword123}
MYSQL_ROOT_PASSWORD: ${MYSQL_ROOT_PASSWORD:-lsxd2026}
MYSQL_DATABASE: ${MYSQL_DATABASE:-myapp}
MYSQL_USER: ${MYSQL_USER:-myuser}
MYSQL_PASSWORD: ${MYSQL_PASSWORD:-mypassword}
@ -36,9 +36,8 @@ services:
- "27017:27017"
volumes:
- mongodb_data:/data/db
- ./mongodb/mongod.conf:/etc/mongod.conf
command:
--config /etc/mongod.conf
--auth
--bind_ip_all # 允许所有IP连接
networks:
- ai_scheduler_network

11
go.mod
View File

@ -26,7 +26,6 @@ require (
github.com/gofiber/websocket/v2 v2.2.1
github.com/google/uuid v1.6.0
github.com/google/wire v0.7.0
github.com/lukasjarosch/go-docx v0.5.0
github.com/ollama/ollama v0.12.7
github.com/redis/go-redis/v9 v9.16.0
github.com/robfig/cron/v3 v3.0.1
@ -34,9 +33,10 @@ require (
github.com/spf13/viper v1.17.0
github.com/stretchr/testify v1.11.1
github.com/tmc/langchaingo v0.1.13
github.com/unidoc/unioffice v1.39.0
github.com/valyala/fasthttp v1.51.0
github.com/volcengine/volcengine-go-sdk v1.2.9
github.com/xuri/excelize/v2 v2.10.0
go.mongodb.org/mongo-driver v1.14.0
golang.org/x/sync v0.17.0
google.golang.org/grpc v1.64.0
gorm.io/driver/mysql v1.6.0
@ -73,6 +73,7 @@ require (
github.com/fsnotify/fsnotify v1.6.0 // indirect
github.com/go-sql-driver/mysql v1.8.1 // indirect
github.com/golang-jwt/jwt/v5 v5.2.2 // indirect
github.com/golang/snappy v0.0.4 // indirect
github.com/goph/emperror v0.17.2 // indirect
github.com/gorilla/websocket v1.5.3 // indirect
github.com/hashicorp/hcl v1.0.0 // indirect
@ -92,6 +93,7 @@ require (
github.com/mitchellh/mapstructure v1.5.0 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe // indirect
github.com/nikolalohinski/gonja v1.5.3 // indirect
github.com/pelletier/go-toml/v2 v2.2.2 // indirect
github.com/pkg/errors v0.9.1 // indirect
@ -114,13 +116,16 @@ require (
github.com/tjfoc/gmsm v1.4.1 // indirect
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
github.com/valyala/bytebufferpool v1.0.0 // indirect
github.com/valyala/fasthttp v1.51.0 // indirect
github.com/valyala/tcplisten v1.0.0 // indirect
github.com/volcengine/volc-sdk-golang v1.0.23 // indirect
github.com/wk8/go-ordered-map/v2 v2.1.8 // indirect
github.com/xdg-go/pbkdf2 v1.0.0 // indirect
github.com/xdg-go/scram v1.1.2 // indirect
github.com/xdg-go/stringprep v1.0.4 // indirect
github.com/xuri/efp v0.0.1 // indirect
github.com/xuri/nfp v0.0.2-0.20250530014748-2ddeb826f9a9 // indirect
github.com/yargevad/filepathx v1.0.0 // indirect
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect
go.uber.org/atomic v1.9.0 // indirect
go.uber.org/multierr v1.9.0 // indirect
golang.org/x/arch v0.11.0 // indirect

20
go.sum
View File

@ -239,6 +239,8 @@ github.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QD
github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
github.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=
github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
github.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM=
github.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
github.com/google/btree v0.0.0-20180813153112-4030bb1f1f0c/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
github.com/google/btree v1.0.0/go.mod h1:lNA+9X1NB3Zf8V7Ke586lFgjr2dZNuvo3lPJSGZ5JPQ=
github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=
@ -327,8 +329,6 @@ github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
github.com/lukasjarosch/go-docx v0.5.0 h1:4vU+gJ4WMdqwRvRVFF+XMw3rPfUGSXlToPJIX3mHQsQ=
github.com/lukasjarosch/go-docx v0.5.0/go.mod h1:ka/NZgDIJId48vMvcfWfduVTY7uV0/f8EgsmCjuS9X0=
github.com/magiconair/properties v1.8.7 h1:IeQXZAiQcpL9mgcAe1Nu6cX9LLw6ExEHKjN0VQdvPDY=
github.com/magiconair/properties v1.8.7/go.mod h1:Dhd985XPs7jluiymwWYZ0G4Z61jb3vdS329zhj2hYo0=
github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0=
@ -354,6 +354,8 @@ github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lN
github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6fbMZ9s0scYfZQ84/6SPL6zC8ACM2oIL0=
github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=
github.com/nikolalohinski/gonja v1.5.3 h1:GsA+EEaZDZPGJ8JtpeGN78jidhOlxeJROpqMT9fTj9c=
github.com/nikolalohinski/gonja v1.5.3/go.mod h1:RmjwxNiXAEqcq1HeK5SSMmqFJvKOfTfXhkJv6YBtPa4=
@ -454,8 +456,6 @@ github.com/tmc/langchaingo v0.1.13 h1:rcpMWBIi2y3B90XxfE4Ao8dhCQPVDMaNPnN5cGB1Ca
github.com/tmc/langchaingo v0.1.13/go.mod h1:vpQ5NOIhpzxDfTZK9B6tf2GM/MoaHewPWM5KXXGh7hg=
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
github.com/unidoc/unioffice v1.39.0 h1:Wo5zvrzCqhyK/1Zi5dg8a5F5+NRftIMZPnFPYwruLto=
github.com/unidoc/unioffice v1.39.0/go.mod h1:Axz6ltIZZTUUyHoEnPe4Mb3VmsN4TRHT5iZCGZ1rgnU=
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
github.com/valyala/fasthttp v1.51.0 h1:8b30A5JlZ6C7AS81RsWjYMQmrZG6feChmgAolCl1SqA=
@ -470,6 +470,12 @@ github.com/wk8/go-ordered-map/v2 v2.1.8 h1:5h/BUHu93oj4gIdvHHHGsScSTMijfx5PeYkE/
github.com/wk8/go-ordered-map/v2 v2.1.8/go.mod h1:5nJHM5DyteebpVlHnWMV0rPz6Zp7+xBAnxjb1X5vnTw=
github.com/x-cray/logrus-prefixed-formatter v0.5.2 h1:00txxvfBM9muc0jiLIEAkAcIMJzfthRT6usrui8uGmg=
github.com/x-cray/logrus-prefixed-formatter v0.5.2/go.mod h1:2duySbKsL6M18s5GU7VPsoEPHyzalCE06qoARUCeBBE=
github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
github.com/xdg-go/scram v1.1.2 h1:FHX5I5B4i4hKRVRBCFRxq1iQRej7WO3hhBuJf+UUySY=
github.com/xdg-go/scram v1.1.2/go.mod h1:RT/sEzTbU5y00aCK8UOx6R7YryM0iF1N2MOmC3kKLN4=
github.com/xdg-go/stringprep v1.0.4 h1:XLI/Ng3O1Atzq0oBs3TWm+5ZVgkq2aqdlvP9JtoZ6c8=
github.com/xdg-go/stringprep v1.0.4/go.mod h1:mPGuuIYwz7CmR2bT9j4GbQqutWS1zV24gijq1dTyGkM=
github.com/xuri/efp v0.0.1 h1:fws5Rv3myXyYni8uwj2qKjVaRP30PdjeYe2Y6FDsCL8=
github.com/xuri/efp v0.0.1/go.mod h1:ybY/Jr0T0GTCnYjKqmdwxyxn2BQf2RcQIIvex5QldPI=
github.com/xuri/excelize/v2 v2.10.0 h1:8aKsP7JD39iKLc6dH5Tw3dgV3sPRh8uRVXu/fMstfW4=
@ -478,6 +484,8 @@ github.com/xuri/nfp v0.0.2-0.20250530014748-2ddeb826f9a9 h1:+C0TIdyyYmzadGaL/HBL
github.com/xuri/nfp v0.0.2-0.20250530014748-2ddeb826f9a9/go.mod h1:WwHg+CVyzlv/TX9xqBFXEZAuxOPxn2k1GNHwG41IIUQ=
github.com/yargevad/filepathx v1.0.0 h1:SYcT+N3tYGi+NvazubCNlvgIPbzAk7i7y2dwg3I5FYc=
github.com/yargevad/filepathx v1.0.0/go.mod h1:BprfX/gpYNJHJfc35GjRRpVcwWXS89gGulUIU5tK3tA=
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 h1:ilQV1hzziu+LLM3zUTJ0trRztfwgjqKnBWNtSRkbmwM=
github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78/go.mod h1:aL8wCCfTfSfmXjznFBSZNN13rSJjlIOI1fUNAtF7rmI=
github.com/yuin/goldmark v1.1.25/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
github.com/yuin/goldmark v1.1.30/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
@ -485,6 +493,8 @@ github.com/yuin/goldmark v1.1.32/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9de
github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
go.etcd.io/bbolt v1.3.4/go.mod h1:G5EMThwa9y8QZGBClrRx5EY+Yw9kAhnjy3bSjsnlVTQ=
go.mongodb.org/mongo-driver v1.14.0 h1:P98w8egYRjYe3XDjxhYJagTokP/H6HzlsnojRgZRd80=
go.mongodb.org/mongo-driver v1.14.0/go.mod h1:Vzb0Mk/pa7e6cWw85R4F/endUC3u0U9jGcNU603k65c=
go.opencensus.io v0.21.0/go.mod h1:mSImk1erAIZhrmZN+AvHh14ztQfjbGwt4TtuofqLduU=
go.opencensus.io v0.22.0/go.mod h1:+kGneAE2xo2IficOXnaByMWTGM9T73dGwxeWcUqIpI8=
go.opencensus.io v0.22.2/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw=
@ -584,7 +594,6 @@ golang.org/x/net v0.0.0-20200520182314-0ba52f642ac2/go.mod h1:qpuaurCH72eLCgpAm/
golang.org/x/net v0.0.0-20200625001655-4c5254603344/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
golang.org/x/net v0.0.0-20200707034311-ab3426394381/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
golang.org/x/net v0.0.0-20200822124328-c89045814202/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
golang.org/x/net v0.0.0-20200925080053-05aa5d4ee321/go.mod h1:/O7V0waA8r7cgGh81Ro3o1hOxt32SMVPicZroKQ2sZA=
golang.org/x/net v0.0.0-20201010224723-4f7140c49acb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
golang.org/x/net v0.0.0-20201031054903-ff519b6c9102/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
@ -699,6 +708,7 @@ golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.4/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=

View File

@ -3,26 +3,35 @@ package biz
import (
"ai_scheduler/internal/data/impl"
"ai_scheduler/internal/data/model"
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"errors"
"fmt"
"time"
"context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"xorm.io/builder"
)
type AdviceAdvicerBiz struct {
advicerImpl *impl.AdviceAdvicerImpl
adviceAdvicerVersionImpl *impl.AdviceAdvicerVersionImpl
advicerImpl *impl.AdviceAdvicerImpl
advicerVersionMongo *mongo_model.AdvicerVersionMongo
mongo *pkg.Mongo
}
func NewAdviceAdvicerBiz(
advicerImpl *impl.AdviceAdvicerImpl,
adviceAdvicerVersionImpl *impl.AdviceAdvicerVersionImpl,
advicerVersionMongo *mongo_model.AdvicerVersionMongo,
mongo *pkg.Mongo,
) *AdviceAdvicerBiz {
return &AdviceAdvicerBiz{
advicerImpl: advicerImpl,
adviceAdvicerVersionImpl: adviceAdvicerVersionImpl,
advicerImpl: advicerImpl,
advicerVersionMongo: advicerVersionMongo,
mongo: mongo,
}
}
@ -33,6 +42,7 @@ func (a *AdviceAdvicerBiz) Update(ctx context.Context, data *entitys.AdvicerInit
}
param := &model.AiAdviceAdvicer{
AdvicerID: data.AdvicerID,
ProjectID: data.ProjectID,
Name: data.Name,
Birth: birth,
Gender: data.Gender,
@ -56,21 +66,143 @@ func (a *AdviceAdvicerBiz) List(ctx context.Context, data *entitys.AdvicerListRe
return list, err
}
func (a *AdviceAdvicerBiz) VersionUpdate(ctx context.Context, param *entitys.AdvicerVersionInitReq) (err error) {
if param.VersionID == 0 {
_, err = a.adviceAdvicerVersionImpl.Add(param)
} else {
cond := builder.NewCond()
cond = cond.And(builder.Eq{"version_id": param.VersionID})
err = a.adviceAdvicerVersionImpl.UpdateByCond(&cond, param)
func (a *AdviceAdvicerBiz) VersionAdd(ctx context.Context, param *entitys.AdvicerVersionAddReq) (err error) {
cond := builder.NewCond()
cond = cond.And(builder.Eq{"advicer_id": param.AdvicerID})
_, err = a.advicerImpl.GetOneBySearch(&cond)
if err != nil {
return errors.New("顾问不存在")
}
_, err = a.mongo.Co(a.advicerVersionMongo).InsertOne(ctx, &mongo_model.AdvicerVersionMongo{
AdvicerId: param.AdvicerID,
VersionDesc: param.VersionDesc,
DialectFeatures: param.DialectFeatures,
SentencePatterns: param.SentencePatterns,
ToneTags: param.ToneTags,
PersonalityTags: param.PersonalityTags,
SignatureDialogues: param.SignatureDialogues,
LastUpdateTime: time.Now(),
})
return err
}
func (a *AdviceAdvicerBiz) VersionList(ctx context.Context, data *entitys.AdvicerVersionListReq) ([]map[string]interface{}, error) {
cond := builder.NewCond()
cond = cond.And(builder.Eq{"advicer_id": data.AdvicerID})
list, err := a.adviceAdvicerVersionImpl.GetRange(&cond)
func (a *AdviceAdvicerBiz) VersionUpdate(ctx context.Context, param *entitys.AdvicerVersionUpdateReq) (err error) {
filter := bson.M{}
if len(param.Id) == 0 {
return errors.New("ID不能为空")
}
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
update := bson.M{
"$set": &mongo_model.AdvicerVersionMongo{
AdvicerId: param.AdvicerID,
VersionDesc: param.VersionDesc,
DialectFeatures: param.DialectFeatures,
SentencePatterns: param.SentencePatterns,
ToneTags: param.ToneTags,
PersonalityTags: param.PersonalityTags,
SignatureDialogues: param.SignatureDialogues,
LastUpdateTime: time.Now(),
},
}
res := a.mongo.Co(a.advicerVersionMongo).FindOneAndUpdate(ctx, filter, update)
return res.Err()
}
func (a *AdviceAdvicerBiz) VersionList(ctx context.Context, param *entitys.AdvicerVersionListReq) (list []mongo_model.AdvicerVersionMongo, err error) {
filter := bson.M{}
// 1. advicer_id 条件
if param.AdvicerId != 0 {
filter["AdvicerId"] = param.AdvicerId
}
// 2. _id 条件
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return nil, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
// 3. version_desc 模糊查询
if len(param.VersionDesc) != 0 {
// 正确的方式:指定字段名
filter["VersionDesc"] = bson.M{
"$regex": primitive.Regex{
Pattern: param.VersionDesc,
Options: "i",
},
}
}
cursor, err := a.mongo.Co(a.advicerVersionMongo).Find(ctx, filter)
if err != nil {
return nil, err
}
// 遍历结果
for cursor.Next(ctx) {
var advicerVersion mongo_model.AdvicerVersionMongo
if err := cursor.Decode(&advicerVersion); err != nil {
return nil, err
}
list = append(list, advicerVersion)
}
if err := cursor.Err(); err != nil {
return nil, err
}
return list, err
}
func (a *AdviceAdvicerBiz) VersionDel(ctx context.Context, param *entitys.AdvicerVersionDelReq) (err error) {
filter := bson.M{}
// 1. advicer_id 条件
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
_, err = a.mongo.Co(a.advicerVersionMongo).DeleteOne(ctx, filter)
return err
}
func (a *AdviceAdvicerBiz) VersionInfo(ctx context.Context, param *entitys.AdvicerVersionInfoReq) (info mongo_model.AdvicerVersionMongo, err error) {
filter := bson.M{}
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return info, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
res := a.mongo.Co(a.advicerVersionMongo).FindOne(ctx, filter)
if res.Err() != nil {
return info, res.Err()
}
if err := res.Decode(&info); err != nil {
return info, err
}
return info, nil
}
func (a *AdviceAdvicerBiz) AdvicerInfo(ctx context.Context, param *entitys.AdvicerInfoReq) (info model.AiAdviceAdvicer, err error) {
cond := builder.NewCond()
cond = cond.And(builder.Eq{"advicer_id": param.AdvicerID})
err = a.advicerImpl.GetOneBySearchToStrut(&cond, &info)
return
}

124
internal/biz/advice_chat.go Normal file
View File

@ -0,0 +1,124 @@
package biz
import (
"ai_scheduler/internal/biz/llm_service/third_party"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"ai_scheduler/utils"
"context"
"encoding/json"
"strings"
"time"
"github.com/google/uuid"
"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
"github.com/volcengine/volcengine-go-sdk/volcengine"
)
type AdviceChatBiz struct {
hsyq *third_party.Hsyq
rdb *utils.Rdb
}
func NewAdviceChatBiz(
hsyq *third_party.Hsyq,
rdb *utils.Rdb,
) *AdviceChatBiz {
return &AdviceChatBiz{
hsyq: hsyq,
rdb: rdb,
}
}
func (a *AdviceChatBiz) Regis(ctx context.Context, chatData *entitys.ChatData) (string, error) {
sessionId := uuid.New().String()
prompt, err := a.buildBasePrompt(ctx, chatData)
if err != nil {
return "", err
}
err = a.rdb.Rdb.SetEx(ctx, sessionId, pkg.JsonStringIgonErr(prompt), 3600*time.Second).Err()
return sessionId, err
}
func (a *AdviceChatBiz) Chat(ctx context.Context, chat *entitys.AdvicerChatReq) ([]string, error) {
if len(chat.Content) == 0 {
return nil, nil
}
basePromptJson, err := a.getChatDataFromStringSessionId(ctx, chat.SessionId)
if err != nil {
return nil, err
}
prompt, err := a.setContent(ctx, basePromptJson, chat.Content)
if err != nil {
return nil, err
}
resContent, err := a.callLlm(ctx, prompt, fileModel)
if err != nil {
return nil, err
}
resSlice := strings.Split(resContent, "\n")
return resSlice, nil
}
func (a *AdviceChatBiz) buildBasePrompt(ctx context.Context, chatData *entitys.ChatData) ([]*model.ChatCompletionMessage, error) {
var message = make([]*model.ChatCompletionMessage, 3)
message[0] = &model.ChatCompletionMessage{
Role: model.ChatMessageRoleSystem,
Content: &model.ChatCompletionMessageContent{
StringValue: volcengine.String(a.sysPrompt(chatData)),
},
}
message[1] = &model.ChatCompletionMessage{
Role: model.ChatMessageRoleUser,
Content: &model.ChatCompletionMessageContent{
StringValue: volcengine.String("{{chat_content}}"),
},
}
message[2] = &model.ChatCompletionMessage{
Role: model.ChatMessageRoleAssistant,
Content: &model.ChatCompletionMessageContent{
StringValue: volcengine.String(a.assistantPrompt(chatData)),
},
}
return message, nil
}
func (a *AdviceChatBiz) setContent(ctx context.Context, basePromptJson string, content string) ([]*model.ChatCompletionMessage, error) {
promptJson := strings.ReplaceAll(basePromptJson, "{{chat_content}}", content)
var basePrompt []*model.ChatCompletionMessage
err := json.Unmarshal([]byte(promptJson), &basePrompt)
if err != nil {
return nil, err
}
return basePrompt, nil
}
func (a *AdviceChatBiz) sysPrompt(chatData *entitys.ChatData) string {
return "【场景】现在你是专业的顾问我们在聊天软件里面聊天当前时间是2026/01/30 16:13:30\n【要求】提供的json内容包含顾问项目对话技巧客户信息请根据json内容进行模仿真人进行对话** 绝对不能让对方看出来你是ai!**\n【规则】\n1.必须模仿顾问的聊天方式不能有markdown格式聊天内容必须贴近聊天软件的正常聊天非客户要求不要自顾自的介绍项目开场白不要太长\n2.不要有任何特殊格式,有比较长的对话内容注意提行\n3.尽量邀约对方来项目上来\n4.不能有任何画外音的文字格式出现!比如(等待对方回复)"
}
func (a *AdviceChatBiz) assistantPrompt(chatData *entitys.ChatData) string {
return pkg.JsonStringIgonErr(chatData)
}
func (a *AdviceChatBiz) getChatDataFromStringSessionId(ctx context.Context, sessionId string) (basePromptJson string, err error) {
cache := a.rdb.Rdb.Get(ctx, sessionId)
if cache.Err() != nil {
err = cache.Err()
return
}
return cache.Val(), cache.Err()
}
func (a *AdviceChatBiz) callLlm(ctx context.Context, prompt []*model.ChatCompletionMessage, modelName string) (string, error) {
res, err := a.hsyq.RequestHsyq(ctx, key, modelName, prompt)
if err != nil {
return "", err
}
return *res.Choices[0].Message.Content.StringValue, nil
}

View File

@ -0,0 +1,150 @@
package biz
import (
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"errors"
"fmt"
"time"
"context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type AdviceClientBiz struct {
AdvicerClientMongo *mongo_model.AdvicerClientMongo
mongo *pkg.Mongo
}
func NewAdviceClientBiz(
advicerClientMongo *mongo_model.AdvicerClientMongo,
mongo *pkg.Mongo,
) *AdviceClientBiz {
return &AdviceClientBiz{
AdvicerClientMongo: advicerClientMongo,
mongo: mongo,
}
}
func (a *AdviceClientBiz) Add(ctx context.Context, param *entitys.AdvicerClientAddReq) (err error) {
_, err = a.mongo.Co(a.AdvicerClientMongo).InsertOne(ctx, &mongo_model.AdvicerClientMongo{
ProjectId: param.ProjectId,
AdvicerId: param.AdvicerId,
PersonalInfo: param.PersonalInfo,
PurchasePurpose: param.PurchasePurpose,
CoreDemands: param.CoreDemands,
Concerns: param.Concerns,
DecisionProfile: param.DecisionProfile,
LastUpdateTime: time.Now(),
})
return err
}
func (a *AdviceClientBiz) Update(ctx context.Context, param *entitys.AdvicerrClientUpdateReq) (err error) {
filter := bson.M{}
if len(param.Id) == 0 {
return errors.New("ID不能为空")
}
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
update := bson.M{
"$set": &mongo_model.AdvicerClientMongo{
ProjectId: param.ProjectId,
AdvicerId: param.AdvicerId,
PersonalInfo: param.PersonalInfo,
PurchasePurpose: param.PurchasePurpose,
CoreDemands: param.CoreDemands,
Concerns: param.Concerns,
DecisionProfile: param.DecisionProfile,
LastUpdateTime: time.Now(),
},
}
res := a.mongo.Co(a.AdvicerClientMongo).FindOneAndUpdate(ctx, filter, update)
return res.Err()
}
func (a *AdviceClientBiz) List(ctx context.Context, param *entitys.AdvicerClientListReq) (list []mongo_model.AdvicerClientMongo, err error) {
filter := bson.M{}
// 1. advicer_id 条件
if param.AdvicerId != 0 {
filter["AdvicerId"] = param.AdvicerId
}
if param.ProjectId != 0 {
filter["projectId"] = param.ProjectId
}
// 2. _id 条件
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return nil, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
cursor, err := a.mongo.Co(a.AdvicerClientMongo).Find(ctx, filter)
if err != nil {
return nil, err
}
// 遍历结果
for cursor.Next(ctx) {
var advicerVersion mongo_model.AdvicerClientMongo
if err := cursor.Decode(&advicerVersion); err != nil {
return nil, err
}
list = append(list, advicerVersion)
}
if err := cursor.Err(); err != nil {
return nil, err
}
return list, err
}
func (a *AdviceClientBiz) Del(ctx context.Context, param *entitys.AdvicerClientDelReq) (err error) {
filter := bson.M{}
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
_, err = a.mongo.Co(a.AdvicerClientMongo).DeleteOne(ctx, filter)
return err
}
func (a *AdviceClientBiz) Info(ctx context.Context, param *entitys.AdvicerClientInfoReq) (info mongo_model.AdvicerClientMongo, err error) {
filter := bson.M{}
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return info, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
res := a.mongo.Co(a.AdvicerClientMongo).FindOne(ctx, filter)
if res.Err() != nil {
return info, res.Err()
}
if err = res.Decode(&info); err != nil {
return info, err
}
return
}

View File

@ -2,7 +2,7 @@ package biz
import (
"ai_scheduler/internal/biz/llm_service/third_party"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/pkg"
"context"
"encoding/json"
@ -11,7 +11,6 @@ import (
"strings"
"time"
"github.com/gofiber/fiber/v2/log"
"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
"github.com/volcengine/volcengine-go-sdk/volcengine"
)
@ -32,26 +31,26 @@ const (
jsonModel = "doubao-seed-1-6-flash-250828"
)
var DataMap = map[string]entitys.AdviceData{
"dialectFeatures": &entitys.DialectFeatures{},
"sentencePatterns": &entitys.SentencePatterns{},
"personalityTags": &entitys.PersonalityTags{},
"toneTags": &entitys.ToneTags{},
"signatureDialogues": &entitys.SignatureDialogues{},
"regionValue": &entitys.RegionValue{},
"competitionComparison": &entitys.CompetitionComparison{},
"coreSellingPoints": &entitys.CoreSellingPoints{},
"supportingFacilities": &entitys.SupportingFacilities{},
"developerBacking": &entitys.DeveloperBacking{},
"needsMining": &entitys.NeedsMining{},
"painPointResponse": &entitys.PainPointResponse{},
"valueBuilding": &entitys.ValueBuilding{},
"closingTechniques": &entitys.ClosingTechniques{},
"communicationRhythm": &entitys.CommunicationRhythm{},
"customer": &entitys.Customer{},
var DataMap = map[string]mongo_model.AdviceData{
"dialectFeatures": &mongo_model.DialectFeatures{},
"sentencePatterns": &mongo_model.SentencePatterns{},
"personalityTags": &mongo_model.PersonalityTags{},
"toneTags": &mongo_model.ToneTags{},
"signatureDialogues": &mongo_model.SignatureDialogues{},
"regionValue": &mongo_model.RegionValue{},
"competitionComparison": &mongo_model.CompetitionComparison{},
"coreSellingPoints": &mongo_model.CoreSellingPoints{},
"supportingFacilities": &mongo_model.SupportingFacilities{},
"developerBacking": &mongo_model.DeveloperBacking{},
"needsMining": &mongo_model.NeedsMining{},
"painPointResponse": &mongo_model.PainPointResponse{},
"valueBuilding": &mongo_model.ValueBuilding{},
"closingTechniques": &mongo_model.ClosingTechniques{},
"communicationRhythm": &mongo_model.CommunicationRhythm{},
"customer": &mongo_model.Customer{},
}
func (a *AdviceFileBiz) WordAna(ctx context.Context, wordContent string) (map[entitys.AdviceRole]map[string]entitys.AdviceData, error) {
func (a *AdviceFileBiz) WordAna(ctx context.Context, wordContent string) (map[mongo_model.AdviceRole]map[string]mongo_model.AdviceData, error) {
timeSte := time.Now().Format("200601021504")
dir := "./cache/" + timeSte
os.Mkdir(dir, 0755)
@ -81,18 +80,18 @@ func (a *AdviceFileBiz) WordAna(ctx context.Context, wordContent string) (map[en
return resData, err
}
func (a *AdviceFileBiz) cateData(data map[string]entitys.AdviceData) map[entitys.AdviceRole]map[string]entitys.AdviceData {
var res = make(map[entitys.AdviceRole]map[string]entitys.AdviceData)
func (a *AdviceFileBiz) cateData(data map[string]mongo_model.AdviceData) map[mongo_model.AdviceRole]map[string]mongo_model.AdviceData {
var res = make(map[mongo_model.AdviceRole]map[string]mongo_model.AdviceData)
for k, v := range data {
if _, ok := res[v.Role()]; !ok {
res[v.Role()] = make(map[string]entitys.AdviceData)
res[v.Role()] = make(map[string]mongo_model.AdviceData)
}
res[v.Role()][k] = v
}
return res
}
func (a *AdviceFileBiz) parseResponse(ctx context.Context, responseByte []byte) (resultOutPut map[string]entitys.AdviceData, err error) {
func (a *AdviceFileBiz) parseResponse(ctx context.Context, responseByte []byte) (resultOutPut map[string]mongo_model.AdviceData, err error) {
//只尝试修复一次
if isValid := json.Valid(responseByte); !isValid {
responseByte, err = a.fixJson(ctx, responseByte)
@ -108,7 +107,7 @@ func (a *AdviceFileBiz) parseResponse(ctx context.Context, responseByte []byte)
result map[string]interface{}
)
resultOutPut = make(map[string]entitys.AdviceData)
resultOutPut = make(map[string]mongo_model.AdviceData)
if err = json.Unmarshal(responseByte, &result); err != nil {
return
@ -150,20 +149,19 @@ func (a *AdviceFileBiz) callLlm(ctx context.Context, prompt string, modelName st
StringValue: volcengine.String(prompt),
},
}
res, err := a.hsyq.RequestHsyq(ctx, key, modelName, message)
if err != nil {
return "", err
}
log.Info("token用量", res.Usage.TotalTokens)
return *res.Choices[0].Message.Content.StringValue, nil
}
func (a *AdviceFileBiz) getAllExamples() map[string]entitys.AdviceData {
func (a *AdviceFileBiz) getAllExamples() map[string]mongo_model.AdviceData {
return DataMap
}
func (a *AdviceFileBiz) buildSimplePrompt(wordContent string, examples map[string]entitys.AdviceData) string {
func (a *AdviceFileBiz) buildSimplePrompt(wordContent string, examples map[string]mongo_model.AdviceData) string {
// 最简单的提示词模板
template := `分析以下房地产销售对话按指定格式提取信息
@ -182,7 +180,7 @@ func (a *AdviceFileBiz) buildSimplePrompt(wordContent string, examples map[strin
// 构建格式部分
var formats strings.Builder
for name, example := range examples {
formats.WriteString(fmt.Sprintf("=== %s (%s:%s)===\n示例%s\n\n", name, entitys.RoleDesc[example.Role()], example.Desc(), example.Example()))
formats.WriteString(fmt.Sprintf("=== %s (%s:%s)===\n示例%s\n\n", name, mongo_model.RoleDesc[example.Role()], example.Desc(), example.Example()))
}
return fmt.Sprintf(template, wordContent, formats.String())

View File

@ -0,0 +1,100 @@
package biz
import (
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"errors"
"fmt"
"time"
"context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type AdviceProjectBiz struct {
AdvicerProjectMongo *mongo_model.AdvicerProjectMongo
mongo *pkg.Mongo
}
func NewAdviceProjectBiz(
advicerProjectMongo *mongo_model.AdvicerProjectMongo,
mongo *pkg.Mongo,
) *AdviceProjectBiz {
return &AdviceProjectBiz{
AdvicerProjectMongo: advicerProjectMongo,
mongo: mongo,
}
}
func (a *AdviceProjectBiz) Add(ctx context.Context, param *entitys.AdvicerProjectAddReq) (err error) {
_, err = a.mongo.Co(a.AdvicerProjectMongo).InsertOne(ctx, &mongo_model.AdvicerProjectMongo{
ProjectId: param.ProjectId,
ProjectInfo: param.ProjectInfo,
RegionValue: param.RegionValue,
CompetitionComparison: param.CompetitionComparison,
CoreSellingPoints: param.CoreSellingPoints,
SupportingFacilities: param.SupportingFacilities,
DeveloperBacking: param.DeveloperBacking,
LastUpdateTime: time.Now(),
})
return err
}
func (a *AdviceProjectBiz) Update(ctx context.Context, param *entitys.AdvicerrProjectUpdateReq) (err error) {
filter := bson.M{}
if len(param.Id) == 0 {
return errors.New("ID不能为空")
}
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
update := bson.M{
"$set": &mongo_model.AdvicerProjectMongo{
ProjectId: param.ProjectId,
RegionValue: param.RegionValue,
CompetitionComparison: param.CompetitionComparison,
CoreSellingPoints: param.CoreSellingPoints,
SupportingFacilities: param.SupportingFacilities,
DeveloperBacking: param.DeveloperBacking,
LastUpdateTime: time.Now(),
},
}
res := a.mongo.Co(a.AdvicerProjectMongo).FindOneAndUpdate(ctx, filter, update)
return res.Err()
}
func (a *AdviceProjectBiz) Info(ctx context.Context, param *entitys.AdvicerProjectInfoReq) (info mongo_model.AdvicerProjectMongo, err error) {
filter := bson.M{}
if param.ProjectId != 0 {
filter["projectId"] = param.ProjectId
}
// 2. _id 条件
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return info, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
res := a.mongo.Co(a.AdvicerProjectMongo).FindOne(ctx, filter)
if res.Err() != nil {
return info, res.Err()
}
// 遍历结果
if err := res.Decode(&info); err != nil {
return info, err
}
return info, nil
}

View File

@ -0,0 +1,164 @@
package biz
import (
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"errors"
"fmt"
"time"
"context"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type AdviceSkillBiz struct {
AdvicerTalkSkillMongo *mongo_model.AdvicerTalkSkillMongo
mongo *pkg.Mongo
}
func NewAdviceSkillBiz(
advicerTalkSkillMongo *mongo_model.AdvicerTalkSkillMongo,
mongo *pkg.Mongo,
) *AdviceSkillBiz {
return &AdviceSkillBiz{
AdvicerTalkSkillMongo: advicerTalkSkillMongo,
mongo: mongo,
}
}
func (a *AdviceSkillBiz) VersionAdd(ctx context.Context, param *entitys.AdvicerTalkSkillAddReq) (err error) {
_, err = a.mongo.Co(a.AdvicerTalkSkillMongo).InsertOne(ctx, &mongo_model.AdvicerTalkSkillMongo{
ProjectId: param.ProjectId,
AdvicerId: param.AdvicerId,
Desc: param.Desc,
NeedsMining: param.NeedsMining,
PainPointResponse: param.PainPointResponse,
ValueBuilding: param.ValueBuilding,
ClosingTechniques: param.ClosingTechniques,
CommunicationRhythm: param.CommunicationRhythm,
LastUpdateTime: time.Now(),
})
return err
}
func (a *AdviceSkillBiz) VersionUpdate(ctx context.Context, param *entitys.AdvicerTalkSkillUpdateReq) (err error) {
filter := bson.M{}
if len(param.Id) == 0 {
return errors.New("ID不能为空")
}
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
update := bson.M{
"$set": &mongo_model.AdvicerTalkSkillMongo{
AdvicerId: param.AdvicerId,
ProjectId: param.ProjectId,
Desc: param.Desc,
NeedsMining: param.NeedsMining,
PainPointResponse: param.PainPointResponse,
ValueBuilding: param.ValueBuilding,
ClosingTechniques: param.ClosingTechniques,
CommunicationRhythm: param.CommunicationRhythm,
LastUpdateTime: time.Now(),
},
}
res := a.mongo.Co(a.AdvicerTalkSkillMongo).FindOneAndUpdate(ctx, filter, update)
return res.Err()
}
func (a *AdviceSkillBiz) VersionList(ctx context.Context, param *entitys.AdvicerTalkSkillListReq) (list []mongo_model.AdvicerTalkSkillMongo, err error) {
filter := bson.M{}
// 1. advicer_id 条件
if param.AdvicerId != 0 {
filter["AdvicerId"] = param.AdvicerId
}
if param.ProjectId != 0 {
filter["projectId"] = param.ProjectId
}
// 2. _id 条件
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return nil, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
// 3. version_desc 模糊查询
if len(param.Desc) != 0 {
// 正确的方式:指定字段名
filter["desc"] = bson.M{
"$regex": primitive.Regex{
Pattern: param.Desc,
Options: "i",
},
}
}
cursor, err := a.mongo.Co(a.AdvicerTalkSkillMongo).Find(ctx, filter)
if err != nil {
return nil, err
}
// 遍历结果
for cursor.Next(ctx) {
var advicerVersion mongo_model.AdvicerTalkSkillMongo
if err := cursor.Decode(&advicerVersion); err != nil {
return nil, err
}
list = append(list, advicerVersion)
}
if err := cursor.Err(); err != nil {
return nil, err
}
return list, err
}
func (a *AdviceSkillBiz) VersionDel(ctx context.Context, param *entitys.AdvicerTalkSkillDelReq) (err error) {
filter := bson.M{}
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
_, err = a.mongo.Co(a.AdvicerTalkSkillMongo).DeleteOne(ctx, filter)
return err
}
func (a *AdviceSkillBiz) Info(ctx context.Context, param *entitys.AdvicerTalkSkillInfoReq) (info mongo_model.AdvicerTalkSkillMongo, err error) {
filter := bson.M{}
if len(param.Id) != 0 {
objectID, err := primitive.ObjectIDFromHex(param.Id)
if err != nil {
return info, fmt.Errorf("ID转换失败: %w", err)
}
filter["_id"] = objectID
}
res := a.mongo.Co(a.AdvicerTalkSkillMongo).FindOne(ctx, filter)
if res.Err() != nil {
return info, err
}
// 遍历结果
if err = res.Decode(&info); err != nil {
return info, err
}
return info, nil
}

View File

@ -2,10 +2,13 @@ package third_party
import (
"context"
"time"
"github.com/gofiber/fiber/v2/log"
"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model/responses"
)
type Hsyq struct {
@ -47,6 +50,27 @@ func (h *Hsyq) RequestHsyq(ctx context.Context, key string, modelName string, pr
if err != nil {
return model.ChatCompletionResponse{ID: ""}, err
}
log.Info("token用量", resp.Usage.TotalTokens, "输入:", resp.Usage.PromptTokens, "输出:", resp.Usage.CompletionTokens)
return resp, err
}
func (h *Hsyq) RequestHsyqJson(ctx context.Context, key string, modelName string, prompt []*responses.InputItem) (*responses.ResponseObject, error) {
req := responses.ResponsesRequest{
Model: modelName,
Input: &responses.ResponsesInput{
Union: &responses.ResponsesInput_ListValue{
ListValue: &responses.InputItemList{ListValue: prompt},
},
},
Stream: new(bool),
Thinking: &responses.ResponsesThinking{Type: responses.ThinkingType_disabled.Enum()},
Text: &responses.ResponsesText{Format: &responses.TextFormat{Type: responses.TextType_json_object}},
}
resp, err := h.getClient(key).CreateResponses(ctx, &req)
if err != nil {
return resp, err
}
log.Info("token用量", resp.Usage.TotalTokens)
return resp, err
}

View File

@ -25,4 +25,8 @@ var ProviderSetBiz = wire.NewSet(
NewAdviceFileBiz,
third_party.NewHsyq,
NewAdviceAdvicerBiz,
NewAdviceSkillBiz,
NewAdviceProjectBiz,
NewAdviceClientBiz,
NewAdviceChatBiz,
)

View File

@ -158,7 +158,8 @@ type Redis struct {
}
type DB struct {
Driver string `mapstructure:"driver"`
Driver string `mapstructure:"driver"`
Source string `mapstructure:"source"`
MaxIdle int32 `mapstructure:"maxIdle"`
MaxOpen int32 `mapstructure:"maxOpen"`
@ -168,6 +169,7 @@ type DB struct {
type Mongo struct {
Source string `mapstructure:"source"`
DataBase string `mapstructure:"dataBase"`
MaxPoolSize uint64 `mapstructure:"maxPoolSize"`
MinPoolSize uint64 `mapstructure:"minPoolSize"`
MaxConnIdleTime int32 `mapstructure:"maxConnIdleTime"`
@ -319,7 +321,7 @@ func LoadConfigWithEnv() (*Config, error) {
if err != nil {
return nil, err
}
viper.SetConfigFile(modularDir + "/config/config_env.yaml")
viper.SetConfigFile(modularDir + "/config/config_test.yaml")
viper.SetConfigType("yaml")
// 读取配置文件
if err := viper.ReadInConfig(); err != nil {

View File

@ -12,12 +12,13 @@ const TableNameAiAdviceAdvicer = "ai_advice_advicer"
// AiAdviceAdvicer mapped from table <ai_advice_advicer>
type AiAdviceAdvicer struct {
AdvicerID int32 `gorm:"column:advicer_id;primaryKey;autoIncrement:true" json:"advicer_id"`
Name string `gorm:"column:name;not null;comment:姓名" json:"name"` // 姓名
Birth time.Time `gorm:"column:birth;not null;comment:用户名称" json:"birth"` // 用户名称
Gender int32 `gorm:"column:gender;not null;comment:1:男2女" json:"gender"` // 1:男2
WorkingYears int32 `gorm:"column:working_years;not null;default:1;comment:工作年限" json:"working_years"` // 工作年限
CreateAt *time.Time `gorm:"column:create_at;default:CURRENT_TIMESTAMP" json:"create_at"`
AdvicerID int32 `gorm:"column:advicer_id;primaryKey;autoIncrement:true" json:"advicer_id"`
ProjectID int32 `gorm:"column:project_id;not null" json:"project_id"`
Name string `gorm:"column:name;not null;comment:姓名" json:"name"` // 姓名
Birth time.Time `gorm:"column:birth;not null;comment:用户名称" json:"birth"` // 用户名称
Gender int32 `gorm:"column:gender;not null;comment:1:男2女" json:"gender"` // 1:男2
WorkingYears int32 `gorm:"column:working_years;not null;default:1;comment:工作年限" json:"working_years"` // 工作年限
CreateAt time.Time `gorm:"column:create_at;default:CURRENT_TIMESTAMP" json:"create_at"`
}
// TableName AiAdviceAdvicer's table name

View File

@ -0,0 +1,37 @@
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
// Code generated by gorm.io/gen. DO NOT EDIT.
package model
import (
"fmt"
"time"
)
type AiAdviceAdvicerEntity struct {
Name string `json:"name"` // 姓名
Birth string `json:"birth"` // 用户名称
Gender string `json:"gender"` // 1:男2
WorkingYears string `json:"working_years"` // 工作年限
}
func (a *AiAdviceAdvicer) Entity() *AiAdviceAdvicerEntity {
var (
gender string
)
switch a.Gender {
case 1:
gender = "男"
case 2:
gender = "女"
default:
gender = "未知"
}
return &AiAdviceAdvicerEntity{
Name: a.Name,
Birth: a.Birth.Format(time.DateOnly),
Gender: gender,
WorkingYears: fmt.Sprintf("%d年", a.WorkingYears),
}
}

View File

@ -0,0 +1,97 @@
package mongo_model
import (
"time"
)
type AdvicerClientMongo struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
PersonalInfo PersonalInfo `json:"personalInfo" bson:"personalInfo"`
PurchasePurpose PurchasePurpose `json:"purchasePurpose" bson:"purchasePurpose"`
CoreDemands CoreDemands `json:"coreDemands" bson:"coreDemands"`
Concerns []string `json:"concerns" bson:"concerns"`
DecisionProfile []string `json:"decisionProfile" bson:"decisionProfile"`
LastUpdateTime time.Time `json:"lastUpdateTime" bson:"lastUpdateTime"`
}
func NewAdvicerClientMongo() *AdvicerClientMongo {
return &AdvicerClientMongo{}
}
func (a *AdvicerClientMongo) MongoTableName() string {
return "advicer_client"
}
type AdvicerClientMongoEntity struct {
PersonalInfo PersonalInfo `json:"personalInfo"`
PurchasePurpose PurchasePurpose `json:"purchasePurpose"`
CoreDemands CoreDemands `json:"coreDemands"`
Concerns []string `json:"concerns"`
DecisionProfile []string `json:"decisionProfile"`
}
func (a *AdvicerClientMongo) Entity() *AdvicerClientMongoEntity {
return &AdvicerClientMongoEntity{
PersonalInfo: a.PersonalInfo,
PurchasePurpose: a.PurchasePurpose,
CoreDemands: a.CoreDemands,
Concerns: a.Concerns,
DecisionProfile: a.DecisionProfile,
}
}
// Customer 客户信息
type Customer []ClientInfo
type ClientInfo struct {
// 个人信息
PersonalInfo PersonalInfo `json:"personalInfo"`
// 购房目的
PurchasePurpose PurchasePurpose `json:"purchasePurpose"`
// 核心需求
CoreDemands CoreDemands `json:"coreDemands"`
// 关注点与顾虑
Concerns []string `json:"concerns"`
// 决策建议
DecisionProfile []string `json:"decisionProfile"`
}
type PersonalInfo struct {
Name string `json:"name"` // 姓氏
Gender string `json:"gender"` // 性别
Location string `json:"location"` // 来源地/当前居住地
IsFirstHome bool `json:"isFirstHome"` // 是否首套房
FamilyOrganize string `json:"familyOrganize"` // 家庭人数
}
type PurchasePurpose struct {
PrimaryPurpose string `json:"primaryPurpose"` // 主要目的
SecondaryPurpose string `json:"secondaryPurpose"` // 次要目的
DecisionMakers string `json:"decisionMakers"` // 决策人
}
type CoreDemands struct {
TotalBudget string `json:"totalBudget"` // 预算范围
PreferredLayout string `json:"preferredLayout"` // 偏好户型
CoreAppeal string `json:"coreAppeal"` // 核心述求
}
func (e *Customer) Example() string {
return `[{"personalInfo":{"name":"唐","gender":"男","location":"成都北门","isFirstHome":true,"familyOrganize":"夫妻"},"purchasePurpose":{"primaryPurpose":"首次置业,解决自住","secondaryPurpose":"资产保值,未来可出租","decisionMakers":"夫妻双方"},"coreDemands":{"totalBudget":"350-400"","preferredLayout":"118㎡四房三卫双套房","coreAppeal":"在有限预算内满足家庭居住功能,确保房产保值"},"concerns":["总价超预算风险","板块保值能力"],"decisionProfile":["预算导向,严格控制总价","重点关注户型功能性和实用性"]},{"personalInfo":{"name":"冯女士","gender":"女","location":"","isFirstHome":false,"familyOrganize":"夫妻+1孩+父母同住"},"purchasePurpose":{"primaryPurpose":"改善居住条件","secondaryPurpose":"子女教育质量提升","decisionMakers":"夫妻双方需家庭共同商议]},"coreDemands":{"totalBudget":"400-500","preferredLayout":"118㎡四房三卫非全景户型","coreAppeal":"安静舒适、学区有保障的改善型住房"},"concerns":["临路噪音影响老人休息","学区质量和稳定性","社区小,绿化空间有限","得房率是否足够","二八板块学区对比"],"decisionProfile":["对噪音敏感,需要安静环境","重视教育资源配置","关注社区品质和舒适度","需要详细对比不同板块学区优势"]}]`
}
func (e *Customer) Copy() AdviceData {
return new(Customer)
}
func (e *Customer) Role() AdviceRole {
return RoleClient
}
func (e *Customer) Desc() string {
return "客户信息"
}

View File

@ -0,0 +1,150 @@
package mongo_model
import (
"time"
)
type AdvicerProjectMongo struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
ProjectInfo ProjectInfo `json:"projectInfo" bson:"projectInfo"`
RegionValue RegionValue `json:"regionValue" bson:"regionValue"`
CompetitionComparison CompetitionComparison `json:"competitionComparison" bson:"competitionComparison"`
CoreSellingPoints CoreSellingPoints `json:"coreSellingPoints" bson:"coreSellingPoints"`
SupportingFacilities SupportingFacilities `json:"supportingFacilities" bson:"supportingFacilities"`
DeveloperBacking DeveloperBacking `json:"developerBacking" bson:"developerBacking"`
LastUpdateTime time.Time `json:"lastUpdateTime" bson:"lastUpdateTime"`
}
func NewAdvicerProjectMongo() *AdvicerProjectMongo {
return &AdvicerProjectMongo{}
}
func (a *AdvicerProjectMongo) MongoTableName() string {
return "advicer_project"
}
type AdvicerProjectMongoEntity struct {
RegionValue RegionValue `json:"regionValue" bson:"regionValue"`
CompetitionComparison CompetitionComparison `json:"competitionComparison" bson:"competitionComparison"`
CoreSellingPoints CoreSellingPoints `json:"coreSellingPoints" bson:"coreSellingPoints"`
SupportingFacilities SupportingFacilities `json:"supportingFacilities" bson:"supportingFacilities"`
DeveloperBacking DeveloperBacking `json:"developerBacking" bson:"developerBacking"`
}
func (a *AdvicerProjectMongo) Entity() *AdvicerProjectMongoEntity {
return &AdvicerProjectMongoEntity{
RegionValue: a.RegionValue,
CompetitionComparison: a.CompetitionComparison,
CoreSellingPoints: a.CoreSellingPoints,
SupportingFacilities: a.SupportingFacilities,
DeveloperBacking: a.DeveloperBacking,
}
}
type ProjectInfo struct {
Name string `json:"projectName" bson:"projectName"`
Address string `json:"projectAddress" bson:"projectAddress"`
Area string `json:"area" bson:"area"`
HouseTypes []HouseType `json:"houseTypes" bson:"houseTypes"`
}
type HouseType struct {
Name string `json:"name" bson:"name"`
BuildArea string `json:"buildArea" bson:"buildArea"`
InnerArea string `json:"innerArea" bson:"innerArea"`
UnitPrice string `json:"unitPrice" bson:"unitPrice"`
TotalPrice string `json:"totalPrice" bson:"totalPrice"`
}
// RegionValue 区域价值话术库
type RegionValue map[string][]string
func (e *RegionValue) Example() string {
return `{"区位层级":["成华区2.5环内侧,这个位置真的稀缺","槐树店板块现在是成华区的number one板块","北接三板桥商圈,西靠万象城,东临火车东站","属于淮舜板块,万象城东的核心位置"],"地价论证":["我们地价19500华晨府20400棕榈也是2万+","2.5环内现在地价没有低于19000的","面粉贵了,面包不可能便宜"],"板块热度":["从21年新希望锦麟一品开始这边全是高端盘","龙湖最高端的滨江系列在这里,新希望的锦麟系列也在这里","各大品牌开发商争相恐后都在这边拿地"],"发展规划":["槐树店板块是棋盘成钢之后第二个富人区","整个板块都是300万到900万的总价段","未来全是改善型住宅,没有刚需盘"]}`
}
func (e *RegionValue) Copy() AdviceData {
return new(RegionValue)
}
func (e *RegionValue) Role() AdviceRole {
return RoleProject
}
func (e *RegionValue) Desc() string {
return "区域价值话术"
}
// CompetitionComparison 竞品对比话术
type CompetitionComparison map[string]map[string]string
func (e *CompetitionComparison) Example() string {
return `{"龙湖滨江云河颂":{"优点承认":"龙湖位置确实好,看沙河公园","价格对比":"他们单价32000-35000但得房率只有95%套内算下来36000+","优势突出":"我们得房率118平实得132平套内单价才33000"},"邦泰云锦":{"定位相似":"邦泰也是首个项目,要打造口碑","价格参考":"他们当时12800拿地现在卖34000","品质对比":"我们外立面全玻璃幕墙比他们成本高30%"}}`
}
func (e *CompetitionComparison) Copy() AdviceData {
return new(CompetitionComparison)
}
func (e *CompetitionComparison) Role() AdviceRole {
return RoleProject
}
func (e *CompetitionComparison) Desc() string {
return "竞品对比话术"
}
// CoreSellingPoints 核心卖点
type CoreSellingPoints map[string]string
func (e *CoreSellingPoints) Example() string {
return `{"产品配置高端":"全玻璃幕墙+铝单板外立面三层中空氩气玻璃3.2米层高方太Y9烟机灶具高仪卫浴","地段稀缺性":"成华区2.5环内侧核心地段槐树店板块是成华区number one板块被三板桥、万象城、火车东站包围","得房率高":"118平实得132平套内单价33000比龙湖滨江云河颂套内单价低3000"}`
}
func (e *CoreSellingPoints) Copy() AdviceData {
return new(CoreSellingPoints)
}
func (e *CoreSellingPoints) Role() AdviceRole {
return RoleProject
}
func (e *CoreSellingPoints) Desc() string {
return "核心卖点"
}
// SupportingFacilities 配套体系
type SupportingFacilities map[string]map[string]string
func (e *SupportingFacilities) Example() string {
return `{"交通配套":{"地铁":"双店路站350米7号线槐树店站550米4号线未来12号线","道路":"中环路、成洛大道到春熙路5个站","通达性":"到火车东站2个站到华西30分钟内"},"教育配套":{"幼儿园":"楼下公立幼儿园","小学":"城市附小锦汇东城(成华区生源最好的学校)","生源优势":"周边新盘都是300万+,生源纯粹"},"医疗配套":{"三甲医院":"市六医院、市二医院3公里内","顶尖医疗":"华西医院锦江院区30分钟车程","便利性":"到华西本部也是30分钟内"}}`
}
func (e *SupportingFacilities) Copy() AdviceData {
return new(SupportingFacilities)
}
func (e *SupportingFacilities) Role() AdviceRole {
return RoleProject
}
func (e *SupportingFacilities) Desc() string {
return "配套体系"
}
// DeveloperBacking 开发商背书
type DeveloperBacking map[string]string
func (e *DeveloperBacking) Example() string {
return `{"公司实力":"中信资产,多元化民营企业","资金安全":"在河南渑池有铝土矿每年稳定收入10亿","开发经验":"宜宾有5个项目贵州2个成都是首个项目","合作方":"招商铂金物业,首次与外部企业合作"}`
}
func (e *DeveloperBacking) Copy() AdviceData {
return new(DeveloperBacking)
}
func (e *DeveloperBacking) Role() AdviceRole {
return RoleProject
}
func (e *DeveloperBacking) Desc() string {
return "开发商背书"
}

View File

@ -0,0 +1,137 @@
package mongo_model
import (
"time"
)
type AdvicerTalkSkillMongo struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
Desc string `json:"desc" bson:"desc"`
NeedsMining NeedsMining `json:"needsMining" bson:"needsMining"`
PainPointResponse PainPointResponse `json:"painPointResponse" bson:"painPointResponse"`
ValueBuilding ValueBuilding `json:"valueBuilding" bson:"valueBuilding"`
ClosingTechniques ClosingTechniques `json:"closingTechniques" bson:"closingTechniques"`
CommunicationRhythm CommunicationRhythm `json:"communicationRhythm" bson:"communicationRhythm"`
LastUpdateTime time.Time `json:"lastUpdateTime" bson:"lastUpdateTime"`
}
func NewAdvicerTalkSkillMongo() *AdvicerTalkSkillMongo {
return &AdvicerTalkSkillMongo{}
}
func (a *AdvicerTalkSkillMongo) MongoTableName() string {
return "advicer_talk_skill"
}
type AdvicerTalkSkillMongoEntity struct {
NeedsMining NeedsMining `json:"needsMining"`
PainPointResponse PainPointResponse `json:"painPointResponse"`
ValueBuilding ValueBuilding `json:"valueBuilding"`
ClosingTechniques ClosingTechniques `json:"closingTechniques"`
CommunicationRhythm CommunicationRhythm `json:"communicationRhythm"`
}
func (a *AdvicerTalkSkillMongo) Entity() *AdvicerTalkSkillMongoEntity {
return &AdvicerTalkSkillMongoEntity{
NeedsMining: a.NeedsMining,
PainPointResponse: a.PainPointResponse,
ValueBuilding: a.ValueBuilding,
ClosingTechniques: a.ClosingTechniques,
CommunicationRhythm: a.CommunicationRhythm,
}
}
// NeedsMining 需求挖掘话术
type NeedsMining map[string][]string
func (e *NeedsMining) Example() string {
return `{"预算需求":["你们总价想控制在多少以内?","是考虑按揭还是一次性?","月供能接受多少范围?"],"居住需求":["几个人住?有老人小孩吗?","主要是自住还是考虑投资?","现在住哪里?想改善哪些方面?"],"通勤需求":["在哪个位置上班?","主要开车还是坐地铁?","对地铁距离有要求吗?"]}`
}
func (e *NeedsMining) Copy() AdviceData {
return new(NeedsMining)
}
func (e *NeedsMining) Role() AdviceRole {
return RoleSkill
}
func (e *NeedsMining) Desc() string {
return "需求挖掘话术"
}
// PainPointResponse 痛点应对策略
type PainPointResponse map[string]map[string]string
func (e *PainPointResponse) Example() string {
return `{"地块太小":{"承认事实":"14亩确实不大","普遍现象":"2.5环内都是小地块万景13亩中铁建8.8亩","转化优势":"但人少安静,楼间距反而更开阔","对比竞品":"339的邦泰才11亩人家上千万豪宅"},"物业费高":{"理解感受":"我懂你,我们也觉得有点贵","价值分析":"但6块里3块是增值服务保洁、送外卖","价格补贴":"前三年补贴到5块跟其他盘差不多"}}`
}
func (e *PainPointResponse) Copy() AdviceData {
return new(PainPointResponse)
}
func (e *PainPointResponse) Role() AdviceRole {
return RoleSkill
}
func (e *PainPointResponse) Desc() string {
return "痛点应对策略"
}
// ValueBuilding 价值塑造技巧
type ValueBuilding map[string][]string
func (e *ValueBuilding) Example() string {
return `{"地段价值塑造":["买房最重要的是地段、地段、还是地段","核心地段的核心资产才保值增值","2.5环内的地卖一块少一块,不可再生"],"产品价值塑造":["我们是用改善的价格,买豪宅的标准","很多细节都是3000万豪宅才有的配置","外立面成本比竞品高30%,但单价差不多"]}`
}
func (e *ValueBuilding) Copy() AdviceData {
return new(ValueBuilding)
}
func (e *ValueBuilding) Role() AdviceRole {
return RoleSkill
}
func (e *ValueBuilding) Desc() string {
return "价值塑造技巧"
}
// ClosingTechniques 促单话术
type ClosingTechniques map[string]map[string][]string
func (e *ClosingTechniques) Example() string {
return `{"紧迫感营造":{"时间紧迫":["今天是月底最后一天,领导有压力价格可谈","我们刚刚开盘,还有额外优惠","月底冲业绩,价格最有弹性"],"房源稀缺":["118只剩20多套了好楼层不多","这栋楼就60户卖一套少一套","特价房只有这几套,今天不定可能就没了"]},"优惠策略":{"价格优惠":["今天定的话,我可以跟领导申请额外折扣","买车位的话,总价多给两个点优惠","一次性付款再优惠一个点"],"附加价值":["送一年物业费","送品牌家电礼包","优先选车位"]},"决策推动":{"小步推进":["要不先交个小定保留房源?","可以先排个号,有优惠优先通知你","今天不定的话,我帮你留意好楼层"]}}`
}
func (e *ClosingTechniques) Copy() AdviceData {
return new(ClosingTechniques)
}
func (e *ClosingTechniques) Role() AdviceRole {
return RoleSkill
}
func (e *ClosingTechniques) Desc() string {
return "促单话术"
}
// CommunicationRhythm 沟通节奏控制
type CommunicationRhythm map[string]map[string]string
func (e *CommunicationRhythm) Example() string {
return `{"开场阶段":{"时间占比":"5%","目标":"建立关系,了解需求","关键动作":"亲切称呼,简单寒暄,确认看房重点"},"沙盘讲解":{"时间占比":"30%","目标":"建立价值认知","关键动作":"板块价值→周边配套→项目亮点→开发商介绍"}}`
}
func (e *CommunicationRhythm) Copy() AdviceData {
return new(CommunicationRhythm)
}
func (e *CommunicationRhythm) Role() AdviceRole {
return RoleSkill
}
func (e *CommunicationRhythm) Desc() string {
return "沟通节奏控制"
}

View File

@ -0,0 +1,155 @@
package mongo_model
import (
"time"
)
type AdvicerVersionMongo struct {
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
VersionDesc string `json:"versionDesc" bson:"versionDesc"`
DialectFeatures DialectFeatures `json:"dialectFeatures" bson:"DialectFeatures"`
SentencePatterns SentencePatterns `json:"sentencePatterns" bson:"sentencePatterns"`
ToneTags ToneTags `json:"toneTags" bson:"toneTags"`
PersonalityTags PersonalityTags `json:"personalityTags" bson:"personalityTags"`
SignatureDialogues SignatureDialogues `json:"signatureDialogues" bson:"signatureDialogues"`
LastUpdateTime time.Time `json:"lastUpdateTime" bson:"lastUpdateTime"`
}
func NewAdvicerVersionMongo() *AdvicerVersionMongo {
return &AdvicerVersionMongo{}
}
func (a *AdvicerVersionMongo) MongoTableName() string {
return "advicer_version"
}
type AdvicerVersionMongoEntity struct {
DialectFeatures DialectFeatures `json:"dialectFeatures"`
SentencePatterns SentencePatterns `json:"sentencePatterns"`
ToneTags ToneTags `json:"toneTags"`
PersonalityTags PersonalityTags `json:"personalityTags"`
SignatureDialogues SignatureDialogues `json:"signatureDialogues"`
}
func (a *AdvicerVersionMongo) Entity() *AdvicerVersionMongoEntity {
return &AdvicerVersionMongoEntity{
DialectFeatures: a.DialectFeatures,
SentencePatterns: a.SentencePatterns,
ToneTags: a.ToneTags,
PersonalityTags: a.PersonalityTags,
SignatureDialogues: a.SignatureDialogues,
}
}
// SignatureDialogues 代表性对话示例
type SignatureDialogues []struct {
Context string `json:"context"`
Dialogue string `json:"dialogue"` //解释
}
// DialectFeatures 方言特征
type DialectFeatures struct {
Region string `json:"region"` //方言使用程度
Intensity float64 `json:"intensity"` // 方言使用强度0-1
KeyWords []string `json:"KeyWords"`
}
func (e *DialectFeatures) Example() string {
return `{"region":"四川成都话","intensity":0.4,"key_words":["噻","要得","没得","不晓得","是不是"]}`
}
func (e *DialectFeatures) Copy() AdviceData {
return new(DialectFeatures)
}
func (e *DialectFeatures) Role() AdviceRole {
return RoleAdvicer
}
func (e *DialectFeatures) Desc() string {
return "方言特征"
}
// SentencePatterns 句子模式
type SentencePatterns struct {
OpeningMode []string `json:"openingMode"` //开场模式
ExplanationMode []string `json:"explanationMode"` //解释模式
ConfirmationMode []string `json:"confirmationMode"` //确认模式
SummaryMode []string `json:"summaryMode"` //总结模式
TransitionMode []string `json:"transitionMode"` //过渡模式
}
func (e *SentencePatterns) Example() string {
return `{"openingMode":["我给你介绍一下","我们先来看一下"],"explanationMode":["是这样的","我跟你讲","你发现没得"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?"],"summaryMode":["所以说","简单说就是"],"transitionMode":["然后的话","再其次","还有一点"]}`
}
func (e *SentencePatterns) Copy() AdviceData {
return new(SentencePatterns)
}
func (e *SentencePatterns) Role() AdviceRole {
return RoleAdvicer
}
func (e *SentencePatterns) Desc() string {
return "句子模式"
}
// PersonalityTags 个性标签
type PersonalityTags []string
func (e *PersonalityTags) Example() string {
return `["耐心细致","细节控"]`
}
func (e *PersonalityTags) Copy() AdviceData {
return new(PersonalityTags)
}
func (e *PersonalityTags) Role() AdviceRole {
return RoleAdvicer
}
func (e *PersonalityTags) Desc() string {
return "个性标签"
}
// ToneTags 语气标签
type ToneTags struct {
Enthusiasm float64 `json:"enthusiasm"`
Patience float64 `json:"patience"`
Confidence float64 `json:"confidence"`
Friendliness float64 `json:"friendliness"`
Persuasion float64 `json:"persuasion"`
}
func (e *ToneTags) Example() string {
return `{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.75,"persuasion":0.7}`
}
func (e *ToneTags) Copy() AdviceData {
return new(ToneTags)
}
func (e *ToneTags) Role() AdviceRole {
return RoleAdvicer
}
func (e *ToneTags) Desc() string {
return "语气标签"
}
func (e *SignatureDialogues) Example() string {
return `[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都是2.5环内城买房这种是个普遍存在的一个现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有管家送外卖、免费宠物喂养这些增值服务。你算一下就算贵一块钱十年也就多14000但好物业让房子增值不止这点"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比28板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"}]`
}
func (e *SignatureDialogues) Copy() AdviceData {
return new(SignatureDialogues)
}
func (e *SignatureDialogues) Role() AdviceRole {
return RoleAdvicer
}
func (e *SignatureDialogues) Desc() string {
return "代表性对话示例"
}

View File

@ -0,0 +1,24 @@
package mongo_model
type AdviceRole string
const (
RoleAdvicer AdviceRole = "advicer"
RoleProject AdviceRole = "project"
RoleSkill AdviceRole = "skill"
RoleClient AdviceRole = "client"
)
var RoleDesc = map[AdviceRole]string{
RoleAdvicer: "顾问",
RoleProject: "项目",
RoleSkill: "沟通技巧",
RoleClient: "客户",
}
type AdviceData interface {
Example() string
Copy() AdviceData
Role() AdviceRole
Desc() string
}

View File

@ -0,0 +1,10 @@
package mongo_model
import "github.com/google/wire"
var ProviderSetMongo = wire.NewSet(
NewAdvicerVersionMongo,
NewAdvicerTalkSkillMongo,
NewAdvicerProjectMongo,
NewAdvicerClientMongo,
)

View File

@ -1,387 +1,14 @@
package entitys
type AdviceData interface {
Example() string
Copy() AdviceData
Role() AdviceRole
Desc() string
}
type AdviceRole string
const (
RoleAdvicer AdviceRole = "advicer"
RoleProject AdviceRole = "project"
RoleSkill AdviceRole = "skill"
RoleClient AdviceRole = "client"
import (
"ai_scheduler/internal/data/model"
"ai_scheduler/internal/data/mongo_model"
)
var RoleDesc = map[AdviceRole]string{
RoleAdvicer: "顾问",
RoleProject: "项目",
RoleSkill: "沟通技巧",
RoleClient: "客户",
}
// -------顾问
// DialectFeatures 方言特征
type DialectFeatures struct {
Region string `json:"region"` //方言使用程度
Intensity float64 `json:"intensity"` // 方言使用强度0-1
KeyWords []string `json:"KeyWords"`
}
func (e *DialectFeatures) Example() string {
return `{"region":"四川成都话","intensity":0.4,"key_words":["噻","要得","没得","不晓得","是不是"]}`
}
func (e *DialectFeatures) Copy() AdviceData {
return new(DialectFeatures)
}
func (e *DialectFeatures) Role() AdviceRole {
return RoleAdvicer
}
func (e *DialectFeatures) Desc() string {
return "方言特征"
}
// SentencePatterns 句子模式
type SentencePatterns struct {
OpeningMode []string `json:"openingMode"` //开场模式
ExplanationMode []string `json:"explanationMode"` //解释模式
ConfirmationMode []string `json:"confirmationMode"` //确认模式
SummaryMode []string `json:"summaryMode"` //总结模式
TransitionMode []string `json:"transitionMode"` //过渡模式
}
func (e *SentencePatterns) Example() string {
return `{"openingMode":["我给你介绍一下","我们先来看一下"],"explanationMode":["是这样的","我跟你讲","你发现没得"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?"],"summaryMode":["所以说","简单说就是"],"transitionMode":["然后的话","再其次","还有一点"]}`
}
func (e *SentencePatterns) Copy() AdviceData {
return new(SentencePatterns)
}
func (e *SentencePatterns) Role() AdviceRole {
return RoleAdvicer
}
func (e *SentencePatterns) Desc() string {
return "句子模式"
}
// PersonalityTags 个性标签
type PersonalityTags []string
func (e *PersonalityTags) Example() string {
return `["耐心细致","细节控"]`
}
func (e *PersonalityTags) Copy() AdviceData {
return new(PersonalityTags)
}
func (e *PersonalityTags) Role() AdviceRole {
return RoleAdvicer
}
func (e *PersonalityTags) Desc() string {
return "个性标签"
}
// ToneTags 语气标签
type ToneTags struct {
Enthusiasm float64 `json:"enthusiasm"`
Patience float64 `json:"patience"`
Confidence float64 `json:"confidence"`
Friendliness float64 `json:"friendliness"`
Persuasion float64 `json:"persuasion"`
}
func (e *ToneTags) Example() string {
return `{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.75,"persuasion":0.7}`
}
func (e *ToneTags) Copy() AdviceData {
return new(ToneTags)
}
func (e *ToneTags) Role() AdviceRole {
return RoleAdvicer
}
func (e *ToneTags) Desc() string {
return "语气标签"
}
// SignatureDialogues 代表性对话示例
type SignatureDialogues []struct {
Context string `json:"context"`
Dialogue string `json:"dialogue"` //解释
}
func (e *SignatureDialogues) Example() string {
return `[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都是2.5环内城买房这种是个普遍存在的一个现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有管家送外卖、免费宠物喂养这些增值服务。你算一下就算贵一块钱十年也就多14000但好物业让房子增值不止这点"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比28板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"}]`
}
func (e *SignatureDialogues) Copy() AdviceData {
return new(SignatureDialogues)
}
func (e *SignatureDialogues) Role() AdviceRole {
return RoleAdvicer
}
func (e *SignatureDialogues) Desc() string {
return "代表性对话示例"
}
// -------项目
// RegionValue 区域价值话术库
type RegionValue map[string][]string
func (e *RegionValue) Example() string {
return `{"区位层级":["成华区2.5环内侧,这个位置真的稀缺","槐树店板块现在是成华区的number one板块","北接三板桥商圈,西靠万象城,东临火车东站","属于淮舜板块,万象城东的核心位置"],"地价论证":["我们地价19500华晨府20400棕榈也是2万+","2.5环内现在地价没有低于19000的","面粉贵了,面包不可能便宜"],"板块热度":["从21年新希望锦麟一品开始这边全是高端盘","龙湖最高端的滨江系列在这里,新希望的锦麟系列也在这里","各大品牌开发商争相恐后都在这边拿地"],"发展规划":["槐树店板块是棋盘成钢之后第二个富人区","整个板块都是300万到900万的总价段","未来全是改善型住宅,没有刚需盘"]}`
}
func (e *RegionValue) Copy() AdviceData {
return new(RegionValue)
}
func (e *RegionValue) Role() AdviceRole {
return RoleProject
}
func (e *RegionValue) Desc() string {
return "区域价值话术"
}
// CompetitionComparison 竞品对比话术
type CompetitionComparison map[string]map[string]string
func (e *CompetitionComparison) Example() string {
return `{"龙湖滨江云河颂":{"优点承认":"龙湖位置确实好,看沙河公园","价格对比":"他们单价32000-35000但得房率只有95%套内算下来36000+","优势突出":"我们得房率118平实得132平套内单价才33000"},"邦泰云锦":{"定位相似":"邦泰也是首个项目,要打造口碑","价格参考":"他们当时12800拿地现在卖34000","品质对比":"我们外立面全玻璃幕墙比他们成本高30%"}}`
}
func (e *CompetitionComparison) Copy() AdviceData {
return new(CompetitionComparison)
}
func (e *CompetitionComparison) Role() AdviceRole {
return RoleProject
}
func (e *CompetitionComparison) Desc() string {
return "竞品对比话术"
}
// CoreSellingPoints 核心卖点
type CoreSellingPoints map[string]string
func (e *CoreSellingPoints) Example() string {
return `{"产品配置高端":"全玻璃幕墙+铝单板外立面三层中空氩气玻璃3.2米层高方太Y9烟机灶具高仪卫浴","地段稀缺性":"成华区2.5环内侧核心地段槐树店板块是成华区number one板块被三板桥、万象城、火车东站包围","得房率高":"118平实得132平套内单价33000比龙湖滨江云河颂套内单价低3000"}`
}
func (e *CoreSellingPoints) Copy() AdviceData {
return new(CoreSellingPoints)
}
func (e *CoreSellingPoints) Role() AdviceRole {
return RoleProject
}
func (e *CoreSellingPoints) Desc() string {
return "核心卖点"
}
// SupportingFacilities 配套体系
type SupportingFacilities map[string]map[string]string
func (e *SupportingFacilities) Example() string {
return `{"交通配套":{"地铁":"双店路站350米7号线槐树店站550米4号线未来12号线","道路":"中环路、成洛大道到春熙路5个站","通达性":"到火车东站2个站到华西30分钟内"},"教育配套":{"幼儿园":"楼下公立幼儿园","小学":"城市附小锦汇东城(成华区生源最好的学校)","生源优势":"周边新盘都是300万+,生源纯粹"},"医疗配套":{"三甲医院":"市六医院、市二医院3公里内","顶尖医疗":"华西医院锦江院区30分钟车程","便利性":"到华西本部也是30分钟内"}}`
}
func (e *SupportingFacilities) Copy() AdviceData {
return new(SupportingFacilities)
}
func (e *SupportingFacilities) Role() AdviceRole {
return RoleProject
}
func (e *SupportingFacilities) Desc() string {
return "配套体系"
}
// DeveloperBacking 开发商背书
type DeveloperBacking map[string]string
func (e *DeveloperBacking) Example() string {
return `{"公司实力":"中信资产,多元化民营企业","资金安全":"在河南渑池有铝土矿每年稳定收入10亿","开发经验":"宜宾有5个项目贵州2个成都是首个项目","合作方":"招商铂金物业,首次与外部企业合作"}`
}
func (e *DeveloperBacking) Copy() AdviceData {
return new(DeveloperBacking)
}
func (e *DeveloperBacking) Role() AdviceRole {
return RoleProject
}
func (e *DeveloperBacking) Desc() string {
return "开发商背书"
}
// -------销售话术
// NeedsMining 需求挖掘话术
type NeedsMining map[string][]string
func (e *NeedsMining) Example() string {
return `{"预算需求":["你们总价想控制在多少以内?","是考虑按揭还是一次性?","月供能接受多少范围?"],"居住需求":["几个人住?有老人小孩吗?","主要是自住还是考虑投资?","现在住哪里?想改善哪些方面?"],"通勤需求":["在哪个位置上班?","主要开车还是坐地铁?","对地铁距离有要求吗?"]}`
}
func (e *NeedsMining) Copy() AdviceData {
return new(NeedsMining)
}
func (e *NeedsMining) Role() AdviceRole {
return RoleSkill
}
func (e *NeedsMining) Desc() string {
return "需求挖掘话术"
}
// PainPointResponse 痛点应对策略
type PainPointResponse map[string]map[string]string
func (e *PainPointResponse) Example() string {
return `{"地块太小":{"承认事实":"14亩确实不大","普遍现象":"2.5环内都是小地块万景13亩中铁建8.8亩","转化优势":"但人少安静,楼间距反而更开阔","对比竞品":"339的邦泰才11亩人家上千万豪宅"},"物业费高":{"理解感受":"我懂你,我们也觉得有点贵","价值分析":"但6块里3块是增值服务保洁、送外卖","价格补贴":"前三年补贴到5块跟其他盘差不多"}}`
}
func (e *PainPointResponse) Copy() AdviceData {
return new(PainPointResponse)
}
func (e *PainPointResponse) Role() AdviceRole {
return RoleSkill
}
func (e *PainPointResponse) Desc() string {
return "痛点应对策略"
}
// ValueBuilding 价值塑造技巧
type ValueBuilding map[string][]string
func (e *ValueBuilding) Example() string {
return `{"地段价值塑造":["买房最重要的是地段、地段、还是地段","核心地段的核心资产才保值增值","2.5环内的地卖一块少一块,不可再生"],"产品价值塑造":["我们是用改善的价格,买豪宅的标准","很多细节都是3000万豪宅才有的配置","外立面成本比竞品高30%,但单价差不多"]}`
}
func (e *ValueBuilding) Copy() AdviceData {
return new(ValueBuilding)
}
func (e *ValueBuilding) Role() AdviceRole {
return RoleSkill
}
func (e *ValueBuilding) Desc() string {
return "价值塑造技巧"
}
// ClosingTechniques 促单话术
type ClosingTechniques map[string]map[string][]string
func (e *ClosingTechniques) Example() string {
return `{"紧迫感营造":{"时间紧迫":["今天是月底最后一天,领导有压力价格可谈","我们刚刚开盘,还有额外优惠","月底冲业绩,价格最有弹性"],"房源稀缺":["118只剩20多套了好楼层不多","这栋楼就60户卖一套少一套","特价房只有这几套,今天不定可能就没了"]},"优惠策略":{"价格优惠":["今天定的话,我可以跟领导申请额外折扣","买车位的话,总价多给两个点优惠","一次性付款再优惠一个点"],"附加价值":["送一年物业费","送品牌家电礼包","优先选车位"]},"决策推动":{"小步推进":["要不先交个小定保留房源?","可以先排个号,有优惠优先通知你","今天不定的话,我帮你留意好楼层"]}}`
}
func (e *ClosingTechniques) Copy() AdviceData {
return new(ClosingTechniques)
}
func (e *ClosingTechniques) Role() AdviceRole {
return RoleSkill
}
func (e *ClosingTechniques) Desc() string {
return "促单话术"
}
// CommunicationRhythm 沟通节奏控制
type CommunicationRhythm map[string]map[string]string
func (e *CommunicationRhythm) Example() string {
return `{"开场阶段":{"时间占比":"5%","目标":"建立关系,了解需求","关键动作":"亲切称呼,简单寒暄,确认看房重点"},"沙盘讲解":{"时间占比":"30%","目标":"建立价值认知","关键动作":"板块价值→周边配套→项目亮点→开发商介绍"}}`
}
func (e *CommunicationRhythm) Copy() AdviceData {
return new(CommunicationRhythm)
}
func (e *CommunicationRhythm) Role() AdviceRole {
return RoleSkill
}
func (e *CommunicationRhythm) Desc() string {
return "沟通节奏控制"
}
//----------客户
// Customer 客户信息
type Customer []ClientInfo
type ClientInfo struct {
// 个人信息
PersonalInfo PersonalInfo `json:"personalInfo"`
// 购房目的
PurchasePurpose PurchasePurpose `json:"purchasePurpose"`
// 核心需求
CoreDemands CoreDemands `json:"coreDemands"`
// 关注点与顾虑
Concerns []string `json:"concerns"`
// 决策建议
DecisionProfile []string `json:"decisionProfile"`
}
type PersonalInfo struct {
Name string `json:"name"` // 姓氏
Gender string `json:"gender"` // 性别
Location string `json:"location"` // 来源地/当前居住地
IsFirstHome bool `json:"isFirstHome"` // 是否首套房
FamilyOrganize string `json:"familyOrganize"` // 家庭人数
}
type PurchasePurpose struct {
PrimaryPurpose string `json:"primaryPurpose"` // 主要目的
SecondaryPurpose string `json:"secondaryPurpose"` // 次要目的
DecisionMakers string `json:"decisionMakers"` // 决策人
}
type CoreDemands struct {
TotalBudget string `json:"totalBudget"` // 预算范围
PreferredLayout string `json:"preferredLayout"` // 偏好户型
CoreAppeal string `json:"coreAppeal"` // 核心述求
}
func (e *Customer) Example() string {
return `[{"personalInfo":{"name":"唐","gender":"男","location":"成都北门","isFirstHome":true,"familyOrganize":"夫妻+1孩+父母同住"},"purchasePurpose":{"primaryPurpose":"首次置业,解决自住","secondaryPurpose":"资产保值,未来可出租","decisionMakers":"夫妻双方"},"coreDemands":{"totalBudget":"350-400"","preferredLayout":"118㎡四房三卫双套房","coreAppeal":"在有限预算内满足家庭居住功能,确保房产保值"},"concerns":["总价超预算风险","板块保值能力","未来租金回报率","开发商资金实力"],"decisionProfile":["预算导向,严格控制总价","重点关注户型功能性和实用性","需要对比板块发展潜力","对开发商交付能力有顾虑"]},{"personalInfo":{"name":"冯女士","gender":"女","location":"","isFirstHome":false,"familyOrganize":"夫妻+1孩+父母同住"},"purchasePurpose":{"primaryPurpose":"改善居住条件","secondaryPurpose":"子女教育质量提升","decisionMakers":"夫妻双方需家庭共同商议]},"coreDemands":{"totalBudget":"400-500","preferredLayout":"118㎡四房三卫非全景户型","coreAppeal":"安静舒适、学区有保障的改善型住房"},"concerns":["临路噪音影响老人休息","学区质量和稳定性","社区小,绿化空间有限","得房率是否足够","二八板块学区对比"],"decisionProfile":["对噪音敏感,需要安静环境","重视教育资源配置","关注社区品质和舒适度","需要详细对比不同板块学区优势"]}]`
}
func (e *Customer) Copy() AdviceData {
return new(Customer)
}
func (e *Customer) Role() AdviceRole {
return RoleClient
}
func (e *Customer) Desc() string {
return "客户信息"
type ChatData struct {
ClientInfo *mongo_model.AdvicerClientMongoEntity `json:"clientInfo"`
TalkSkill *mongo_model.AdvicerTalkSkillMongoEntity `json:"talkSkill"`
ProjectInfo *mongo_model.AdvicerProjectMongoEntity `json:"projectInfo"`
AdvicerInfo *model.AiAdviceAdvicerEntity `json:"advicerInfo"`
AdvicerVersion *mongo_model.AdvicerVersionMongoEntity `json:"advicerVersion"`
}

View File

@ -1,29 +1,173 @@
package entitys
import "ai_scheduler/internal/data/mongo_model"
type AdvicerInitReq struct {
AdvicerID int32 `json:"advicer_id"`
ProjectID int32 `json:"project_id"`
Name string `json:"name"` // 姓名
Birth string `json:"birth"` // 用户名称
Gender int32 `json:"gender"` // 1:男2
WorkingYears int32 `json:"working_years"` // 工作年限
AdvicerID int32 `json:"AdvicerId"`
ProjectID int32 `json:"ProjectId"`
Name string `json:"name"` // 姓名
Birth string `json:"birth"` // 用户名称
Gender int32 `json:"gender"` // 1:男2
WorkingYears int32 `json:"WorkingYears"` // 工作年限
}
type AdvicerInfoReq struct {
AdvicerID int32 `json:"AdvicerId"`
}
type AdvicerListReq struct {
ProjectId int32 `json:"project_id"`
ProjectId int32 `json:"ProjectId"`
}
type AdvicerVersionInitReq struct {
VersionID int32 `json:"version_id"`
AdvicerID int32 `json:"advicer_id"`
VersionDesc string `json:"version_desc"` // 版本名称
DialectFeatures string `json:"dialect_features"` // 语言风格
SentencePatterns string `json:"sentence_patterns"` // 句子模式
ToneTags string `json:"tone_tags"` // 语气标签
PersonalityTags string `json:"personality_tags"` // 个性标签
SignatureDialogues string `json:"signature_dialogues"` // 代表性对话示例
type AdvicerVersionAddReq struct {
AdvicerID int32 `json:"advicerId"`
VersionDesc string `json:"versionDesc"`
DialectFeatures mongo_model.DialectFeatures `json:"dialectFeatures"`
PersonalityTags mongo_model.PersonalityTags `json:"personalityTags"`
SentencePatterns mongo_model.SentencePatterns `json:"sentencePatterns"`
SignatureDialogues mongo_model.SignatureDialogues `json:"signatureDialogues"`
ToneTags mongo_model.ToneTags `json:"toneTags"`
}
type AdvicerVersionUpdateReq struct {
Id string `json:"id"`
AdvicerID int32 `json:"advicerId"`
VersionDesc string `json:"versionDesc"`
DialectFeatures mongo_model.DialectFeatures `json:"dialectFeatures"`
PersonalityTags mongo_model.PersonalityTags `json:"personalityTags"`
SentencePatterns mongo_model.SentencePatterns `json:"sentencePatterns"`
SignatureDialogues mongo_model.SignatureDialogues `json:"signatureDialogues"`
ToneTags mongo_model.ToneTags `json:"toneTags"`
}
type AdvicerVersionListReq struct {
AdvicerID int32 `json:"advicer_id"`
Id string `json:"id"`
AdvicerId int32 `json:"advicerId"`
VersionDesc string `json:"versionDesc"`
}
type AdvicerVersionDelReq struct {
Id string `json:"id"`
}
type AdvicerVersionInfoReq struct {
Id string `json:"id"`
}
type AdvicerTalkSkillAddReq struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
Desc string `json:"desc" bson:"desc"`
NeedsMining mongo_model.NeedsMining `json:"needsMining" bson:"needsMining"`
PainPointResponse mongo_model.PainPointResponse `json:"painPointResponse" bson:"painPointResponse"`
ValueBuilding mongo_model.ValueBuilding `json:"valueBuilding" bson:"valueBuilding"`
ClosingTechniques mongo_model.ClosingTechniques `json:"closingTechniques" bson:"closingTechniques"`
CommunicationRhythm mongo_model.CommunicationRhythm `json:"communicationRhythm" bson:"communicationRhythm"`
}
type AdvicerTalkSkillUpdateReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId" :"advicer-id"`
Desc string `json:"desc" bson:"desc" :"desc"`
NeedsMining mongo_model.NeedsMining `json:"needsMining" bson:"needsMining" :"needs-mining"`
PainPointResponse mongo_model.PainPointResponse `json:"painPointResponse" bson:"painPointResponse" :"pain-point-response"`
ValueBuilding mongo_model.ValueBuilding `json:"valueBuilding" bson:"valueBuilding" :"value-building"`
ClosingTechniques mongo_model.ClosingTechniques `json:"closingTechniques" bson:"closingTechniques" :"closing-techniques"`
CommunicationRhythm mongo_model.CommunicationRhythm `json:"communicationRhythm" bson:"communicationRhythm" :"communication-rhythm"`
}
type AdvicerTalkSkillListReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
Desc string `json:"desc" bson:"desc"`
}
type AdvicerTalkSkillDelReq struct {
Id string `json:"id"`
}
type AdvicerTalkSkillInfoReq struct {
Id string `json:"id"`
}
type AdvicerProjectAddReq struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
ProjectInfo mongo_model.ProjectInfo `json:"projectInfo" bson:"projectInfo"`
RegionValue mongo_model.RegionValue `json:"regionValue" bson:"regionValue"`
CompetitionComparison mongo_model.CompetitionComparison `json:"competitionComparison" bson:"competitionComparison"`
CoreSellingPoints mongo_model.CoreSellingPoints `json:"coreSellingPoints" bson:"coreSellingPoints"`
SupportingFacilities mongo_model.SupportingFacilities `json:"supportingFacilities" bson:"supportingFacilities"`
DeveloperBacking mongo_model.DeveloperBacking `json:"developerBacking" bson:"developerBacking"`
}
type AdvicerrProjectUpdateReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
ProjectInfo mongo_model.ProjectInfo `json:"projectInfo" bson:"projectInfo"`
RegionValue mongo_model.RegionValue `json:"regionValue" bson:"regionValue"`
CompetitionComparison mongo_model.CompetitionComparison `json:"competitionComparison" bson:"competitionComparison"`
CoreSellingPoints mongo_model.CoreSellingPoints `json:"coreSellingPoints" bson:"coreSellingPoints"`
SupportingFacilities mongo_model.SupportingFacilities `json:"supportingFacilities" bson:"supportingFacilities"`
DeveloperBacking mongo_model.DeveloperBacking `json:"developerBacking" bson:"developerBacking"`
}
type AdvicerProjectInfoReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
}
type AdvicerClientAddReq struct {
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
PersonalInfo mongo_model.PersonalInfo `json:"personalInfo" bson:"personalInfo"`
PurchasePurpose mongo_model.PurchasePurpose `json:"purchasePurpose" bson:"purchasePurpose"`
CoreDemands mongo_model.CoreDemands `json:"coreDemands" bson:"coreDemands"`
Concerns []string `json:"concerns" bson:"concerns"`
DecisionProfile []string `json:"decisionProfile" bson:"decisionProfile"`
}
type AdvicerrClientUpdateReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
PersonalInfo mongo_model.PersonalInfo `json:"personalInfo" bson:"personalInfo"`
PurchasePurpose mongo_model.PurchasePurpose `json:"purchasePurpose" bson:"purchasePurpose"`
CoreDemands mongo_model.CoreDemands `json:"coreDemands" bson:"coreDemands"`
Concerns []string `json:"concerns" bson:"concerns"`
DecisionProfile []string `json:"decisionProfile" bson:"decisionProfile"`
}
type AdvicerClientListReq struct {
Id string `json:"id"`
ProjectId int32 `json:"projectId" bson:"projectId"`
AdvicerId int32 `json:"advicerId" bson:"advicerId"`
}
type AdvicerClientDelReq struct {
Id string `json:"id"`
}
type AdvicerClientInfoReq struct {
Id string `json:"id"`
}
type AdvicerChatRegistReq struct {
AdvicerVersionId string `json:"advicerVersionId"`
ClientId string `json:"clientId"`
TalkSkillId string `json:"talkSkillId"`
}
type AdvicerChatRegistRes struct {
SessionId string `json:"sessionId"`
}
type AdvicerChatReq struct {
SessionId string `json:"sessionId"`
Content string `json:"content"`
}
type AdvicerChatRes struct {
Content int32 `json:"content"`
}

View File

@ -12,6 +12,7 @@ import (
type Mongo struct {
Client *mongo.Client
c *config.Config
}
func NewMongoDb(ctx context.Context, c *config.Config) (*Mongo, func()) {
@ -23,13 +24,25 @@ func NewMongoDb(ctx context.Context, c *config.Config) (*Mongo, func()) {
ConnectTimeout: time.Duration(c.Mongo.ConnectTimeout) * time.Second,
SocketTimeout: time.Duration(c.Mongo.SocketTimeout) * time.Second,
})
if err != nil {
panic(fmt.Sprintf("mongo数据库错误: %v", err))
}
if err = transDBClient.Ping(ctx, nil); err != nil {
panic(fmt.Sprintf("mongo链接失败: %v", err))
}
return &Mongo{
Client: transDBClient,
c: c,
}, func() {
transDBClient.Disconnect(ctx)
}
}
type MongoModel interface {
MongoTableName() string
}
func (m *Mongo) Co(mongoModel MongoModel) *mongo.Collection {
return m.Client.Database(m.c.Mongo.DataBase).Collection(mongoModel.MongoTableName())
}

View File

@ -4,22 +4,13 @@ import (
"ai_scheduler/internal/gateway"
"ai_scheduler/internal/server/router"
"ai_scheduler/internal/services"
"ai_scheduler/internal/services/advice"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/logger"
"github.com/gofiber/fiber/v2/middleware/recover"
)
type HTTPServer struct {
app *fiber.App
service *services.ChatService
session *services.SessionService
gateway *gateway.Gateway
callback *services.CallbackService
chatHis *services.HistoryService
capabilityService *services.CapabilityService
}
func NewHTTPServer(
service *services.ChatService,
session *services.SessionService,
@ -28,10 +19,16 @@ func NewHTTPServer(
callback *services.CallbackService,
chatHis *services.HistoryService,
capabilityService *services.CapabilityService,
adviceFile *advice.FileService,
adviceData *advice.AdvicerService,
adviceChat *advice.ChatService,
adviceProject *advice.ProjectService,
adviceTalkSkill *advice.TalkSkillService,
adviceClient *advice.ClientService,
) *fiber.App {
//构建 server
app := initRoute()
router.SetupRoutes(app, service, session, task, gateway, callback, chatHis, capabilityService)
router.SetupRoutes(app, service, session, task, gateway, callback, chatHis, capabilityService, adviceFile, adviceData, adviceChat, adviceProject, adviceTalkSkill, adviceClient)
return app
}

View File

@ -1,6 +1,7 @@
package router
import (
errorcode "ai_scheduler/internal/data/error"
errors "ai_scheduler/internal/data/error"
"ai_scheduler/internal/gateway"
"ai_scheduler/internal/services"
@ -15,19 +16,11 @@ import (
"github.com/gofiber/websocket/v2"
)
type RouterServer struct {
app *fiber.App
service *services.ChatService
session *services.SessionService
gateway *gateway.Gateway
chatHist *services.HistoryService
capabilityService *services.CapabilityService
}
// SetupRoutes 设置路由
func SetupRoutes(app *fiber.App, ChatService *services.ChatService, sessionService *services.SessionService, task *services.TaskService,
gateway *gateway.Gateway, callbackService *services.CallbackService, chatHist *services.HistoryService,
capabilityService *services.CapabilityService, adviceFile *advice.FileService, adviceData *advice.DataService,
capabilityService *services.CapabilityService, adviceFile *advice.FileService, adviceData *advice.AdvicerService,
adviceChat *advice.ChatService, adviceProject *advice.ProjectService, adviceTalkSkill *advice.TalkSkillService, adviceClient *advice.ClientService,
) {
app.Use(func(c *fiber.Ctx) error {
// 设置 CORS 头
@ -103,28 +96,34 @@ func SetupRoutes(app *fiber.App, ChatService *services.ChatService, sessionServi
advicer := r.Group("advice/")
advicer.Post("file/word/ana", adviceFile.WordAna)
//顾问
advicer.Post("file/advicer/add", adviceData.AdvicerAdd)
advicer.Post("file/advicer/update", adviceData.AdvicerUpdate)
advicer.Post("file/advicer/list", adviceData.AdvicerList)
advicer.Post("file/advicer/version/add", adviceData.AdvicerVersionAdd)
advicer.Post("file/advicer/version/update", adviceFile.WordAna)
advicer.Post("file/advicer/version/del", adviceFile.WordAna)
advicer.Post("file/advicer/version/list", adviceFile.WordAna)
advicer.Post("advicer/add", adviceData.AdvicerUpdate)
advicer.Post("advicer/update", adviceData.AdvicerUpdate)
advicer.Post("advicer/list", adviceData.AdvicerList)
advicer.Post("advicer/version/add", adviceData.AdvicerVersionAdd)
advicer.Post("advicer/version/update", adviceData.AdvicerVersionUpdate)
advicer.Post("advicer/version/del", adviceData.AdvicerVersionDel)
advicer.Post("advicer/version/list", adviceData.AdvicerVersionList)
//聊天技巧
advicer.Post("file/skill/list", adviceFile.WordAna)
advicer.Post("file/skill/init", adviceFile.WordAna)
advicer.Post("file/skill/add", adviceFile.WordAna)
advicer.Post("file/skill/update", adviceFile.WordAna)
advicer.Post("file/skill/del", adviceFile.WordAna)
advicer.Post("file/skill/list", adviceFile.WordAna)
advicer.Post("skill/list", adviceTalkSkill.TalkSkillList)
advicer.Post("skill/add", adviceTalkSkill.TalkSkillAdd)
advicer.Post("skill/update", adviceTalkSkill.TalkSkillUpdate)
advicer.Post("skill/del", adviceTalkSkill.TalkSkillUpdate)
//项目
advicer.Post("file/project/init", adviceFile.WordAna)
advicer.Post("file/project/add", adviceFile.WordAna)
advicer.Post("file/project/update", adviceFile.WordAna)
advicer.Post("project/add", adviceProject.Add)
advicer.Post("project/update", adviceProject.Update)
advicer.Post("project/info", adviceProject.Info)
//客户
advicer.Post("file/client/init", adviceFile.WordAna)
advicer.Post("file/client/add", adviceFile.WordAna)
advicer.Post("file/client/update", adviceFile.WordAna)
advicer.Post("client/add", adviceClient.Add)
advicer.Post("client/update", adviceClient.Update)
advicer.Post("client/list", adviceClient.List)
advicer.Post("client/del", adviceClient.Del)
//客户
advicer.Post("chat/regis", adviceChat.Regis)
advicer.Post("chat/chat", adviceChat.Chat)
}
func routerSocket(app *fiber.App, chatService *services.ChatService) {
@ -161,12 +160,13 @@ func registerCommon(c *fiber.Ctx, err error) error {
if c.Path() == "/api/v1/qywx/callback" {
return nil
}
bsErr, ok := err.(*errors.BusinessErr)
if !ok {
bsErr = errors.SystemError
}
// 如果有错误发生
if err != nil {
bsErr, ok := err.(*errors.BusinessErr)
if !ok {
bsErr = errorcode.SysErr(err.Error())
}
// 返回自定义错误响应
return c.JSON(fiber.Map{
"message": bsErr.Error(),
@ -179,10 +179,17 @@ func registerCommon(c *fiber.Ctx, err error) error {
// 是 SSE 请求
return c.SendString("这是 SSE 请求")
}
var data interface{}
json.Unmarshal(c.Response().Body(), &data)
body := c.Response().Body()
var rawData json.RawMessage
if len(body) > 0 {
if err := json.Unmarshal(body, &rawData); err != nil {
// 解析失败作为字符串包装成JSON
rawData = json.RawMessage(`"` + strings.ReplaceAll(string(body), `"`, `\"`) + `"`)
}
}
return c.JSON(fiber.Map{
"data": data,
"data": rawData,
"message": errors.Success.Error(),
"code": errors.Success.Code(),
})

View File

@ -0,0 +1,78 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"github.com/gofiber/fiber/v2"
)
// AdvicerService 数据处理
type AdvicerService struct {
adviceBiz *biz.AdviceAdvicerBiz
cfg *config.Config
}
// NewDataService
func NewAdvicerService(
adviceBiz *biz.AdviceAdvicerBiz,
cfg *config.Config,
) *AdvicerService {
return &AdvicerService{
adviceBiz: adviceBiz,
cfg: cfg,
}
}
func (d *AdvicerService) AdvicerUpdate(c *fiber.Ctx) error {
req := &entitys.AdvicerInitReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.Update(c.UserContext(), req)
}
func (d *AdvicerService) AdvicerList(c *fiber.Ctx) error {
req := &entitys.AdvicerListReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.adviceBiz.List(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}
func (d *AdvicerService) AdvicerVersionAdd(c *fiber.Ctx) error {
req := &entitys.AdvicerVersionAddReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.VersionAdd(c.UserContext(), req)
}
func (d *AdvicerService) AdvicerVersionUpdate(c *fiber.Ctx) error {
req := &entitys.AdvicerVersionUpdateReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.VersionUpdate(c.UserContext(), req)
}
func (d *AdvicerService) AdvicerVersionList(c *fiber.Ctx) error {
req := &entitys.AdvicerVersionListReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.adviceBiz.VersionList(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}
func (d *AdvicerService) AdvicerVersionDel(c *fiber.Ctx) error {
req := &entitys.AdvicerVersionDelReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.VersionDel(c.UserContext(), req)
}

View File

@ -5,8 +5,10 @@ import (
"ai_scheduler/internal/biz/llm_service/third_party"
"ai_scheduler/internal/config"
"ai_scheduler/internal/data/impl"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/pkg"
"ai_scheduler/utils"
"context"
"encoding/json"
@ -18,19 +20,47 @@ import (
)
func Test_WordAna(t *testing.T) {
Run(nil)
ana, err := file.WordAnat("https://attachment-public.oss-cn-hangzhou.aliyuncs.com/ai-scheduler/data-analytics/word/content3.docx")
Run(context.Background(), nil)
ana, err := file.WordAnat("https://attachment-public.oss-cn-hangzhou.aliyuncs.com/ai-scheduler/data-analytics/word/content2.docx")
t.Log(ana, err)
}
func Test_AdvicerInit(t *testing.T) {
reqBody := `{"advicer_id": 124, "name": "张三111", "birth": "1990-01-01", "gender": 1, "working_years": 10}`
Run([]byte(reqBody))
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerUpdate(fiberCtx)
t.Log(err)
}
func Test_AdvicerVersionAdd(t *testing.T) {
reqBody := `{"advicerId":124,"versionDesc":"第三个版本","dialectFeatures":{"region":"四川成都话","intensity":0.6,"KeyWords":null},"personalityTags":["耐心细致","专业务实","经验丰富","善于引导"],"sentencePatterns":{"openingMode":["我给你介绍一下","我们先来看一下","这边请"],"explanationMode":["是这样的","我跟你讲","你发现没得","说白了"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?","对不对?"],"summaryMode":["所以说","简单说就是","其实"],"transitionMode":["然后的话","再其次","还有一点","另外"]},"signatureDialogues":[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都2.5环内城买房这种是普遍现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米而且小小区人少安静圈层更纯粹"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有夜间送外卖、免费宠物喂养、年度保洁这些增值服务。而且前三年开发商补贴一块钱只需要交5块跟其他盘差不多好物业能让房子后期保值增值更多"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比二八板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"},{"context":"客户担心小区小不保值","dialogue":"哥你不用担心小地块不保值东大街的九龙仓擎天半岛只有两栋楼现在二手房还能卖3万左右是当年的豪宅项目。还有望江名门、仁和春天29号院都是小地块但照样是高端保值盘。核心还是地段我们在槐树店这个成华区最贵的板块保值根本没问题"}],"toneTags":{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.8,"persuasion":0.75}}`
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerVersionAdd(fiberCtx)
t.Log(err)
}
func Test_AdvicerVersionUpdate(t *testing.T) {
reqBody := `{"id":"69804b5a6532131383aeda3a","advicerId":124,"versionDesc":"第三个版本","dialectFeatures":{"region":"四川成都话","intensity":0.6,"KeyWords":null},"personalityTags":["耐心细致","专业务实","经验丰富","善于引导"],"sentencePatterns":{"openingMode":["我给你介绍一下","我们先来看一下","这边请"],"explanationMode":["是这样的","我跟你讲","你发现没得","说白了"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?","对不对?"],"summaryMode":["所以说","简单说就是","其实"],"transitionMode":["然后的话","再其次","还有一点","另外"]},"signatureDialogues":[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都2.5环内城买房这种是普遍现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米而且小小区人少安静圈层更纯粹"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有夜间送外卖、免费宠物喂养、年度保洁这些增值服务。而且前三年开发商补贴一块钱只需要交5块跟其他盘差不多好物业能让房子后期保值增值更多"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比二八板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"},{"context":"客户担心小区小不保值","dialogue":"哥你不用担心小地块不保值东大街的九龙仓擎天半岛只有两栋楼现在二手房还能卖3万左右是当年的豪宅项目。还有望江名门、仁和春天29号院都是小地块但照样是高端保值盘。核心还是地段我们在槐树店这个成华区最贵的板块保值根本没问题"}],"toneTags":{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.8,"persuasion":0.75}}`
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerVersionUpdate(fiberCtx)
t.Log(err)
}
func Test_VersionList(t *testing.T) {
reqBody := `{"id":"69804060c17976e5e21858a8"}`
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerVersionList(fiberCtx)
t.Log(err)
}
func Test_AdvicerVersionDel(t *testing.T) {
reqBody := `{"id":"698056073059550befc4f0da"}`
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerVersionDel(fiberCtx)
t.Log(err)
}
func Test_Json(t *testing.T) {
responseByte, err := os.ReadFile("./res.json")
if err != nil {
@ -38,7 +68,7 @@ func Test_Json(t *testing.T) {
}
var (
result map[string]interface{}
res = make(map[string]entitys.AdviceData)
res = make(map[string]mongo_model.AdviceData)
)
if err = json.Unmarshal(responseByte, &result); err != nil {
@ -64,13 +94,13 @@ func Test_Json(t *testing.T) {
var (
file *FileService
advicer *DataService
advicer *AdvicerService
configConfig *config.Config
fiberCtx *fiber.Ctx
)
// run 函数是程序的入口函数,负责初始化和配置各个组件
func Run(reqBody []byte) {
func Run(ctx context.Context, reqBody []byte) {
if reqBody != nil {
app := fiber.New()
fctx := &fasthttp.RequestCtx{}
@ -82,29 +112,42 @@ func Run(reqBody []byte) {
configConfig, _ = config.LoadConfigWithEnv()
// 初始化数据库连接
db, _ := utils.NewGormDb(configConfig)
rdb := utils.NewRdb(configConfig)
advicerImpl := impl.NewAdviceAdvicerImpl(db)
advicerVersionImpl := impl.NewAdviceAdvicerVersionImpl(db)
advicerVersionMongo := mongo_model.NewAdvicerVersionMongo()
advicerTalkSkillMongo := mongo_model.NewAdvicerTalkSkillMongo()
advicerClientMongo := mongo_model.NewAdvicerClientMongo()
advicerProjectMongo := mongo_model.NewAdvicerProjectMongo()
hsyq := third_party.NewHsyq()
advicerfilebiz := biz.NewAdviceFileBiz(hsyq)
advicerbiz := biz.NewAdviceAdvicerBiz(advicerImpl, advicerVersionImpl)
mongo, _ := pkg.NewMongoDb(ctx, configConfig)
adviceAdvicerBiz := biz.NewAdviceAdvicerBiz(advicerImpl, advicerVersionMongo, mongo)
skillBiz := biz.NewAdviceSkillBiz(advicerTalkSkillMongo, mongo)
clientBiz := biz.NewAdviceClientBiz(advicerClientMongo, mongo)
projectBiz := biz.NewAdviceProjectBiz(advicerProjectMongo, mongo)
chatBiz := biz.NewAdviceChatBiz(hsyq, rdb)
file = NewFileService(advicerfilebiz, configConfig)
advicer = NewDataService(advicerbiz, configConfig)
advicer = NewAdvicerService(adviceAdvicerBiz, configConfig)
skill = NewTalkSkillService(skillBiz, configConfig)
client = NewClientService(clientBiz, configConfig)
project = NewProjectService(projectBiz, configConfig)
chat = NewChatService(chatBiz, clientBiz, adviceAdvicerBiz, projectBiz, skillBiz, configConfig)
}
var dataMap = map[string]entitys.AdviceData{
"DialectFeatures": &entitys.DialectFeatures{},
"SentencePatterns": &entitys.SentencePatterns{},
"PersonalityTags": &entitys.PersonalityTags{},
"ToneTags": &entitys.ToneTags{},
"SignatureDialogues": &entitys.SignatureDialogues{},
"RegionValue": &entitys.RegionValue{},
"CompetitionComparison": &entitys.CompetitionComparison{},
"CoreSellingPoints": &entitys.CoreSellingPoints{},
"SupportingFacilities": &entitys.SupportingFacilities{},
"DeveloperBacking": &entitys.DeveloperBacking{},
"NeedsMining": &entitys.NeedsMining{},
"PainPointResponse": &entitys.PainPointResponse{},
"ValueBuilding": &entitys.ValueBuilding{},
"ClosingTechniques": &entitys.ClosingTechniques{},
"CommunicationRhythm": &entitys.CommunicationRhythm{},
var dataMap = map[string]mongo_model.AdviceData{
"DialectFeatures": &mongo_model.DialectFeatures{},
"SentencePatterns": &mongo_model.SentencePatterns{},
"PersonalityTags": &mongo_model.PersonalityTags{},
"ToneTags": &mongo_model.ToneTags{},
"SignatureDialogues": &mongo_model.SignatureDialogues{},
"RegionValue": &mongo_model.RegionValue{},
"CompetitionComparison": &mongo_model.CompetitionComparison{},
"CoreSellingPoints": &mongo_model.CoreSellingPoints{},
"SupportingFacilities": &mongo_model.SupportingFacilities{},
"DeveloperBacking": &mongo_model.DeveloperBacking{},
"NeedsMining": &mongo_model.NeedsMining{},
"PainPointResponse": &mongo_model.PainPointResponse{},
"ValueBuilding": &mongo_model.ValueBuilding{},
"ClosingTechniques": &mongo_model.ClosingTechniques{},
"CommunicationRhythm": &mongo_model.CommunicationRhythm{},
}

View File

@ -0,0 +1,117 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
errorcode "ai_scheduler/internal/data/error"
"ai_scheduler/internal/pkg"
"ai_scheduler/internal/data/mongo_model"
"ai_scheduler/internal/entitys"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/log"
)
// FileService 文件处理
type ChatService struct {
adviceChatBiz *biz.AdviceChatBiz
adviceClientBiz *biz.AdviceClientBiz
adviceAdvicerBiz *biz.AdviceAdvicerBiz
adviceProjectBiz *biz.AdviceProjectBiz
adviceSkillBiz *biz.AdviceSkillBiz
cfg *config.Config
}
// NewFileService
func NewChatService(
adviceChatBiz *biz.AdviceChatBiz,
adviceClientBiz *biz.AdviceClientBiz,
adviceAdvicerBiz *biz.AdviceAdvicerBiz,
adviceProjectBiz *biz.AdviceProjectBiz,
adviceSkillBiz *biz.AdviceSkillBiz,
cfg *config.Config,
) *ChatService {
return &ChatService{
adviceChatBiz: adviceChatBiz,
cfg: cfg,
adviceClientBiz: adviceClientBiz,
adviceAdvicerBiz: adviceAdvicerBiz,
adviceProjectBiz: adviceProjectBiz,
adviceSkillBiz: adviceSkillBiz,
}
}
func (a *ChatService) Regis(c *fiber.Ctx) error {
req := &entitys.AdvicerChatRegistReq{}
if err := c.BodyParser(req); err != nil {
return err
}
if len(req.AdvicerVersionId) == 0 {
return errorcode.ParamErr("AdvicerVersionId is empty")
}
if len(req.TalkSkillId) == 0 {
return errorcode.ParamErr("talkSkillId is empty")
}
//顾问版本信息
versionInfo, err := a.adviceAdvicerBiz.VersionInfo(c.UserContext(), &entitys.AdvicerVersionInfoReq{
Id: req.AdvicerVersionId,
})
if err != nil {
return err
}
//顾问信息
advicerInfo, err := a.adviceAdvicerBiz.AdvicerInfo(c.UserContext(), &entitys.AdvicerInfoReq{
AdvicerID: versionInfo.AdvicerId,
})
if err != nil {
return err
}
//项目信息
projectInfo, err := a.adviceProjectBiz.Info(c.UserContext(), &entitys.AdvicerProjectInfoReq{
ProjectId: advicerInfo.ProjectID,
})
if err != nil {
return err
}
//销售技巧
talkSkill, err := a.adviceSkillBiz.Info(c.UserContext(), &entitys.AdvicerTalkSkillInfoReq{
Id: req.TalkSkillId,
})
if err != nil {
return err
}
//客户信息
var clientInfo mongo_model.AdvicerClientMongo
if len(req.ClientId) != 0 {
clientInfo, err = a.adviceClientBiz.Info(c.UserContext(), &entitys.AdvicerClientInfoReq{
Id: req.ClientId,
})
if err != nil {
return err
}
}
chat := entitys.ChatData{
ClientInfo: clientInfo.Entity(),
TalkSkill: talkSkill.Entity(),
ProjectInfo: projectInfo.Entity(),
AdvicerInfo: advicerInfo.Entity(),
AdvicerVersion: versionInfo.Entity(),
}
sessionId, err := a.adviceChatBiz.Regis(c.UserContext(), &chat)
log.Info(sessionId)
return pkg.HandleResponse(c, sessionId, err)
}
func (a *ChatService) Chat(c *fiber.Ctx) error {
req := &entitys.AdvicerChatReq{}
if err := c.BodyParser(req); err != nil {
return err
}
res, err := a.adviceChatBiz.Chat(c.UserContext(), req)
log.Info(res)
return pkg.HandleResponse(c, res, err)
}

File diff suppressed because one or more lines are too long

View File

@ -0,0 +1,61 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"github.com/gofiber/fiber/v2"
)
// ClientService 数据处理
type ClientService struct {
AdviceClientBiz *biz.AdviceClientBiz
cfg *config.Config
}
// NewDataService
func NewClientService(
AdviceClientBiz *biz.AdviceClientBiz,
cfg *config.Config,
) *ClientService {
return &ClientService{
AdviceClientBiz: AdviceClientBiz,
cfg: cfg,
}
}
func (d *ClientService) Add(c *fiber.Ctx) error {
req := &entitys.AdvicerClientAddReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.AdviceClientBiz.Add(c.UserContext(), req)
}
func (d *ClientService) Update(c *fiber.Ctx) error {
req := &entitys.AdvicerrClientUpdateReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.AdviceClientBiz.Update(c.UserContext(), req)
}
func (d *ClientService) List(c *fiber.Ctx) error {
req := &entitys.AdvicerClientListReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.AdviceClientBiz.List(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}
func (d *ClientService) Del(c *fiber.Ctx) error {
req := &entitys.AdvicerClientDelReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.AdviceClientBiz.Del(c.UserContext(), req)
}

View File

@ -0,0 +1,39 @@
package advice
import (
"context"
"testing"
)
func Test_ClientAdd(t *testing.T) {
reqBody := `{"projectId":1,"AdvicerId":1,"personalInfo":{"name":"杜先生","gender":"男","location":"","isFirstHome":false,"familyOrganize":"夫妻+2孩"},"purchasePurpose":{"primaryPurpose":"改善居住条件","secondaryPurpose":"资产保值","decisionMakers":"夫妻双方"},"coreDemands":{"totalBudget":"450-500万","preferredLayout":"140㎡四房三卫","coreAppeal":"户型实用、景观好、社区品质高"},"concerns":["小区小是否保值","价格是否有优惠","开发商交付能力"],"decisionProfile":["注重户型实用性和景观","关注社区品质和后期保值","对价格敏感,希望拿到优惠"]}`
Run(context.Background(), []byte(reqBody))
err := client.Add(fiberCtx)
t.Log(err)
}
func Test_ClientUpdate(t *testing.T) {
reqBody := `{"id":"698199fa0c5f4ae098e009ab","projectId":1,"AdvicerId":1,"personalInfo":{"name":"唐先生1","gender":"男","location":"成都北门","isFirstHome":false,"familyOrganize":"夫妻+1孩+父母同住"},"purchasePurpose":{"primaryPurpose":"改善居住条件","secondaryPurpose":"资产保值,方便子女上学","decisionMakers":"夫妻双方"},"coreDemands":{"totalBudget":"350-400万","preferredLayout":"118㎡四房三卫双套房","coreAppeal":"在预算内满足家庭居住功能,确保房产保值,临近学校"},"concerns":["总价超预算风险","板块保值能力","小区小是否影响居住体验","开发商资金实力"],"decisionProfile":["预算导向,严格控制总价","重点关注户型功能性和实用性","需要对比板块发展潜力","对开发商交付能力有顾虑"]}`
Run(context.Background(), []byte(reqBody))
err := client.Update(fiberCtx)
t.Log(err)
}
func Test_ClientList(t *testing.T) {
reqBody := `{"projectId":1}`
Run(context.Background(), []byte(reqBody))
err := client.List(fiberCtx)
t.Log(err)
}
func Test_ClientDel(t *testing.T) {
reqBody := `{"id":"698056073059550befc4f0da"}`
Run(context.Background(), []byte(reqBody))
err := advicer.AdvicerVersionDel(fiberCtx)
t.Log(err)
}
var (
client *ClientService
)

View File

@ -1,60 +0,0 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"github.com/gofiber/fiber/v2"
)
// DataService 数据处理
type DataService struct {
adviceBiz *biz.AdviceAdvicerBiz
cfg *config.Config
}
// NewDataService
func NewDataService(
adviceBiz *biz.AdviceAdvicerBiz,
cfg *config.Config,
) *DataService {
return &DataService{
adviceBiz: adviceBiz,
cfg: cfg,
}
}
func (d *DataService) AdvicerAdd(c *fiber.Ctx) error {
req := &entitys.AdvicerInitReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.Update(c.UserContext(), req)
}
func (d *DataService) AdvicerUpdate(c *fiber.Ctx) error {
req := &entitys.AdvicerInitReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.Update(c.UserContext(), req)
}
func (d *DataService) AdvicerList(c *fiber.Ctx) error {
req := &entitys.AdvicerListReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.adviceBiz.List(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}
func (d *DataService) AdvicerVersionAdd(c *fiber.Ctx) error {
req := &entitys.AdvicerVersionInitReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceBiz.VersionUpdate(c.UserContext(), req)
}

View File

@ -0,0 +1,52 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"github.com/gofiber/fiber/v2"
)
// ProjectService 数据处理
type ProjectService struct {
adviceProjectBiz *biz.AdviceProjectBiz
cfg *config.Config
}
// NewProjectService
func NewProjectService(
adviceProjectBiz *biz.AdviceProjectBiz,
cfg *config.Config,
) *ProjectService {
return &ProjectService{
adviceProjectBiz: adviceProjectBiz,
cfg: cfg,
}
}
func (d *ProjectService) Add(c *fiber.Ctx) error {
req := &entitys.AdvicerProjectAddReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceProjectBiz.Add(c.UserContext(), req)
}
func (d *ProjectService) Update(c *fiber.Ctx) error {
req := &entitys.AdvicerrProjectUpdateReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceProjectBiz.Update(c.UserContext(), req)
}
func (d *ProjectService) Info(c *fiber.Ctx) error {
req := &entitys.AdvicerProjectInfoReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.adviceProjectBiz.Info(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}

View File

@ -0,0 +1,30 @@
package advice
import (
"context"
"testing"
)
func Test_ProjectAdd(t *testing.T) {
reqBody := `{"projectId":1,"projectInfo":{"projectName":"中信资产项目 *","projectAddress":"成华区槐树店板块2.5环内侧 *","area":"成华区","houseTypes":[{"name":"118平户型","buildArea":"118㎡","innerArea":"132㎡ *","unitPrice":"约33000元/㎡ * (套内单价)","totalPrice":"约389万元 *"},{"name":"核心主力户型 *","buildArea":"120-140㎡ *","innerArea":"高得房率约132-156㎡ *","unitPrice":"约32000-35000元/㎡ *","totalPrice":"约384-490万元 *"},{"name":"大平层户型 *","buildArea":"140㎡+ *","innerArea":"实得约156㎡+ * (得房率110%+)","unitPrice":"约35000元/㎡ *","totalPrice":"约490万元+ *"}]},"competitionComparison":{"华润置地云上":{"价格对比":"他们单价35000左右还靠近安置小区","优势突出":"我们小区纯粹,没有安置小区,居住环境更安静","优点承认":"华润品牌影响力大"},"招商璟宸序":{"价格对比":"他们单价32000左右但地段在28板块地价比我们便宜5000","优势突出":"我们地段在槐树店,是成华区核心板块,未来增值空间更大","优点承认":"招商品牌大,物业也是自己的"},"龙湖滨江云河颂":{"价格对比":"他们单价32000-35000但得房率只有95%套内算下来36000+","优势突出":"我们118平实得132平套内单价才33000还做四房三卫他们143平才双卫","优点承认":"龙湖位置确实好,能看沙河公园"}},"coreSellingPoints":{"产品配置高端":"3.2米层高、全玻璃幕墙+三层中空玻璃、无机磨石车库、方太Y9烟机、高仪卫浴、国千木作柜体","地段稀缺性":"成华区2.5环内侧槐树店板块成华区房价天花板区域被万象城、339、火车东站包围","得房率高":"118平实得132平得房率超110%,四房三卫双套房设计,市面上同面积段没有竞品","物业优质":"招商局铂金物业,有夜间送外卖、免费宠物喂养、全屋保洁等增值服务"},"developerBacking":{"公司实力":"中信资产,多元化民营企业,涉及矿产、有色金属、生态农业、地产开发","合作方":"招商局物业,百年央企,首次与外部企业合作提供铂金服务","开发经验":"2011年开始做地产在宜宾、贵州开发超过500万平米成都是首个项目","资金安全":"在河南渑池有两座优质铝矿每年稳定收入10亿现金流充足"},"regionValue":{"区位层级":["成华区2.5环内侧槐树店板块是成华区number one板块","北接339商圈西靠万象城东临火车东站","属于槐树店崔怀板块,成华区目前最好的开发板块"],"发展规划":["槐树店是成华区未来的富人区,板块还有大量待开发土地","未来这个区域会形成连片高端居住区,城市界面会越来越好"],"地价论证":["我们地价19500比28板块贵5000多","华润华城府地价20400我们和它同属一个板块地价差距小","面粉都这么贵,面包不可能便宜"],"板块热度":["从2021年新希望锦麟一品开始这边全是高端盘","龙湖最高端的滨江云河颂在这里,卖得特别火","各大品牌开发商都在这边拿地,未来全是改善盘"]},"supportingFacilities":{"交通配套":{"地铁":"7号线双店路站350米4号线槐树店站550米未来还有12号线","通达性":"到万象城2个站到华西锦江院区30分钟车程","道路":"中环路、成洛大道到春熙路5个站到火车东站2个站"},"医疗配套":{"三甲医院":"市六医院、市二医院3公里内","顶尖医疗":"华西医院锦江院区、华西本部30分钟车程"},"商业配套":{"便利性":"到万象城2个站到339商圈3个站","核心商圈":"万象城商圈、339商圈","社区商业":"和悦广场、东方希望上东里"},"教育配套":{"小学":"成华小学1-3年级在项目附近4-6年级在二环内","生源优势":"周边新盘都是300万+,生源纯粹"}}}`
Run(context.Background(), []byte(reqBody))
err := project.Add(fiberCtx)
t.Log(err)
}
func Test_ProjectUpdate(t *testing.T) {
reqBody := `{"id":"69804b5a6532131383aeda3a","advicerId":124,"versionDesc":"第三个版本","dialectFeatures":{"region":"四川成都话","intensity":0.6,"KeyWords":null},"personalityTags":["耐心细致","专业务实","经验丰富","善于引导"],"sentencePatterns":{"openingMode":["我给你介绍一下","我们先来看一下","这边请"],"explanationMode":["是这样的","我跟你讲","你发现没得","说白了"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?","对不对?"],"summaryMode":["所以说","简单说就是","其实"],"transitionMode":["然后的话","再其次","还有一点","另外"]},"signatureDialogues":[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都2.5环内城买房这种是普遍现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米而且小小区人少安静圈层更纯粹"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有夜间送外卖、免费宠物喂养、年度保洁这些增值服务。而且前三年开发商补贴一块钱只需要交5块跟其他盘差不多好物业能让房子后期保值增值更多"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比二八板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"},{"context":"客户担心小区小不保值","dialogue":"哥你不用担心小地块不保值东大街的九龙仓擎天半岛只有两栋楼现在二手房还能卖3万左右是当年的豪宅项目。还有望江名门、仁和春天29号院都是小地块但照样是高端保值盘。核心还是地段我们在槐树店这个成华区最贵的板块保值根本没问题"}],"toneTags":{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.8,"persuasion":0.75}}`
Run(context.Background(), []byte(reqBody))
err := project.Update(fiberCtx)
t.Log(err)
}
func Test_ProjectInfo(t *testing.T) {
reqBody := `{"id":"69804b5a6532131383aeda3a","advicerId":124,"versionDesc":"第三个版本","dialectFeatures":{"region":"四川成都话","intensity":0.6,"KeyWords":null},"personalityTags":["耐心细致","专业务实","经验丰富","善于引导"],"sentencePatterns":{"openingMode":["我给你介绍一下","我们先来看一下","这边请"],"explanationMode":["是这样的","我跟你讲","你发现没得","说白了"],"confirmationMode":["对吧?","是不是嘛?","你晓得不?","明白了噻?","对不对?"],"summaryMode":["所以说","简单说就是","其实"],"transitionMode":["然后的话","再其次","还有一点","另外"]},"signatureDialogues":[{"context":"客户质疑地块大小","dialogue":"哥14亩确实不大但你要在成都2.5环内城买房这种是普遍现象。你看万景和绿城都是13亩中铁建只有8.8亩339那个帮泰只有11亩。我们虽然地小但楼间距开阔啊看过去都是200多米而且小小区人少安静圈层更纯粹"},{"context":"客户担心物业费高","dialogue":"姐我懂你意思我们也觉得物业费是有点贵。但招商物业是铂金服务有夜间送外卖、免费宠物喂养、年度保洁这些增值服务。而且前三年开发商补贴一块钱只需要交5块跟其他盘差不多好物业能让房子后期保值增值更多"},{"context":"客户犹豫价格","dialogue":"说实话这个地段的地价都比二八板块贵5000多但我们单价只贵3000。你看龙湖滨江云河颂套内单价都36000了我们才33000真的性价比高现在不买以后这个板块可能就买不起了。"},{"context":"客户担心小区小不保值","dialogue":"哥你不用担心小地块不保值东大街的九龙仓擎天半岛只有两栋楼现在二手房还能卖3万左右是当年的豪宅项目。还有望江名门、仁和春天29号院都是小地块但照样是高端保值盘。核心还是地段我们在槐树店这个成华区最贵的板块保值根本没问题"}],"toneTags":{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.8,"persuasion":0.75}}`
Run(context.Background(), []byte(reqBody))
err := project.Info(fiberCtx)
t.Log(err)
}
var project *ProjectService

View File

@ -0,0 +1,61 @@
package advice
import (
"ai_scheduler/internal/biz"
"ai_scheduler/internal/config"
"ai_scheduler/internal/entitys"
"ai_scheduler/internal/pkg"
"github.com/gofiber/fiber/v2"
)
// TalkSkillService 数据处理
type TalkSkillService struct {
adviceSkillBiz *biz.AdviceSkillBiz
cfg *config.Config
}
// NewTalkSkillService
func NewTalkSkillService(
adviceSkillBiz *biz.AdviceSkillBiz,
cfg *config.Config,
) *TalkSkillService {
return &TalkSkillService{
adviceSkillBiz: adviceSkillBiz,
cfg: cfg,
}
}
func (d *TalkSkillService) TalkSkillAdd(c *fiber.Ctx) error {
req := &entitys.AdvicerTalkSkillAddReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceSkillBiz.VersionAdd(c.UserContext(), req)
}
func (d *TalkSkillService) TalkSkillUpdate(c *fiber.Ctx) error {
req := &entitys.AdvicerTalkSkillUpdateReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceSkillBiz.VersionUpdate(c.UserContext(), req)
}
func (d *TalkSkillService) TalkSkillList(c *fiber.Ctx) error {
req := &entitys.AdvicerTalkSkillListReq{}
if err := c.BodyParser(req); err != nil {
return err
}
list, err := d.adviceSkillBiz.VersionList(c.UserContext(), req)
return pkg.HandleResponse(c, list, err)
}
func (d *TalkSkillService) TalkSkillDel(c *fiber.Ctx) error {
req := &entitys.AdvicerTalkSkillDelReq{}
if err := c.BodyParser(req); err != nil {
return err
}
return d.adviceSkillBiz.VersionDel(c.UserContext(), req)
}

View File

@ -0,0 +1,41 @@
package advice
import (
"context"
"testing"
)
func Test_TalkSkillAdd(t *testing.T) {
reqBody := `{"advicerId":124,"projectId":1,"desc":"第一版本","closingTechniques":{"优惠策略":{"价格优惠":["今天定的话我可以跟领导申请额外1个点的折扣","买车位的话总价再给你优惠2万块","一次性付款的话还能再降1个点"],"附加价值":["送一年物业费","送品牌家电礼包","优先选车位"]},"决策推动":{"小步推进":["要不先交个小定保留房源?","可以先排个号,有优惠我第一时间通知你","今天不定的话,我帮你留意着这个好楼层"]},"紧迫感营造":{"房源稀缺":["118的户型只剩20多套了好楼层只有这几套了","这栋楼一共就60户卖一套少一套现在不订可能就没了"],"时间紧迫":["今天是周末活动最后一天,这个价格只有今天能申请","月底冲业绩,领导给的权限最大,过了今天就没这个优惠了"]}},"communicationRhythm":{"开场阶段":{"关键动作":"亲切称呼,简单寒暄,确认客户关注点","时间占比":"5%","目标":"建立关系,了解需求"},"样板间带看":{"关键动作":"讲解户型功能→展示装修标准→强调细节品质","时间占比":"40%","目标":"体验产品优势"},"沙盘讲解":{"关键动作":"板块价值→周边配套→项目亮点→开发商介绍","时间占比":"30%","目标":"建立价值认知"},"洽谈阶段":{"关键动作":"算价格→对比竞品→解决顾虑→逼定成交","时间占比":"25%","目标":"促单成交"}},"needsMining":{"居住需求":["几个人住?有老人小孩吗?","主要是自住还是考虑投资?","现在住哪里?想改善哪些方面?"],"教育需求":["小孩在哪里上学?对学校距离有要求吗?","看重学校的哪些方面?"],"通勤需求":["在哪个位置上班?","主要开车还是坐地铁?","对地铁距离有要求吗?"],"预算需求":["你们总价想控制在多少以内?","是考虑按揭还是一次性?","月供能接受多少范围?"]},"painPointResponse":{"小区太小":{"对比竞品":"仁和春天29号院才29亩照样是千万级豪宅","承认事实":"14亩确实不大","普遍现象":"2.5环内都是小地块万景13亩中铁建8.8亩339的邦泰才11亩","转化优势":"但小区人少安静楼间距开阔200多米的楼间距比很多大楼盘还宽"},"担心保值":{"举例论证":"你看九龙仓擎天半岛就两栋楼现在二手房还是卖3万多望江名门一栋楼照样是千万级豪宅","承认顾虑":"我理解你担心小小区不保值","核心逻辑":"保值看的是地段我们槐树店是成华区核心板块地价19500未来只会涨不会跌"},"物业费高":{"价值分析":"但6块里有2块是增值服务招商物业是铂金服务这些服务外面花钱都买不来","价格补贴":"前三年开发商补贴1块你只需要交5块和其他改善盘差不多","理解感受":"我懂你觉得6块有点贵"}},"valueBuilding":{"产品价值塑造":["我们是用改善的价格,买豪宅的配置","3.2米层高、全落地窗、无机磨石车库,这些都是千万级豪宅的标配","118平实得132平得房率超过110%,市面上找不到第二家"],"地段价值塑造":["买房最重要的是地段、地段、还是地段","2.5环内的核心地段卖一块少一块,不可再生","槐树店是成华区房价天花板,买这里的房子保值有保障"]}}`
Run(context.Background(), []byte(reqBody))
err := skill.TalkSkillAdd(fiberCtx)
t.Log(err)
}
func Test_TalkSkillUpdate(t *testing.T) {
reqBody := `{"id":"698063ff5215bdb9c6344e88","advicerId":124,"projectId":3,"desc":"第0版本","closingTechniques":{"优惠策略":{"价格优惠":["双十一特价118㎡优惠后360-400万140㎡优惠后450-500万","渠道客户可额外申请优惠,相当于多一个点左右的优惠"],"附加价值":["车位双十一特惠5.3米长车位9.8万5.1米长车位8.8万"]},"决策推动":{"小步推进":["要不先交个小定保留房源?","可以先排个号,有优惠优先通知你","今天不定的话,我帮你留意好楼层"]},"紧迫感营造":{"房源稀缺":["118㎡只剩部分楼层140㎡只有二十多套公园景观房","好楼层卖一套少一套,性价比高的楼层不多了"],"时间紧迫":["现在是双十一/年底冲刺,有特价优惠","优惠是阶段性的,错过就没有了"]}},"communicationRhythm":{"开场阶段":{"关键动作":"亲切称呼,简单寒暄,确认看房重点","时间占比":"5%","目标":"建立关系,了解需求"},"样板间带看":{"关键动作":"细节讲解→户型优势→空间体验→竞品对比","时间占比":"40%","目标":"强化产品感知"},"沙盘讲解":{"关键动作":"板块价值→周边配套→项目亮点→开发商介绍","时间占比":"30%","目标":"建立价值认知"},"洽谈阶段":{"关键动作":"需求匹配→痛点应对→优惠释放→决策推动","时间占比":"25%","目标":"解决顾虑,促进成交"}},"needsMining":{"居住需求":["几个人住?有老人小孩吗?","主要是自住还是考虑投资?","现在住哪里?想改善哪些方面?","对房间数量、卫生间数量有要求吗?"],"通勤需求":["在哪个位置上班?","主要开车还是坐地铁?","对地铁距离有要求吗?"],"预算需求":["你们总价想控制在多少以内?","是考虑按揭还是一次性?","月供能接受多少范围?"]},"painPointResponse":{"地块太小":{"对比竞品":"339的邦泰才11亩人家是千万级豪宅","承认事实":"14亩确实不大","普遍现象":"2.5环内都是小地块万景13亩中铁建8.8亩","转化优势":"但人少安静,圈层更纯粹,楼间距反而更开阔"},"客户质疑开发商实力":{"合作背书":"招商物业首次外部合作,品牌物业认可开发商实力","实力展示":"公司有6000万吨铝矿年稳定收入10亿现金流雄厚","开发经验":"做房地产14年在宜宾、贵州开发超500万平米项目"},"担心南侧住宅用地遮挡阳光":{"澄清方向":"我们主采光面朝南,南侧住宅用地规划会错开楼间距,不会遮挡","竞品类比":"南侧用地会做高端大户型,开发商会考虑业主采光,不会影响我们的日照"},"担心新小区不保值":{"产品稀缺":"新规产品得房率高,未来政策限制赠送,产品竞争力强","地段支撑":"槐树店是成华区地价最高的板块,周边都是高端项目,地价和高端项目带动房价保值","需求保障":"未来大量业主会置换新规产品,该板块是首选,供需决定价值"},"物业费高":{"价值分析":"但6块里1块是增值服务保洁、送外卖、宠物服务","价格补贴":"前三年补贴到5块跟其他盘差不多","未来可协商":"后期业主委员会可以协商调整物业费仁恒滨河湾就从7.9谈到5块","理解感受":"我懂你,我们也觉得有点贵"}},"valueBuilding":{"产品价值塑造":["我们是用改善的价格,买豪宅的标准","很多细节都是千万级豪宅才有的配置","外立面成本比竞品高,单价却相当","3.2米层高、无机磨石车库这些都是高端配置"],"地段价值塑造":["买房最重要的是地段、地段、还是地段","核心地段的核心资产才保值增值","2.5环内的地卖一块少一块,不可再生","槐树店是成华区地价最高的板块,地价高对应房价支撑强"]}}`
Run(context.Background(), []byte(reqBody))
err := skill.TalkSkillUpdate(fiberCtx)
t.Log(err)
}
func Test_TalkSkillList(t *testing.T) {
reqBody := `{"projectId":1}`
Run(context.Background(), []byte(reqBody))
err := skill.TalkSkillList(fiberCtx)
t.Log(err)
}
func Test_TalkSkillDel(t *testing.T) {
reqBody := `{"id":"698056073059550befc4f0da"}`
Run(context.Background(), []byte(reqBody))
err := skill.TalkSkillDel(fiberCtx)
t.Log(err)
}
var (
skill *TalkSkillService
)
// run 函数是程序的入口函数,负责初始化和配置各个组件

View File

@ -16,5 +16,10 @@ var ProviderSetServices = wire.NewSet(
NewHistoryService,
NewCapabilityService,
NewCronService,
advice.NewAdviceService,
advice.NewFileService,
advice.NewAdvicerService,
advice.NewTalkSkillService,
advice.NewProjectService,
advice.NewClientService,
advice.NewChatService,
)