From 2ef631b425070fd3d9a0581ee9f77f563edebe20 Mon Sep 17 00:00:00 2001 From: Xe Iaso Date: Sat, 4 Feb 2023 18:36:39 -0500 Subject: mastodon/robocadey2: contact logos over tsnet Signed-off-by: Xe Iaso --- internal/stablediffusion/stablediffusion.go | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) (limited to 'internal') diff --git a/internal/stablediffusion/stablediffusion.go b/internal/stablediffusion/stablediffusion.go index 218bf51..3ef2bae 100644 --- a/internal/stablediffusion/stablediffusion.go +++ b/internal/stablediffusion/stablediffusion.go @@ -83,7 +83,21 @@ type ImageInfo struct { IsUsingInpaintingConditioning bool `json:"is_using_inpainting_conditioning"` } +var ( + Default *Client = &Client{ + HTTP: http.DefaultClient, + } +) + func Generate(ctx context.Context, inp SimpleImageRequest) (*ImageResponse, error) { + return Default.Generate(ctx, inp) +} + +type Client struct { + HTTP *http.Client +} + +func (c *Client) Generate(ctx context.Context, inp SimpleImageRequest) (*ImageResponse, error) { u, err := buildURL("/sdapi/v1/txt2img") if err != nil { return nil, fmt.Errorf("error building URL: %w", err) @@ -99,7 +113,7 @@ func Generate(ctx context.Context, inp SimpleImageRequest) (*ImageResponse, erro return nil, fmt.Errorf("error making request: %w", err) } - resp, err := http.DefaultClient.Do(req) + resp, err := c.HTTP.Do(req) if err != nil { return nil, fmt.Errorf("error fetching response: %w", err) } -- cgit v1.2.3