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
|
package llm
import (
"bytes"
"context"
"errors"
"net/http"
"strings"
"codeberg.org/snonux/hexai/internal/logging"
)
// small helper to keep return type consistent
func nilStringErr(msg string) (string, error) { return "", errors.New(msg) }
// doJSONRequest is the shared entry point for provider HTTP calls. It applies
// the default retry-with-backoff policy and the shared circuit breaker so that
// transient upstream failures (network resets, 429, 5xx) are smoothed over,
// while client errors (4xx) and successes are returned immediately. The body
// bytes are passed by value so each retry can rebuild a fresh request.
func doJSONRequest(ctx context.Context, httpClient *http.Client, url string, body []byte, headers map[string]string, accept string) (*http.Response, error) {
return doJSONRequestResilient(ctx, httpClient, url, body, headers, accept, defaultRetryPolicy(), sharedBreaker)
}
// doJSONRequestOnce performs exactly one HTTP POST with the given JSON body and
// headers. It is the single-attempt primitive used by the resilience layer.
func doJSONRequestOnce(ctx context.Context, httpClient *http.Client, url string, body []byte, headers map[string]string, accept string) (*http.Response, error) {
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(body))
if err != nil {
return nil, err
}
req.Header.Set("Content-Type", "application/json")
if strings.TrimSpace(accept) != "" {
req.Header.Set("Accept", accept)
}
for k, v := range headers {
req.Header.Set(k, v)
}
return httpClient.Do(req)
}
func logStartMessages(chatLogger logging.ChatLogger, stream bool, o Options, messages []Message) {
logMessages := make([]struct{ Role, Content string }, len(messages))
for i, m := range messages {
logMessages[i] = struct{ Role, Content string }{m.Role, m.Content}
}
chatLogger.LogStart(stream, o.Model, o.Temperature, o.MaxTokens, o.Stop, logMessages)
}
|
