From ec5ff4a0a9c4141362640841864b57643660b67b Mon Sep 17 00:00:00 2001 From: renzhiyuan <465386466@qq.com> Date: Thu, 5 Feb 2026 13:37:42 +0800 Subject: [PATCH] =?UTF-8?q?refactor:=20=E9=87=8D=E6=9E=84advice=E6=A8=A1?= =?UTF-8?q?=E5=9D=97=E5=8F=8A=E6=B7=BB=E5=8A=A0MongoDB=E6=94=AF=E6=8C=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/server/wire.go | 4 +- config/config.yaml | 8 + config/config_env.yaml | 8 + config/config_test.yaml | 3 +- docker-compose.yml | 5 +- go.mod | 11 +- go.sum | 20 +- internal/biz/advice_advicer.go | 166 +++++++- internal/biz/advice_chat.go | 124 ++++++ internal/biz/advice_client.go | 150 +++++++ internal/biz/advice_file.go | 58 ++- internal/biz/advice_project.go | 100 +++++ internal/biz/advice_skill.go | 164 ++++++++ internal/biz/llm_service/third_party/hsyq.go | 26 +- internal/biz/provider_set.go | 4 + internal/config/config.go | 6 +- internal/data/model/ai_advice_advicer.gen.go | 13 +- .../data/model/ai_advice_advicer_fn.gen.go | 37 ++ internal/data/mongo_model/advicer_client.go | 97 +++++ internal/data/mongo_model/advicer_project.go | 150 +++++++ .../data/mongo_model/advicer_talk_skill.go | 137 ++++++ internal/data/mongo_model/advicer_version.go | 155 +++++++ internal/data/mongo_model/common.go | 24 ++ internal/data/mongo_model/provider_set.go | 10 + internal/entitys/advicer.go | 391 +----------------- internal/entitys/advicer_data.go | 178 +++++++- internal/pkg/mongo.go | 15 +- internal/server/http.go | 19 +- internal/server/router/router.go | 79 ++-- internal/services/advice/advicer.go | 78 ++++ internal/services/advice/advicer_test.go | 95 +++-- internal/services/advice/chat.go | 117 ++++++ internal/services/advice/chat_test.go | 37 ++ internal/services/advice/client.go | 61 +++ internal/services/advice/client_test.go | 39 ++ internal/services/advice/data.go | 60 --- internal/services/advice/project.go | 52 +++ internal/services/advice/project_test.go | 30 ++ internal/services/advice/talk_skill.go | 61 +++ internal/services/advice/talk_skill_test.go | 41 ++ internal/services/provider_set.go | 7 +- 41 files changed, 2237 insertions(+), 603 deletions(-) create mode 100644 internal/biz/advice_chat.go create mode 100644 internal/biz/advice_client.go create mode 100644 internal/biz/advice_project.go create mode 100644 internal/biz/advice_skill.go create mode 100644 internal/data/model/ai_advice_advicer_fn.gen.go create mode 100644 internal/data/mongo_model/advicer_client.go create mode 100644 internal/data/mongo_model/advicer_project.go create mode 100644 internal/data/mongo_model/advicer_talk_skill.go create mode 100644 internal/data/mongo_model/advicer_version.go create mode 100644 internal/data/mongo_model/common.go create mode 100644 internal/data/mongo_model/provider_set.go create mode 100644 internal/services/advice/advicer.go create mode 100644 internal/services/advice/chat.go create mode 100644 internal/services/advice/chat_test.go create mode 100644 internal/services/advice/client.go create mode 100644 internal/services/advice/client_test.go delete mode 100644 internal/services/advice/data.go create mode 100644 internal/services/advice/project.go create mode 100644 internal/services/advice/project_test.go create mode 100644 internal/services/advice/talk_skill.go create mode 100644 internal/services/advice/talk_skill_test.go diff --git a/cmd/server/wire.go b/cmd/server/wire.go index b4f4437..711e7b2 100644 --- a/cmd/server/wire.go +++ b/cmd/server/wire.go @@ -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, )) } diff --git a/config/config.yaml b/config/config.yaml index 6166634..c1300c2 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -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" diff --git a/config/config_env.yaml b/config/config_env.yaml index 1b1a8bd..e9f7bbb 100644 --- a/config/config_env.yaml +++ b/config/config_env.yaml @@ -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" diff --git a/config/config_test.yaml b/config/config_test.yaml index 003be9e..df57522 100644 --- a/config/config_test.yaml +++ b/config/config_test.yaml @@ -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 diff --git a/docker-compose.yml b/docker-compose.yml index ab49b41..26198fd 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -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 diff --git a/go.mod b/go.mod index eb351cb..5634e43 100644 --- a/go.mod +++ b/go.mod @@ -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 diff --git a/go.sum b/go.sum index b069ba2..1d57f5e 100644 --- a/go.sum +++ b/go.sum @@ -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= diff --git a/internal/biz/advice_advicer.go b/internal/biz/advice_advicer.go index af49954..935d58f 100644 --- a/internal/biz/advice_advicer.go +++ b/internal/biz/advice_advicer.go @@ -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 +} diff --git a/internal/biz/advice_chat.go b/internal/biz/advice_chat.go new file mode 100644 index 0000000..58c14fb --- /dev/null +++ b/internal/biz/advice_chat.go @@ -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 +} diff --git a/internal/biz/advice_client.go b/internal/biz/advice_client.go new file mode 100644 index 0000000..77fd54a --- /dev/null +++ b/internal/biz/advice_client.go @@ -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 +} diff --git a/internal/biz/advice_file.go b/internal/biz/advice_file.go index 897843b..65c8431 100644 --- a/internal/biz/advice_file.go +++ b/internal/biz/advice_file.go @@ -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()) diff --git a/internal/biz/advice_project.go b/internal/biz/advice_project.go new file mode 100644 index 0000000..bd48d77 --- /dev/null +++ b/internal/biz/advice_project.go @@ -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 +} diff --git a/internal/biz/advice_skill.go b/internal/biz/advice_skill.go new file mode 100644 index 0000000..38e64c7 --- /dev/null +++ b/internal/biz/advice_skill.go @@ -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 + +} diff --git a/internal/biz/llm_service/third_party/hsyq.go b/internal/biz/llm_service/third_party/hsyq.go index 5257774..6b74e0c 100644 --- a/internal/biz/llm_service/third_party/hsyq.go +++ b/internal/biz/llm_service/third_party/hsyq.go @@ -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 } diff --git a/internal/biz/provider_set.go b/internal/biz/provider_set.go index cb50b0e..4462609 100644 --- a/internal/biz/provider_set.go +++ b/internal/biz/provider_set.go @@ -25,4 +25,8 @@ var ProviderSetBiz = wire.NewSet( NewAdviceFileBiz, third_party.NewHsyq, NewAdviceAdvicerBiz, + NewAdviceSkillBiz, + NewAdviceProjectBiz, + NewAdviceClientBiz, + NewAdviceChatBiz, ) diff --git a/internal/config/config.go b/internal/config/config.go index d9201dd..232168f 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -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 { diff --git a/internal/data/model/ai_advice_advicer.gen.go b/internal/data/model/ai_advice_advicer.gen.go index 67f7035..449547a 100644 --- a/internal/data/model/ai_advice_advicer.gen.go +++ b/internal/data/model/ai_advice_advicer.gen.go @@ -12,12 +12,13 @@ const TableNameAiAdviceAdvicer = "ai_advice_advicer" // AiAdviceAdvicer mapped from table 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 diff --git a/internal/data/model/ai_advice_advicer_fn.gen.go b/internal/data/model/ai_advice_advicer_fn.gen.go new file mode 100644 index 0000000..adeb509 --- /dev/null +++ b/internal/data/model/ai_advice_advicer_fn.gen.go @@ -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), + } +} \ No newline at end of file diff --git a/internal/data/mongo_model/advicer_client.go b/internal/data/mongo_model/advicer_client.go new file mode 100644 index 0000000..053ff42 --- /dev/null +++ b/internal/data/mongo_model/advicer_client.go @@ -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 "客户信息" +} diff --git a/internal/data/mongo_model/advicer_project.go b/internal/data/mongo_model/advicer_project.go new file mode 100644 index 0000000..05f7d64 --- /dev/null +++ b/internal/data/mongo_model/advicer_project.go @@ -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 "开发商背书" +} diff --git a/internal/data/mongo_model/advicer_talk_skill.go b/internal/data/mongo_model/advicer_talk_skill.go new file mode 100644 index 0000000..e3744fa --- /dev/null +++ b/internal/data/mongo_model/advicer_talk_skill.go @@ -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 "沟通节奏控制" +} diff --git a/internal/data/mongo_model/advicer_version.go b/internal/data/mongo_model/advicer_version.go new file mode 100644 index 0000000..1a66049 --- /dev/null +++ b/internal/data/mongo_model/advicer_version.go @@ -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 "代表性对话示例" +} diff --git a/internal/data/mongo_model/common.go b/internal/data/mongo_model/common.go new file mode 100644 index 0000000..3b20438 --- /dev/null +++ b/internal/data/mongo_model/common.go @@ -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 +} diff --git a/internal/data/mongo_model/provider_set.go b/internal/data/mongo_model/provider_set.go new file mode 100644 index 0000000..7520345 --- /dev/null +++ b/internal/data/mongo_model/provider_set.go @@ -0,0 +1,10 @@ +package mongo_model + +import "github.com/google/wire" + +var ProviderSetMongo = wire.NewSet( + NewAdvicerVersionMongo, + NewAdvicerTalkSkillMongo, + NewAdvicerProjectMongo, + NewAdvicerClientMongo, +) diff --git a/internal/entitys/advicer.go b/internal/entitys/advicer.go index 22968c0..eea6d85 100644 --- a/internal/entitys/advicer.go +++ b/internal/entitys/advicer.go @@ -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"` } diff --git a/internal/entitys/advicer_data.go b/internal/entitys/advicer_data.go index ed72545..4a22a67 100644 --- a/internal/entitys/advicer_data.go +++ b/internal/entitys/advicer_data.go @@ -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"` } diff --git a/internal/pkg/mongo.go b/internal/pkg/mongo.go index c6dfe76..5642c32 100644 --- a/internal/pkg/mongo.go +++ b/internal/pkg/mongo.go @@ -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()) +} diff --git a/internal/server/http.go b/internal/server/http.go index 53446c8..91191a1 100644 --- a/internal/server/http.go +++ b/internal/server/http.go @@ -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 } diff --git a/internal/server/router/router.go b/internal/server/router/router.go index b7a6131..2b66d20 100644 --- a/internal/server/router/router.go +++ b/internal/server/router/router.go @@ -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(), }) diff --git a/internal/services/advice/advicer.go b/internal/services/advice/advicer.go new file mode 100644 index 0000000..51d1d86 --- /dev/null +++ b/internal/services/advice/advicer.go @@ -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) +} diff --git a/internal/services/advice/advicer_test.go b/internal/services/advice/advicer_test.go index 3ced2d2..056ac9f 100644 --- a/internal/services/advice/advicer_test.go +++ b/internal/services/advice/advicer_test.go @@ -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{}, } diff --git a/internal/services/advice/chat.go b/internal/services/advice/chat.go new file mode 100644 index 0000000..4759b7b --- /dev/null +++ b/internal/services/advice/chat.go @@ -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) +} diff --git a/internal/services/advice/chat_test.go b/internal/services/advice/chat_test.go new file mode 100644 index 0000000..d60eb91 --- /dev/null +++ b/internal/services/advice/chat_test.go @@ -0,0 +1,37 @@ +package advice + +import ( + "ai_scheduler/internal/entitys" + "context" + "encoding/json" + + "testing" +) + +func Test_Regis(t *testing.T) { + reqBody := `{"advicerVersionId":"698315725d0077330399f704","clientId":"698199fa0c5f4ae098e009ab","talkSkillId":"698063ff5215bdb9c6344e88"}` + Run(context.Background(), []byte(reqBody)) + err := chat.Regis(fiberCtx) + t.Log(err) +} + +func Test_Chat(t *testing.T) { + reqBody := `{"sessionId":"075a0ec7-2163-4cfa-ae88-c8b8e60e1c66","content":"说话！"}` + Run(context.Background(), []byte(reqBody)) + err := chat.Chat(fiberCtx) + t.Log(err) +} + +func Test_Cache(t *testing.T) { + var chatData entitys.ChatData + chatDataJson := `{"clientInfo":{"personalInfo":{"name":"唐先生1","gender":"男","location":"成都北门","isFirstHome":false,"familyOrganize":"夫妻+1孩+父母同住"},"purchasePurpose":{"primaryPurpose":"改善居住条件","secondaryPurpose":"资产保值，方便子女上学","decisionMakers":"夫妻双方"},"coreDemands":{"totalBudget":"350-400万","preferredLayout":"118㎡四房三卫双套房","coreAppeal":"在预算内满足家庭居住功能，确保房产保值，临近学校"},"concerns":["总价超预算风险","板块保值能力","小区小是否影响居住体验","开发商资金实力"],"decisionProfile":["预算导向，严格控制总价","重点关注户型功能性和实用性","需要对比板块发展潜力","对开发商交付能力有顾虑"]},"talkSkill":{"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环内的地卖一块少一块，不可再生","槐树店是成华区地价最高的板块，地价高对应房价支撑强"]},"closingTechniques":{"优惠策略":{"价格优惠":["双十一特价，118㎡优惠后360-400万，140㎡优惠后450-500万","渠道客户可额外申请优惠，相当于多一个点左右的优惠"],"附加价值":["车位双十一特惠，5.3米长车位9.8万，5.1米长车位8.8万"]},"决策推动":{"小步推进":["要不先交个小定保留房源？","可以先排个号，有优惠优先通知你","今天不定的话，我帮你留意好楼层"]},"紧迫感营造":{"房源稀缺":["118㎡只剩部分楼层，140㎡只有二十多套公园景观房","好楼层卖一套少一套，性价比高的楼层不多了"],"时间紧迫":["现在是双十一/年底冲刺，有特价优惠","优惠是阶段性的，错过就没有了"]}},"communicationRhythm":{"开场阶段":{"关键动作":"亲切称呼，简单寒暄，确认看房重点","时间占比":"5%","目标":"建立关系，了解需求"},"样板间带看":{"关键动作":"细节讲解→户型优势→空间体验→竞品对比","时间占比":"40%","目标":"强化产品感知"},"沙盘讲解":{"关键动作":"板块价值→周边配套→项目亮点→开发商介绍","时间占比":"30%","目标":"建立价值认知"},"洽谈阶段":{"关键动作":"需求匹配→痛点应对→优惠释放→决策推动","时间占比":"25%","目标":"解决顾虑，促进成交"}}},"projectInfo":{"regionValue":{"区位层级":["成华区2.5环内侧，槐树店板块是成华区number one板块","北接339商圈，西靠万象城，东临火车东站","属于槐树店崔怀板块，成华区目前最好的开发板块"],"发展规划":["槐树店是成华区未来的富人区，板块还有大量待开发土地","未来这个区域会形成连片高端居住区，城市界面会越来越好"],"地价论证":["我们地价19500，比28板块贵5000多","华润华城府地价20400，我们和它同属一个板块，地价差距小","面粉都这么贵，面包不可能便宜"],"板块热度":["从2021年新希望锦麟一品开始，这边全是高端盘","龙湖最高端的滨江云河颂在这里，卖得特别火","各大品牌开发商都在这边拿地，未来全是改善盘"]},"competitionComparison":{"华润置地云上":{"价格对比":"他们单价35000左右，还靠近安置小区","优势突出":"我们小区纯粹，没有安置小区，居住环境更安静","优点承认":"华润品牌影响力大"},"招商璟宸序":{"价格对比":"他们单价32000左右，但地段在28板块，地价比我们便宜5000","优势突出":"我们地段在槐树店，是成华区核心板块，未来增值空间更大","优点承认":"招商品牌大，物业也是自己的"},"龙湖滨江云河颂":{"价格对比":"他们单价32000-35000，但得房率只有95%，套内算下来36000+","优势突出":"我们118平实得132平，套内单价才33000，还做四房三卫，他们143平才双卫","优点承认":"龙湖位置确实好，能看沙河公园"}},"coreSellingPoints":{"产品配置高端":"3.2米层高、全玻璃幕墙+三层中空玻璃、无机磨石车库、方太Y9烟机、高仪卫浴、国千木作柜体","地段稀缺性":"成华区2.5环内侧槐树店板块，成华区房价天花板区域，被万象城、339、火车东站包围","得房率高":"118平实得132平，得房率超110%，四房三卫双套房设计，市面上同面积段没有竞品","物业优质":"招商局铂金物业，有夜间送外卖、免费宠物喂养、全屋保洁等增值服务"},"supportingFacilities":{"交通配套":{"地铁":"7号线双店路站350米，4号线槐树店站550米，未来还有12号线","通达性":"到万象城2个站，到华西锦江院区30分钟车程","道路":"中环路、成洛大道，到春熙路5个站，到火车东站2个站"},"医疗配套":{"三甲医院":"市六医院、市二医院3公里内","顶尖医疗":"华西医院锦江院区、华西本部30分钟车程"},"商业配套":{"便利性":"到万象城2个站，到339商圈3个站","核心商圈":"万象城商圈、339商圈","社区商业":"和悦广场、东方希望上东里"},"教育配套":{"小学":"成华小学，1-3年级在项目附近，4-6年级在二环内","生源优势":"周边新盘都是300万+，生源纯粹"}},"developerBacking":{"公司实力":"中信资产，多元化民营企业，涉及矿产、有色金属、生态农业、地产开发","合作方":"招商局物业，百年央企，首次与外部企业合作提供铂金服务","开发经验":"2011年开始做地产，在宜宾、贵州开发超过500万平米，成都是首个项目","资金安全":"在河南渑池有两座优质铝矿，每年稳定收入10亿，现金流充足"}},"advicerInfo":{"name":"张三111","birth":"1990-01-01","gender":"男","working_years":"10年"},"advicerVersion":{"dialectFeatures":{"region":"四川成都话","intensity":0.6,"KeyWords":null},"sentencePatterns":{"openingMode":["我给你介绍一下","我们先来看一下","这边请"],"explanationMode":["是这样的","我跟你讲","你发现没得","说白了"],"confirmationMode":["对吧？","是不是嘛？","你晓得不？","明白了噻？","对不对？"],"summaryMode":["所以说","简单说就是","其实"],"transitionMode":["然后的话","再其次","还有一点","另外"]},"toneTags":{"enthusiasm":0.8,"patience":0.9,"confidence":0.85,"friendliness":0.8,"persuasion":0.75},"personalityTags":["耐心细致","专业务实","经验丰富","善于引导"],"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号院，都是小地块但照样是高端保值盘。核心还是地段，我们在槐树店这个成华区最贵的板块，保值根本没问题！"}]}}` + + err := json.Unmarshal([]byte(chatDataJson), &chatData) + t.Log(err) +} + +var ( + chat *ChatService +) + +// run 函数是程序的入口函数，负责初始化和配置各个组件 diff --git a/internal/services/advice/client.go b/internal/services/advice/client.go new file mode 100644 index 0000000..d884414 --- /dev/null +++ b/internal/services/advice/client.go @@ -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) +} diff --git a/internal/services/advice/client_test.go b/internal/services/advice/client_test.go new file mode 100644 index 0000000..f94e2cb --- /dev/null +++ b/internal/services/advice/client_test.go @@ -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 +) diff --git a/internal/services/advice/data.go b/internal/services/advice/data.go deleted file mode 100644 index fc12207..0000000 --- a/internal/services/advice/data.go +++ /dev/null @@ -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) -} diff --git a/internal/services/advice/project.go b/internal/services/advice/project.go new file mode 100644 index 0000000..22cd036 --- /dev/null +++ b/internal/services/advice/project.go @@ -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) +} diff --git a/internal/services/advice/project_test.go b/internal/services/advice/project_test.go new file mode 100644 index 0000000..6375353 --- /dev/null +++ b/internal/services/advice/project_test.go @@ -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 diff --git a/internal/services/advice/talk_skill.go b/internal/services/advice/talk_skill.go new file mode 100644 index 0000000..d47ccec --- /dev/null +++ b/internal/services/advice/talk_skill.go @@ -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) +} diff --git a/internal/services/advice/talk_skill_test.go b/internal/services/advice/talk_skill_test.go new file mode 100644 index 0000000..9de7e30 --- /dev/null +++ b/internal/services/advice/talk_skill_test.go @@ -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 函数是程序的入口函数，负责初始化和配置各个组件 diff --git a/internal/services/provider_set.go b/internal/services/provider_set.go index dfce67a..75178d5 100644 --- a/internal/services/provider_set.go +++ b/internal/services/provider_set.go @@ -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, )