postgres

Documentation

Index

• type Client
  • func NewPostgresClient(ctx context.Context, cfg Config) (*Client, error)
  • func (c Client) Close(ctx context.Context) error
  • func (c Client) GetActiveUserIDByActiveTokenID(ctx context.Context, id string) (string, error)
  • func (c Client) GetDailySuccessfulResultsTimestampsByUserID(ctx context.Context, userID string) ([]time.Time, error)
  • func (c Client) WriteInputPrompt(ctx context.Context, requestID, userID, prompt string) error
  • func (c Client) WriteModelResult(ctx context.Context, ...) error
  • func (c Client) WriteSuccessFlag(ctx context.Context, requestID, userID, token string) error
• type Config
  • func (cfg Config) ConnectionString() string
  • func (cfg Config) Validate() error

Types

type Client

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

func NewPostgresClient

func NewPostgresClient(ctx context.Context, cfg Config) (
	*Client, error,
)

NewPostgresClient initiates the postgres Client.

func (Client) Close

func (c Client) Close(ctx context.Context) error

func (Client) GetActiveUserIDByActiveTokenID

func (c Client) GetActiveUserIDByActiveTokenID(ctx context.Context, id string) (string, error)

func (Client) GetDailySuccessfulResultsTimestampsByUserID

func (c Client) GetDailySuccessfulResultsTimestampsByUserID(ctx context.Context, userID string) ([]time.Time, error)

func (Client) WriteInputPrompt

func (c Client) WriteInputPrompt(ctx context.Context, requestID, userID, prompt string) error

func (Client) WriteModelResult

func (c Client) WriteModelResult(
	ctx context.Context, requestID, userID, predictionRaw, prediction, model string,
	usageTokensPrompt, usageTokensCompletions uint16,
) error

func (Client) WriteSuccessFlag

func (c Client) WriteSuccessFlag(ctx context.Context, requestID, userID, token string) error

type Config

type Config struct {
	DBHost             string `json:"db_host"`
	DBName             string `json:"db_name"`
	DBUser             string `json:"db_user"`
	DBPassword         string `json:"db_password"`
	TablePrompt        string `json:"table_prompt,omitempty"`
	TablePrediction    string `json:"table_prediction,omitempty"`
	TableSuccessStatus string `json:"table_success_status,omitempty"`
	TableUsers         string `json:"table_users,omitempty"`
	TableTokens        string `json:"table_tokens,omitempty"`
	SSLMode            string `json:"ssl_mode"`
}

Config configuration of the postgres Client.

func (Config) ConnectionString

func (cfg Config) ConnectionString() string

func (Config) Validate

func (cfg Config) Validate() error