OPC/chorus
2
0
Fork 0
Code Issues 11 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 实现安全多 Provider worker (#22)

Browse Source
This commit is contained in:
ila
2026-08-21 21:00:25 +08:00
parent 9b2827a727
commit 35fd0342f0
22 changed files with 1498 additions and 375 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/core/queue/queue.go
+2
View File
@@ -15,6 +15,8 @@ type Claim struct {
type Repository interface {
CreateIdempotent(ctx context.Context, generation *model.Generation, inputs []model.GenerationInput) (created bool, err error)
ClaimNext(ctx context.Context, owner string, leaseDuration time.Duration) (*Claim, error)
BeginProviderAttempt(ctx context.Context, generationID uint64, leaseToken string, routeMemberID, providerModelID uint64) (model.Attempt, bool, error)
FinishProviderAttempt(ctx context.Context, generationID uint64, leaseToken string, attempt model.Attempt) (bool, error)
AssignProvider(ctx context.Context, generationID uint64, leaseToken string, providerModelID uint64) (bool, error)
Succeed(ctx context.Context, generationID uint64, leaseToken string, outputs []model.GenerationOutput, attempt model.Attempt) (bool, error)
Fail(ctx context.Context, generationID uint64, leaseToken string, code, message string, attempt model.Attempt) (bool, error)