Beta77

applogic/gensokyo.go

@@ -99,13 +99,20 @@ func (app *App) GensokyoHandler(w http.ResponseWriter, r *http.Request) {
 		fmt.Printf("收到prompt参数: %s\n", promptstr)
 	}
 
-	// 读取URL参数 "prompt"
+	// 读取URL参数 "selfid"
 	selfid := r.URL.Query().Get("selfid")
 	if selfid != "" {
 		// 使用 prompt 变量进行后续处理
 		fmt.Printf("收到selfid参数: %s\n", selfid)
 	}
 
+	// 读取URL参数 "api"
+	api := r.URL.Query().Get("api")
+	if selfid != "" {
+		// 使用 prompt 变量进行后续处理
+		fmt.Printf("收到api参数: %s\n", selfid)
+	}
+
 	// 打印日志信息，包括prompt参数
 	fmtf.Printf("收到onebotv11信息: %+v\n", string(body))
 
@@ -358,10 +365,15 @@ func (app *App) GensokyoHandler(w http.ResponseWriter, r *http.Request) {
 
 		// 构建并发送请求到conversation接口
 		port := config.GetPort()
-		portStr := fmtf.Sprintf(":%d", port)
+		portStr := fmt.Sprintf(":%d", port)
 
-		// 初始化URL
-		baseURL := "http://127.0.0.1" + portStr + "/conversation"
+		// 初始化URL，根据api参数动态调整路径
+		basePath := "/conversation"
+		if api != "" {
+			fmtf.Printf("收到api参数: %s\n", api)
+			basePath = "/" + api // 动态替换conversation部分为api参数值
+		}
+		baseURL := "http://127.0.0.1" + portStr + basePath
 
 		// 使用net/url包来构建和编码URL
 		urlParams := url.Values{}
@@ -375,7 +387,7 @@ func (app *App) GensokyoHandler(w http.ResponseWriter, r *http.Request) {
 			fullURL += "?" + urlParams.Encode()
 		}
 
-		fmtf.Printf("Generated URL:%v", fullURL)
+		fmtf.Printf("Generated URL:%v\n", fullURL)
 
 		// 请求模型还是使用原文请求
 		requestmsg := message.Message.(string)

config/config.go

@@ -1142,3 +1142,13 @@ func GetPathToken() string {
 	}
 	return ""
 }
+
+// 获取开启全部api
+func GetAllApi() bool {
+	mu.Lock()
+	defer mu.Unlock()
+	if instance != nil {
+		return instance.Settings.AllApi
+	}
+	return false
+}

main.go

@@ -157,6 +157,13 @@ func main() {
 		log.Printf("Unknown API type: %d", apiType)
 	}
 
+	if config.GetAllApi() {
+		http.HandleFunc("/conversation_gpt", app.ChatHandlerChatgpt)
+		http.HandleFunc("/conversation_hunyuan", app.ChatHandlerHunyuan)
+		http.HandleFunc("/conversation_ernie", app.ChatHandlerErnie)
+		http.HandleFunc("/conversation_rwkv", app.ChatHandlerRwkv)
+	}
+
 	exePath, err := os.Executable()
 	if err != nil {
 		log.Fatal(err)

structs/struct.go

@@ -208,6 +208,7 @@ type WXFunctionCall struct {
 }
 
 type Settings struct {
+	AllApi       bool     `yaml:"allApi"`
 	SecretId     string   `yaml:"secretId"`
 	SecretKey    string   `yaml:"secretKey"`
 	Region       string   `yaml:"region"`

template/config_template.go

@@ -5,7 +5,8 @@ version: 1
 settings:
 
   #通用配置项
-  useSse : false                                 #智能体场景开启,其他场景,比如普通onebotv11不开启
+  allApi : false                                #以conversation_ernie conversation_hunyuan形式同时开启全部api,请设置好iPWhiteList避免被盗用.
+  useSse : false                                #智能体场景开启,其他场景,比如普通onebotv11不开启
   port : 46233                                  #本程序监听端口,支持gensokyo http上报,   请在gensokyo的反向http配置加入 post_url: ["http://127.0.0.1:port/gensokyo"] 
   path : "http://123.123.123.123:11111"         #调用gensokyo api的地址,填入 gensokyo 的 正向http地址   http_address: "0.0.0.0:46231"  对应填入 "http://127.0.0.1:46231"
   apiType : 0                                   #0=混元 1=文心(文心平台包含了N种模型...) 2=gpt