feat: 集成 AI-Core gRPC 文档解析服务
- 新增 AICoreClient 客户端 - 添加 document_parser_client.go - 知识库服务集成 AI-Core 解析 - 配置中添加 ai_core_service_addr Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -14,8 +14,14 @@ import (
|
||||
type Config struct {
|
||||
Port string
|
||||
JWTSecret string
|
||||
DatabaseURL string
|
||||
DatabaseHost string
|
||||
DatabasePort string
|
||||
DatabaseUser string
|
||||
DatabasePassword string
|
||||
DatabaseName string
|
||||
DatabaseURL string // 拼接后的完整连接字符串
|
||||
PythonServiceURL string
|
||||
AICoreServiceAddr string // AI-Core gRPC 服务地址,如 "localhost:50051"
|
||||
// 文件上传配置
|
||||
UploadMode string // "local" 或 "minio"
|
||||
UploadLocalPath string // 本地存储路径,如 "resource/files"
|
||||
@@ -39,7 +45,13 @@ func Load() *Config {
|
||||
viper.SetDefault("port", "8080")
|
||||
viper.SetDefault("jwt_secret", "your-secret-key-change-in-production")
|
||||
viper.SetDefault("python_service_url", "http://localhost:8081")
|
||||
viper.SetDefault("database_url", "root:root@tcp(localhost:3306)/x_agents?charset=utf8mb4&parseTime=True&loc=Local")
|
||||
viper.SetDefault("ai_core_service_addr", "localhost:50051")
|
||||
// 数据库默认配置
|
||||
viper.SetDefault("database_host", "localhost")
|
||||
viper.SetDefault("database_port", "3306")
|
||||
viper.SetDefault("database_user", "root")
|
||||
viper.SetDefault("database_password", "root")
|
||||
viper.SetDefault("database_name", "x_agents")
|
||||
// 文件上传默认配置
|
||||
viper.SetDefault("upload_mode", "local")
|
||||
viper.SetDefault("upload_local_path", "resource/files")
|
||||
@@ -54,11 +66,26 @@ func Load() *Config {
|
||||
log.Printf("Using default config: %v", err)
|
||||
}
|
||||
|
||||
// 拼接数据库连接字符串
|
||||
dbHost := viper.GetString("database_host")
|
||||
dbPort := viper.GetString("database_port")
|
||||
dbUser := viper.GetString("database_user")
|
||||
dbPassword := viper.GetString("database_password")
|
||||
dbName := viper.GetString("database_name")
|
||||
databaseURL := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||||
dbUser, dbPassword, dbHost, dbPort, dbName)
|
||||
|
||||
return &Config{
|
||||
Port: viper.GetString("port"),
|
||||
JWTSecret: viper.GetString("jwt_secret"),
|
||||
DatabaseURL: viper.GetString("database_url"),
|
||||
PythonServiceURL: viper.GetString("python_service_url"),
|
||||
Port: viper.GetString("port"),
|
||||
JWTSecret: viper.GetString("jwt_secret"),
|
||||
DatabaseURL: databaseURL,
|
||||
DatabaseHost: dbHost,
|
||||
DatabasePort: dbPort,
|
||||
DatabaseUser: dbUser,
|
||||
DatabasePassword: dbPassword,
|
||||
DatabaseName: dbName,
|
||||
PythonServiceURL: viper.GetString("python_service_url"),
|
||||
AICoreServiceAddr: viper.GetString("ai_core_service_addr"),
|
||||
// 文件上传配置
|
||||
UploadMode: viper.GetString("upload_mode"),
|
||||
UploadLocalPath: viper.GetString("upload_local_path"),
|
||||
|
||||
Reference in New Issue
Block a user