feat(tasks): add templates findings workflows and batch controls
This commit is contained in:
1 parent
b7e0b93023
commit
d3183dbbd2
65 files changed
+11603
-2198
No files matched your search
+20
-2
@@ -122,7 +122,7 @@ func TestTaskLLMStateSnapshotIsConcurrentAndDetached(t *testing.T) {
|
||||
defer wg.Done()
|
||||
for i := 0; i < 500; i++ {
|
||||
id := int64(offset*1000 + i + 1)
|
||||
task.setLLMState(&id, &id, []int64{id, id + 1}, "ready", "")
|
||||
task.setLLMState(&id, &id, []int64{id, id + 1}, int64(i), "ready", "")
|
||||
}
|
||||
}(writer)
|
||||
}
|
||||
@@ -146,6 +146,24 @@ func TestTaskLLMStateSnapshotIsConcurrentAndDetached(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestTaskLLMStateRejectsOlderRevision(t *testing.T) {
|
||||
t.Parallel()
|
||||
task := &Task{ID: "7"}
|
||||
current := int64(22)
|
||||
stale := int64(11)
|
||||
if !task.setLLMState(¤t, ¤t, []int64{current}, 2, "ready", "") {
|
||||
t.Fatal("initial LLM state was rejected")
|
||||
}
|
||||
if task.setLLMState(&stale, &stale, []int64{stale}, 1, "ready", "stale") {
|
||||
t.Fatal("older LLM snapshot was accepted")
|
||||
}
|
||||
state := task.llmStateSnapshot()
|
||||
if state.ChainRevision != 2 || state.ActiveID == nil || *state.ActiveID != current ||
|
||||
len(state.ProfileIDs) != 1 || state.ProfileIDs[0] != current || state.FailoverReason != "" {
|
||||
t.Fatalf("older LLM snapshot overwrote current state: %+v", state)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTaskLLMStreamRetriesPreStreamQuotaOnNextProfile(t *testing.T) {
|
||||
t.Parallel()
|
||||
quota := errors.New("openai: status 402: insufficient_quota")
|
||||
@@ -381,7 +399,7 @@ func TestResolvedTaskStatusKeepsExhaustedStartedTaskRunning(t *testing.T) {
|
||||
engine.started.Store("7", true)
|
||||
task := &Task{ID: "7", Status: "running"}
|
||||
profileID := int64(11)
|
||||
task.setLLMState(&profileID, nil, []int64{profileID}, "chain_exhausted", "quota exhausted")
|
||||
task.setLLMState(&profileID, nil, []int64{profileID}, 1, "chain_exhausted", "quota exhausted")
|
||||
s := &Server{engine: engine}
|
||||
if got := s.resolvedTaskStatus(task); got != "running" {
|
||||
t.Fatalf("exhausted started status=%q, want running", got)
|
||||
|
||||
Reference in new issue
Block a user