mirror of
https://github.com/CJackHwang/ds2api.git
synced 2026-05-16 22:25:15 +08:00
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package openai
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"ds2api/internal/auth"
|
|
"ds2api/internal/config"
|
|
"ds2api/internal/deepseek"
|
|
)
|
|
|
|
type AuthResolver interface {
|
|
Determine(req *http.Request) (*auth.RequestAuth, error)
|
|
DetermineCaller(req *http.Request) (*auth.RequestAuth, error)
|
|
Release(a *auth.RequestAuth)
|
|
}
|
|
|
|
type DeepSeekCaller interface {
|
|
CreateSession(ctx context.Context, a *auth.RequestAuth, maxAttempts int) (string, error)
|
|
GetPow(ctx context.Context, a *auth.RequestAuth, maxAttempts int) (string, error)
|
|
CallCompletion(ctx context.Context, a *auth.RequestAuth, payload map[string]any, powResp string, maxAttempts int) (*http.Response, error)
|
|
DeleteAllSessionsForToken(ctx context.Context, token string) (int, error)
|
|
}
|
|
|
|
type ConfigReader interface {
|
|
ModelAliases() map[string]string
|
|
CompatWideInputStrictOutput() bool
|
|
ToolcallMode() string
|
|
ToolcallEarlyEmitConfidence() string
|
|
ResponsesStoreTTLSeconds() int
|
|
EmbeddingsProvider() string
|
|
AutoDeleteSessions() bool
|
|
}
|
|
|
|
var _ AuthResolver = (*auth.Resolver)(nil)
|
|
var _ DeepSeekCaller = (*deepseek.Client)(nil)
|
|
var _ ConfigReader = (*config.Store)(nil)
|