localgateway

type ChatChoice struct {
	Index        int         `json:"index"`
	Message      ChatMessage `json:"message"`
	FinishReason string      `json:"finish_reason"`
}

type ChatMessage struct {
	Role       string     `json:"role"`
	Content    any        `json:"content,omitempty"`
	Name       string     `json:"name,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
}

type ChatRequest struct {
	Model    string        `json:"model"`
	Messages []ChatMessage `json:"messages"`
	Tools    []ToolDef     `json:"tools,omitempty"`
	Stream   bool          `json:"stream,omitempty"`
	// Temperature and other sampling params are forwarded as-is via RawExtra
	// is not used — we re-encode a controlled subset.
	Temperature *float64 `json:"temperature,omitempty"`
	MaxTokens   *int     `json:"max_tokens,omitempty"`
	TopP        *float64 `json:"top_p,omitempty"`
}

type ChatResponse struct {
	ID      string       `json:"id"`
	Object  string       `json:"object"`
	Created int64        `json:"created"`
	Model   string       `json:"model"`
	Choices []ChatChoice `json:"choices"`
	Usage   *Usage       `json:"usage,omitempty"`
}

type ChatUpstream interface {
	ChatCompletions(ctx context.Context, req ChatRequest) (*ChatResponse, error)
	ListModels(ctx context.Context) (*ModelsResponse, error)
}

type Config struct {
	// ListenAddr is the HTTP bind address (e.g. ":8091").
	ListenAddr string
	// AuthToken is the required Bearer token for /v1/* (required, min 16 chars).
	AuthToken string
	// UpstreamURL is the OpenAI-compatible base URL (e.g. http://127.0.0.1:11434).
	// Only operator-configured; never taken from request bodies.
	UpstreamURL string
	// UpstreamTimeout bounds a single upstream chat request.
	UpstreamTimeout time.Duration
	// VaultRunBaseURL and VaultRunAPIKey authenticate to the VaultRun API.
	VaultRunBaseURL string
	VaultRunAPIKey  string
	// DefaultImage is used when auto-creating sessions.
	DefaultImage string
	// NetworkEnabled controls whether auto-created sessions get network.
	NetworkEnabled bool
	// MaxToolLoops caps model→tool→model iterations per request.
	MaxToolLoops int
	// MaxBodyBytes limits inbound JSON body size.
	MaxBodyBytes int64
	// MaxToolResultBytes truncates tool stdout/stderr fed back to the model.
	MaxToolResultBytes int
	// MaxFileBytes caps write_file content size.
	MaxFileBytes int
	// RateLimitPerMin is the per-IP request budget (fixed 1-minute window).
	RateLimitPerMin int
	// DefaultModel is injected when the client omits "model".
	DefaultModel string
	// RunTimeoutSeconds default for vaultrun_run_command when the model omits it.
	RunTimeoutSeconds int
	// MaxRunTimeoutSeconds hard cap for per-run timeouts.
	MaxRunTimeoutSeconds int
	// CaptureMissions saves successful VaultRun tool sequences as missions.
	CaptureMissions bool
}

type ErrorBody struct {
	Error ErrorDetail `json:"error"`
}

type ErrorDetail struct {
	Message string `json:"message"`
	Type    string `json:"type"`
	Code    string `json:"code,omitempty"`
}

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type Gateway struct {
	// contains filtered or unexported fields
}

type GatewayError struct {
	Status  int
	Code    string
	Message string
}

type HTTPUpstream struct {
	// contains filtered or unexported fields
}

type LoopResult struct {
	Response  *ChatResponse
	SessionID string
	Stream    bool
}

type MissionAPI interface {
	CreateMission(ctx context.Context, req createMissionRequest) (string, error)
	RecordMissionRun(ctx context.Context, missionID, sessionID string) error
}

type ModelObject struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	OwnedBy string `json:"owned_by"`
}

type ModelsResponse struct {
	Object string        `json:"object"`
	Data   []ModelObject `json:"data"`
}

type SessionStore struct {
	// contains filtered or unexported fields
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function FunctionCall `json:"function"`
}

type ToolDef struct {
	Type     string       `json:"type"`
	Function ToolFunction `json:"function"`
}

type ToolFunction struct {
	Name        string          `json:"name"`
	Description string          `json:"description,omitempty"`
	Parameters  json.RawMessage `json:"parameters,omitempty"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type VaultRunClient interface {
	CreateSession(ctx context.Context, opts vaultrun.CreateSessionOptions) (*vaultrun.Session, error)
	GetSession(ctx context.Context, sessionID string) (*vaultrun.Session, error)
	DeleteSession(ctx context.Context, sessionID string) error
	Run(ctx context.Context, sessionID string, opts vaultrun.RunOptions) (*vaultrun.Run, error)
	UploadFile(ctx context.Context, sessionID, remotePath string, content io.Reader) (*vaultrun.File, error)
	DownloadFile(ctx context.Context, sessionID, remotePath string) (io.ReadCloser, error)
	ListFiles(ctx context.Context, sessionID string) ([]*vaultrun.File, error)
}