internal

type ChatCompletionMessageRole string

type ChatCompletionMessageToolCall = marshalx.Dynamic[struct {
	// The function that the model called.
	Function marshalx.Dynamic[struct {
		// The arguments to call the function with, as generated by the model
		// in JSON format. Note that the model does not always generate valid JSON,
		// and may hallucinate parameters not defined by your function schema.
		// Validate the arguments in your code before calling your function.
		Arguments string `json:"arguments"`

		// The name of the function to call.
		Name string `json:"name"`
	}] `json:"function"`

	// The ID of the tool call.
	Id string `json:"id"`

	// The type of the tool. Currently, only `function` is supported.
	Type ChatCompletionMessageToolCallType `json:"type"`
}]

type ChatCompletionMessageToolCallType string

type ChatCompletionMessageToolCalls = []ChatCompletionMessageToolCall

type ChatCompletionRequestMessage marshalx.Dynamic[struct {
	// The contents of the message.
	Content string `json:"content"`

	// An optional name for the participant. Provides the model information to differentiate between participants of the same role.
	Name *string `json:"name,omitempty"`

	// The role of the messages author.
	Role ChatCompletionMessageRole `json:"role"`
}]

type ChatCompletionResponseMessage = marshalx.Dynamic[struct {
	// The tool calls generated by the model, such as function calls.
	ToolCalls *ChatCompletionMessageToolCalls `json:"tool_calls,omitempty"`
}]

type ChatCompletionTool struct {
	Function FunctionObject `json:"function"`

	// Type The type of the tool. Currently, only `function` is supported.
	Type ChatCompletionToolType `json:"type"`
}

type ChatCompletionToolType string

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
}

type ClientOption func(*Client) error

type CreateChatCompletionRequest = marshalx.Dynamic[struct {
	// ID of the model to use.
	Model string `json:"model"`

	// A list of messages comprising the conversation so far.
	Messages []ChatCompletionRequestMessage `json:"messages"`

	// A list of tools the model may call. Currently, only functions are
	// supported as a tool. Use this to provide a list of functions the model may
	// generate JSON inputs for. A max of 128 functions are supported.
	Tools *[]ChatCompletionTool `json:"tools,omitempty"`
}]

type CreateChatCompletionResponse = marshalx.Dynamic[struct {
	// A list of chat completion choices. Can be more than one if `n` is greater than 1.
	Choices []marshalx.Dynamic[struct {
		// A chat completion message generated by the model.
		Message ChatCompletionResponseMessage `json:"message"`
	}] `json:"choices"`
}]

type CreateEmbeddingRequest = marshalx.Dynamic[struct {
	// ID of the model to use.
	Model string `json:"model"`
}]

type CreateEmbeddingResponse = marshalx.Dynamic[struct{}]

type FunctionObject marshalx.Dynamic[struct {
	// A description of what the function does, used by the model to choose when and how to call the function.
	Description *string `json:"description,omitempty"`

	// Name The name of the function to be called. Must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length of 64.
	Name string `json:"name"`

	// Parameters The parameters the functions accepts, described as a JSON Schema object.
	//
	// Omitting `parameters` defines a function with an empty parameter list.
	Parameters *FunctionParameters `json:"parameters,omitempty"`

	// Strict Whether to enable strict schema adherence when generating the function call.
	// If set to true, the model will follow the exact schema defined in the `parameters`
	// field. Only a subset of JSON Schema is supported when `strict` is `true`.
	Strict *bool `json:"strict"`
}]

type FunctionParameters map[string]any

type HttpRequestDoer interface {
	Do(req *http.Request) (*http.Response, error)
}

type RequestEditorFn func(ctx context.Context, req *http.Request) error

type ServerInterface interface {
	CreateChatCompletion(w http.ResponseWriter, r *http.Request)
	CreateEmbedding(w http.ResponseWriter, r *http.Request)
}

type ServerInterfaceWrapper struct {
	Handler ServerInterface
	// HandlerMiddlewares []MiddlewareFunc
	ErrorHandlerFunc func(w http.ResponseWriter, r *http.Request, err error)
}

type StdHTTPServerOptions struct {
	BaseURL          string
	BaseRouter       *http.ServeMux
	ErrorHandlerFunc func(w http.ResponseWriter, r *http.Request, err error)
}