(log *zap.Logger, prod, private bool, r *EmbeddingResponseBase64)
| 1091 | } |
| 1092 | |
| 1093 | func logBase64EmbeddingResponse(log *zap.Logger, prod, private bool, r *EmbeddingResponseBase64) { |
| 1094 | if prod { |
| 1095 | log.Info("openai embeddings response", |
| 1096 | zap.Time("createdAt", time.Now()), |
| 1097 | zap.Object("response", zapcore.ObjectMarshalerFunc( |
| 1098 | func(enc zapcore.ObjectEncoder) error { |
| 1099 | enc.AddString("object", r.Object) |
| 1100 | enc.AddString("model", r.Model) |
| 1101 | enc.AddArray("data", zapcore.ArrayMarshalerFunc( |
| 1102 | func(enc zapcore.ArrayEncoder) error { |
| 1103 | for _, d := range r.Data { |
| 1104 | enc.AppendObject(zapcore.ObjectMarshalerFunc( |
| 1105 | func(enc zapcore.ObjectEncoder) error { |
| 1106 | enc.AddInt("index", d.Index) |
| 1107 | enc.AddString("object", d.Object) |
| 1108 | if !private { |
| 1109 | enc.AddString("embedding", string(d.Embedding)) |
| 1110 | } |
| 1111 | |
| 1112 | return nil |
| 1113 | }, |
| 1114 | )) |
| 1115 | } |
| 1116 | return nil |
| 1117 | }, |
| 1118 | )) |
| 1119 | |
| 1120 | enc.AddObject("usage", zapcore.ObjectMarshalerFunc( |
| 1121 | func(enc zapcore.ObjectEncoder) error { |
| 1122 | enc.AddInt("prompt_tokens", r.Usage.PromptTokens) |
| 1123 | enc.AddInt("completion_tokens", r.Usage.CompletionTokens) |
| 1124 | enc.AddInt("total_tokens", r.Usage.TotalTokens) |
| 1125 | return nil |
| 1126 | }, |
| 1127 | )) |
| 1128 | return nil |
| 1129 | }, |
| 1130 | )), |
| 1131 | ) |
| 1132 | } |
| 1133 | } |
| 1134 | |
| 1135 | func logChatCompletionResponse(log *zap.Logger, prod, private bool, r *goopenai.ChatCompletionResponse) { |
| 1136 | if prod { |
no test coverage detected