/
image.go
116 lines (92 loc) 路 2.66 KB
/
image.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package proxy
import (
"encoding/json"
goopenai "github.com/sashabaranov/go-openai"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
func logCreateImageRequest(log *zap.Logger, ir *goopenai.ImageRequest, prod, private bool, cid string) {
if prod {
fields := []zapcore.Field{
zap.String(correlationId, cid),
zap.String("model", ir.Model),
zap.Int("n", ir.N),
zap.String("quality", ir.Quality),
zap.String("size", ir.Size),
zap.String("style", ir.Style),
zap.String("response_format", ir.ResponseFormat),
zap.String("user", ir.User),
}
if !private && len(ir.Prompt) != 0 {
fields = append(fields, zap.String("prompt", ir.Prompt))
}
log.Info("openai create image request", fields...)
}
}
func logEditImageRequest(log *zap.Logger, prompt, model string, n int, size, responseFormat, user string, prod, private bool, cid string) {
if prod {
fields := []zapcore.Field{
zap.String(correlationId, cid),
}
if !private && len(prompt) != 0 {
fields = append(fields, zap.String("prompt", prompt))
}
if len(model) != 0 {
fields = append(fields, zap.String("prompt", model))
}
if n != 0 {
fields = append(fields, zap.Int("n", n))
}
if len(size) != 0 {
fields = append(fields, zap.String("size", size))
}
if len(responseFormat) != 0 {
fields = append(fields, zap.String("response_format", responseFormat))
}
if len(user) != 0 {
fields = append(fields, zap.String("user", user))
}
log.Info("openai edit image request", fields...)
}
}
func logImageVariationsRequest(log *zap.Logger, model string, n int, size, responseFormat, user string, prod bool, cid string) {
if prod {
fields := []zapcore.Field{
zap.String(correlationId, cid),
}
if len(model) != 0 {
fields = append(fields, zap.String("model", model))
}
if n != 0 {
fields = append(fields, zap.Int("n", n))
}
if len(size) != 0 {
fields = append(fields, zap.String("size", size))
}
if len(responseFormat) != 0 {
fields = append(fields, zap.String("response_format", user))
}
if len(user) != 0 {
fields = append(fields, zap.String("user", user))
}
log.Info("openai image variations request", fields...)
}
}
func logImageResponse(log *zap.Logger, data []byte, prod, private bool, cid string) {
ir := &goopenai.ImageResponse{}
err := json.Unmarshal(data, ir)
if err != nil {
logError(log, "error when unmarshalling image response", prod, cid, err)
return
}
if prod {
fields := []zapcore.Field{
zap.String(correlationId, cid),
zap.Int64("created", ir.Created),
}
if !private && len(ir.Data) != 0 {
fields = append(fields, zap.Any("data", ir.Data))
}
log.Info("openai image response", fields...)
}
}