github-clones
/
override
mirror of https://github.com/linux-do/override.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge fcc40956fa into 8603e7429e

This commit is contained in:
lingfeng 2024-12-18 09:05:18 +00:00 committed by GitHub
parent 8603e7429e fcc40956fa
commit ff34922b5d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 98 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -10,7 +10,7 @@ RUN CGO_ENABLED=0 go build -ldflags="-w -s" -o override
FROM alpine:latest FROM alpine:latest
RUN apk --no-cache add ca-certificates RUN apk --no-cache add ca-certificates tzdata && cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo "Asia/Shanghai" > /etc/timezone
COPY --from=builder /app/override /usr/local/bin/ COPY --from=builder /app/override /usr/local/bin/
COPY config.json.example /app/config.json COPY config.json.example /app/config.json

6
docker-compose.yml
View File

@ -4,9 +4,9 @@ services:
container_name: override-app container_name: override-app
restart: always restart: always
build: build:
context: . context: .
dockerfile: Dockerfile dockerfile: Dockerfile
volumes: volumes:
- ./config.json:/app/config.json - ./config.json:/app/config.json
ports: ports:
- "8181:8181" - "8181:8181"

102
main.go
View File

@ -10,8 +10,10 @@ import (
"github.com/tidwall/gjson" "github.com/tidwall/gjson"
"github.com/tidwall/sjson" "github.com/tidwall/sjson"
"golang.org/x/net/http2" "golang.org/x/net/http2"
"html/template"
"io" "io"
"log" "log"
"math/rand"
"net/http" "net/http"
"net/url" "net/url"
"os" "os"
@ -19,7 +21,6 @@ import (
"strconv" "strconv"
"strings" "strings"
"time" "time"
"math/rand"
) )
const DefaultInstructModel = "gpt-3.5-turbo-instruct" const DefaultInstructModel = "gpt-3.5-turbo-instruct"
@ -51,6 +52,17 @@ type config struct {
AuthToken string `json:"auth_token"` AuthToken string `json:"auth_token"`
} }
// 转换结构体为键值对 Map
func structToMap(cfg config) map[string]interface{} {
result := make(map[string]interface{})
v := reflect.ValueOf(cfg)
t := reflect.TypeOf(cfg)
for i := 0; i < v.NumField(); i++ {
result[t.Field(i).Name] = v.Field(i).Interface()
}
return result
}
func readConfig() *config { func readConfig() *config {
var configPath string var configPath string
if len(os.Args) > 1 { if len(os.Args) > 1 {
@ -194,6 +206,7 @@ func (s *ProxyService) InitRoutes(e *gin.Engine) {
e.GET("/_ping", s.pong) e.GET("/_ping", s.pong)
e.GET("/models", s.models) e.GET("/models", s.models)
e.GET("/v1/models", s.models) e.GET("/v1/models", s.models)
e.GET("/", s.index)
authToken := s.cfg.AuthToken // replace with your dynamic value as needed authToken := s.cfg.AuthToken // replace with your dynamic value as needed
if authToken != "" { if authToken != "" {
// 鉴权 // 鉴权
@ -220,6 +233,18 @@ type Pong struct {
Ns1 string `json:"ns1"` Ns1 string `json:"ns1"`
} }
func (s *ProxyService) index(c *gin.Context) {
cfg := *readConfig()
if c.Query("format") == "json" {
c.JSON(http.StatusOK, cfg)
} else {
c.HTML(http.StatusOK, "configPage", gin.H{
"Config": structToMap(cfg),
})
}
}
func (s *ProxyService) pong(c *gin.Context) { func (s *ProxyService) pong(c *gin.Context) {
c.JSON(http.StatusOK, Pong{ c.JSON(http.StatusOK, Pong{
Now: time.Now().Second(), Now: time.Now().Second(),
@ -478,7 +503,7 @@ func (s *ProxyService) completions(c *gin.Context) {
} }
func contains(arr []string, str string) bool { func contains(arr []string, str string) bool {
return strings.Contains(strings.Join(arr, ","), str) return strings.Contains(strings.Join(arr, ","), str)
} }
func (s *ProxyService) codeCompletions(c *gin.Context) { func (s *ProxyService) codeCompletions(c *gin.Context) {
@ -506,7 +531,7 @@ func (s *ProxyService) codeCompletions(c *gin.Context) {
} }
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer " + getRandomApiKey(s.cfg.CodexApiKey)) req.Header.Set("Authorization", "Bearer "+getRandomApiKey(s.cfg.CodexApiKey))
if "" != s.cfg.CodexApiOrganization { if "" != s.cfg.CodexApiOrganization {
req.Header.Set("OpenAI-Organization", s.cfg.CodexApiOrganization) req.Header.Set("OpenAI-Organization", s.cfg.CodexApiOrganization)
} }
@ -547,12 +572,12 @@ func (s *ProxyService) codeCompletions(c *gin.Context) {
// 随机取一个apiKey // 随机取一个apiKey
func getRandomApiKey(paramStr string) string { func getRandomApiKey(paramStr string) string {
params := strings.Split(paramStr, ",") params := strings.Split(paramStr, ",")
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
randomIndex := rand.Intn(len(params)) randomIndex := rand.Intn(len(params))
fmt.Println("Code completion API Key index:", randomIndex) fmt.Println("Code completion API Key index:", randomIndex)
fmt.Println("Code completion API Key:", strings.TrimSpace(params[randomIndex])) fmt.Println("Code completion API Key:", strings.TrimSpace(params[randomIndex]))
return strings.TrimSpace(params[randomIndex]) return strings.TrimSpace(params[randomIndex])
} }
func ConstructRequestBody(body []byte, cfg *config) []byte { func ConstructRequestBody(body []byte, cfg *config) []byte {
@ -607,12 +632,73 @@ func constructWithChatModel(body []byte, messages interface{}) []byte {
return []byte(jsonStr) return []byte(jsonStr)
} }
func getHTMLTemplate() string {
return `
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Config Page</title>
<style>
body {
font-family: Arial, sans-serif;
margin: 20px;
padding: 0;
}
.container {
max-width: 800px;
margin: 0 auto;
}
table {
width: 100%;
border-collapse: collapse;
margin: 20px 0;
}
table, th, td {
border: 1px solid #ccc;
}
th, td {
padding: 10px;
text-align: left;
}
th {
background-color: #f4f4f4;
}
</style>
</head>
<body>
<div class="container">
<h1>Configuration</h1>
<p>Below is the configuration data:</p>
<table>
<thead>
<tr>
<th>Key</th>
<th>Value</th>
</tr>
</thead>
<tbody>
{{ range $key, $value := .Config }}
<tr>
<td>{{ $key }}</td>
<td>{{ $value }}</td>
</tr>
{{ end }}
</tbody>
</table>
</div>
</body>
</html>
`
}
func main() { func main() {
cfg := readConfig() cfg := readConfig()
gin.SetMode(gin.ReleaseMode) gin.SetMode(gin.ReleaseMode)
r := gin.Default() r := gin.Default()
r.SetHTMLTemplate(template.Must(template.New("configPage").Parse(getHTMLTemplate())))
proxyService, err := NewProxyService(cfg) proxyService, err := NewProxyService(cfg)
if nil != err { if nil != err {
log.Fatal(err) log.Fatal(err)