|
|
@@ -1,388 +0,0 @@
|
|
|
-// Package client provides primitives to interact with the openapi HTTP API.
|
|
|
-//
|
|
|
-// Code generated by github.com/oapi-codegen/oapi-codegen/v2 version v2.4.1 DO NOT EDIT.
|
|
|
-package client
|
|
|
-
|
|
|
-import (
|
|
|
- "bytes"
|
|
|
- "context"
|
|
|
- "encoding/json"
|
|
|
- "fmt"
|
|
|
- "io"
|
|
|
- "net/http"
|
|
|
- "net/url"
|
|
|
- "strings"
|
|
|
-)
|
|
|
-
|
|
|
-// PostSessionChatJSONBody defines parameters for PostSessionChat.
|
|
|
-type PostSessionChatJSONBody struct {
|
|
|
- Parts *interface{} `json:"parts,omitempty"`
|
|
|
- SessionID string `json:"sessionID"`
|
|
|
-}
|
|
|
-
|
|
|
-// PostSessionChatJSONRequestBody defines body for PostSessionChat for application/json ContentType.
|
|
|
-type PostSessionChatJSONRequestBody PostSessionChatJSONBody
|
|
|
-
|
|
|
-// RequestEditorFn is the function signature for the RequestEditor callback function
|
|
|
-type RequestEditorFn func(ctx context.Context, req *http.Request) error
|
|
|
-
|
|
|
-// Doer performs HTTP requests.
|
|
|
-//
|
|
|
-// The standard http.Client implements this interface.
|
|
|
-type HttpRequestDoer interface {
|
|
|
- Do(req *http.Request) (*http.Response, error)
|
|
|
-}
|
|
|
-
|
|
|
-// Client which conforms to the OpenAPI3 specification for this service.
|
|
|
-type Client struct {
|
|
|
- // The endpoint of the server conforming to this interface, with scheme,
|
|
|
- // https://api.deepmap.com for example. This can contain a path relative
|
|
|
- // to the server, such as https://api.deepmap.com/dev-test, and all the
|
|
|
- // paths in the swagger spec will be appended to the server.
|
|
|
- Server string
|
|
|
-
|
|
|
- // Doer for performing requests, typically a *http.Client with any
|
|
|
- // customized settings, such as certificate chains.
|
|
|
- Client HttpRequestDoer
|
|
|
-
|
|
|
- // A list of callbacks for modifying requests which are generated before sending over
|
|
|
- // the network.
|
|
|
- RequestEditors []RequestEditorFn
|
|
|
-}
|
|
|
-
|
|
|
-// ClientOption allows setting custom parameters during construction
|
|
|
-type ClientOption func(*Client) error
|
|
|
-
|
|
|
-// Creates a new Client, with reasonable defaults
|
|
|
-func NewClient(server string, opts ...ClientOption) (*Client, error) {
|
|
|
- // create a client with sane default values
|
|
|
- client := Client{
|
|
|
- Server: server,
|
|
|
- }
|
|
|
- // mutate client and add all optional params
|
|
|
- for _, o := range opts {
|
|
|
- if err := o(&client); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- }
|
|
|
- // ensure the server URL always has a trailing slash
|
|
|
- if !strings.HasSuffix(client.Server, "/") {
|
|
|
- client.Server += "/"
|
|
|
- }
|
|
|
- // create httpClient, if not already present
|
|
|
- if client.Client == nil {
|
|
|
- client.Client = &http.Client{}
|
|
|
- }
|
|
|
- return &client, nil
|
|
|
-}
|
|
|
-
|
|
|
-// WithHTTPClient allows overriding the default Doer, which is
|
|
|
-// automatically created using http.Client. This is useful for tests.
|
|
|
-func WithHTTPClient(doer HttpRequestDoer) ClientOption {
|
|
|
- return func(c *Client) error {
|
|
|
- c.Client = doer
|
|
|
- return nil
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-// WithRequestEditorFn allows setting up a callback function, which will be
|
|
|
-// called right before sending the request. This can be used to mutate the request.
|
|
|
-func WithRequestEditorFn(fn RequestEditorFn) ClientOption {
|
|
|
- return func(c *Client) error {
|
|
|
- c.RequestEditors = append(c.RequestEditors, fn)
|
|
|
- return nil
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-// The interface specification for the client above.
|
|
|
-type ClientInterface interface {
|
|
|
- // PostSessionChatWithBody request with any body
|
|
|
- PostSessionChatWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)
|
|
|
-
|
|
|
- PostSessionChat(ctx context.Context, body PostSessionChatJSONRequestBody, reqEditors ...RequestEditorFn) (*http.Response, error)
|
|
|
-
|
|
|
- // PostSessionCreate request
|
|
|
- PostSessionCreate(ctx context.Context, reqEditors ...RequestEditorFn) (*http.Response, error)
|
|
|
-}
|
|
|
-
|
|
|
-func (c *Client) PostSessionChatWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error) {
|
|
|
- req, err := NewPostSessionChatRequestWithBody(c.Server, contentType, body)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- req = req.WithContext(ctx)
|
|
|
- if err := c.applyEditors(ctx, req, reqEditors); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return c.Client.Do(req)
|
|
|
-}
|
|
|
-
|
|
|
-func (c *Client) PostSessionChat(ctx context.Context, body PostSessionChatJSONRequestBody, reqEditors ...RequestEditorFn) (*http.Response, error) {
|
|
|
- req, err := NewPostSessionChatRequest(c.Server, body)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- req = req.WithContext(ctx)
|
|
|
- if err := c.applyEditors(ctx, req, reqEditors); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return c.Client.Do(req)
|
|
|
-}
|
|
|
-
|
|
|
-func (c *Client) PostSessionCreate(ctx context.Context, reqEditors ...RequestEditorFn) (*http.Response, error) {
|
|
|
- req, err := NewPostSessionCreateRequest(c.Server)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- req = req.WithContext(ctx)
|
|
|
- if err := c.applyEditors(ctx, req, reqEditors); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return c.Client.Do(req)
|
|
|
-}
|
|
|
-
|
|
|
-// NewPostSessionChatRequest calls the generic PostSessionChat builder with application/json body
|
|
|
-func NewPostSessionChatRequest(server string, body PostSessionChatJSONRequestBody) (*http.Request, error) {
|
|
|
- var bodyReader io.Reader
|
|
|
- buf, err := json.Marshal(body)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- bodyReader = bytes.NewReader(buf)
|
|
|
- return NewPostSessionChatRequestWithBody(server, "application/json", bodyReader)
|
|
|
-}
|
|
|
-
|
|
|
-// NewPostSessionChatRequestWithBody generates requests for PostSessionChat with any type of body
|
|
|
-func NewPostSessionChatRequestWithBody(server string, contentType string, body io.Reader) (*http.Request, error) {
|
|
|
- var err error
|
|
|
-
|
|
|
- serverURL, err := url.Parse(server)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- operationPath := fmt.Sprintf("/session_chat")
|
|
|
- if operationPath[0] == '/' {
|
|
|
- operationPath = "." + operationPath
|
|
|
- }
|
|
|
-
|
|
|
- queryURL, err := serverURL.Parse(operationPath)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- req, err := http.NewRequest("POST", queryURL.String(), body)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- req.Header.Add("Content-Type", contentType)
|
|
|
-
|
|
|
- return req, nil
|
|
|
-}
|
|
|
-
|
|
|
-// NewPostSessionCreateRequest generates requests for PostSessionCreate
|
|
|
-func NewPostSessionCreateRequest(server string) (*http.Request, error) {
|
|
|
- var err error
|
|
|
-
|
|
|
- serverURL, err := url.Parse(server)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- operationPath := fmt.Sprintf("/session_create")
|
|
|
- if operationPath[0] == '/' {
|
|
|
- operationPath = "." + operationPath
|
|
|
- }
|
|
|
-
|
|
|
- queryURL, err := serverURL.Parse(operationPath)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- req, err := http.NewRequest("POST", queryURL.String(), nil)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- return req, nil
|
|
|
-}
|
|
|
-
|
|
|
-func (c *Client) applyEditors(ctx context.Context, req *http.Request, additionalEditors []RequestEditorFn) error {
|
|
|
- for _, r := range c.RequestEditors {
|
|
|
- if err := r(ctx, req); err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- }
|
|
|
- for _, r := range additionalEditors {
|
|
|
- if err := r(ctx, req); err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- }
|
|
|
- return nil
|
|
|
-}
|
|
|
-
|
|
|
-// ClientWithResponses builds on ClientInterface to offer response payloads
|
|
|
-type ClientWithResponses struct {
|
|
|
- ClientInterface
|
|
|
-}
|
|
|
-
|
|
|
-// NewClientWithResponses creates a new ClientWithResponses, which wraps
|
|
|
-// Client with return type handling
|
|
|
-func NewClientWithResponses(server string, opts ...ClientOption) (*ClientWithResponses, error) {
|
|
|
- client, err := NewClient(server, opts...)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return &ClientWithResponses{client}, nil
|
|
|
-}
|
|
|
-
|
|
|
-// WithBaseURL overrides the baseURL.
|
|
|
-func WithBaseURL(baseURL string) ClientOption {
|
|
|
- return func(c *Client) error {
|
|
|
- newBaseURL, err := url.Parse(baseURL)
|
|
|
- if err != nil {
|
|
|
- return err
|
|
|
- }
|
|
|
- c.Server = newBaseURL.String()
|
|
|
- return nil
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-// ClientWithResponsesInterface is the interface specification for the client with responses above.
|
|
|
-type ClientWithResponsesInterface interface {
|
|
|
- // PostSessionChatWithBodyWithResponse request with any body
|
|
|
- PostSessionChatWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*PostSessionChatResponse, error)
|
|
|
-
|
|
|
- PostSessionChatWithResponse(ctx context.Context, body PostSessionChatJSONRequestBody, reqEditors ...RequestEditorFn) (*PostSessionChatResponse, error)
|
|
|
-
|
|
|
- // PostSessionCreateWithResponse request
|
|
|
- PostSessionCreateWithResponse(ctx context.Context, reqEditors ...RequestEditorFn) (*PostSessionCreateResponse, error)
|
|
|
-}
|
|
|
-
|
|
|
-type PostSessionChatResponse struct {
|
|
|
- Body []byte
|
|
|
- HTTPResponse *http.Response
|
|
|
-}
|
|
|
-
|
|
|
-// Status returns HTTPResponse.Status
|
|
|
-func (r PostSessionChatResponse) Status() string {
|
|
|
- if r.HTTPResponse != nil {
|
|
|
- return r.HTTPResponse.Status
|
|
|
- }
|
|
|
- return http.StatusText(0)
|
|
|
-}
|
|
|
-
|
|
|
-// StatusCode returns HTTPResponse.StatusCode
|
|
|
-func (r PostSessionChatResponse) StatusCode() int {
|
|
|
- if r.HTTPResponse != nil {
|
|
|
- return r.HTTPResponse.StatusCode
|
|
|
- }
|
|
|
- return 0
|
|
|
-}
|
|
|
-
|
|
|
-type PostSessionCreateResponse struct {
|
|
|
- Body []byte
|
|
|
- HTTPResponse *http.Response
|
|
|
- JSON200 *struct {
|
|
|
- Id string `json:"id"`
|
|
|
- Title string `json:"title"`
|
|
|
- Tokens struct {
|
|
|
- Input float32 `json:"input"`
|
|
|
- Output float32 `json:"output"`
|
|
|
- Reasoning float32 `json:"reasoning"`
|
|
|
- } `json:"tokens"`
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-// Status returns HTTPResponse.Status
|
|
|
-func (r PostSessionCreateResponse) Status() string {
|
|
|
- if r.HTTPResponse != nil {
|
|
|
- return r.HTTPResponse.Status
|
|
|
- }
|
|
|
- return http.StatusText(0)
|
|
|
-}
|
|
|
-
|
|
|
-// StatusCode returns HTTPResponse.StatusCode
|
|
|
-func (r PostSessionCreateResponse) StatusCode() int {
|
|
|
- if r.HTTPResponse != nil {
|
|
|
- return r.HTTPResponse.StatusCode
|
|
|
- }
|
|
|
- return 0
|
|
|
-}
|
|
|
-
|
|
|
-// PostSessionChatWithBodyWithResponse request with arbitrary body returning *PostSessionChatResponse
|
|
|
-func (c *ClientWithResponses) PostSessionChatWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*PostSessionChatResponse, error) {
|
|
|
- rsp, err := c.PostSessionChatWithBody(ctx, contentType, body, reqEditors...)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return ParsePostSessionChatResponse(rsp)
|
|
|
-}
|
|
|
-
|
|
|
-func (c *ClientWithResponses) PostSessionChatWithResponse(ctx context.Context, body PostSessionChatJSONRequestBody, reqEditors ...RequestEditorFn) (*PostSessionChatResponse, error) {
|
|
|
- rsp, err := c.PostSessionChat(ctx, body, reqEditors...)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return ParsePostSessionChatResponse(rsp)
|
|
|
-}
|
|
|
-
|
|
|
-// PostSessionCreateWithResponse request returning *PostSessionCreateResponse
|
|
|
-func (c *ClientWithResponses) PostSessionCreateWithResponse(ctx context.Context, reqEditors ...RequestEditorFn) (*PostSessionCreateResponse, error) {
|
|
|
- rsp, err := c.PostSessionCreate(ctx, reqEditors...)
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- return ParsePostSessionCreateResponse(rsp)
|
|
|
-}
|
|
|
-
|
|
|
-// ParsePostSessionChatResponse parses an HTTP response from a PostSessionChatWithResponse call
|
|
|
-func ParsePostSessionChatResponse(rsp *http.Response) (*PostSessionChatResponse, error) {
|
|
|
- bodyBytes, err := io.ReadAll(rsp.Body)
|
|
|
- defer func() { _ = rsp.Body.Close() }()
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- response := &PostSessionChatResponse{
|
|
|
- Body: bodyBytes,
|
|
|
- HTTPResponse: rsp,
|
|
|
- }
|
|
|
-
|
|
|
- return response, nil
|
|
|
-}
|
|
|
-
|
|
|
-// ParsePostSessionCreateResponse parses an HTTP response from a PostSessionCreateWithResponse call
|
|
|
-func ParsePostSessionCreateResponse(rsp *http.Response) (*PostSessionCreateResponse, error) {
|
|
|
- bodyBytes, err := io.ReadAll(rsp.Body)
|
|
|
- defer func() { _ = rsp.Body.Close() }()
|
|
|
- if err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
- response := &PostSessionCreateResponse{
|
|
|
- Body: bodyBytes,
|
|
|
- HTTPResponse: rsp,
|
|
|
- }
|
|
|
-
|
|
|
- switch {
|
|
|
- case strings.Contains(rsp.Header.Get("Content-Type"), "json") && rsp.StatusCode == 200:
|
|
|
- var dest struct {
|
|
|
- Id string `json:"id"`
|
|
|
- Title string `json:"title"`
|
|
|
- Tokens struct {
|
|
|
- Input float32 `json:"input"`
|
|
|
- Output float32 `json:"output"`
|
|
|
- Reasoning float32 `json:"reasoning"`
|
|
|
- } `json:"tokens"`
|
|
|
- }
|
|
|
- if err := json.Unmarshal(bodyBytes, &dest); err != nil {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
- response.JSON200 = &dest
|
|
|
-
|
|
|
- }
|
|
|
-
|
|
|
- return response, nil
|
|
|
-}
|
Dax Raad