Skip to content

Commit 0e2da82

Browse files
ETL Tasks States enum (#21)
* ETL Tasks States enum * Parse task date function * Enums * Strategy step type enum * Fix for null values * PR comments
1 parent 6b13b06 commit 0e2da82

File tree

2 files changed

+53
-0
lines changed

2 files changed

+53
-0
lines changed

date-parser.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,10 @@ export function formatTimeDigitalClock(s: number) {
3838
const mm = Math.floor(s / 60);
3939
const ss = s % 60;
4040
return `${String(mm).padStart(2, "0")}:${String(ss).padStart(2, "0")}`;
41+
}
42+
43+
export function parseTaskDate(dateStr: string): number {
44+
if (!dateStr) return null;
45+
const fixed = dateStr.replace(/(\.\d{3})\d+/, '$1');
46+
return new Date(fixed + 'Z').getTime();
4147
}

enums/enums.ts

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,3 +65,50 @@ export enum UserRole {
6565
EXPERT = "EXPERT",
6666
ANNOTATOR = "ANNOTATOR",
6767
}
68+
69+
export enum ETLTaskState {
70+
UPLOADING = 'UPLOADING',
71+
QUEUE = 'QUEUE',
72+
EXTRACTING = 'EXTRACTING',
73+
TOKENIZING = 'TOKENIZING',
74+
SPLITTING = 'SPLITTING',
75+
TRANSFORMING = 'TRANSFORMING',
76+
FINISHED = 'FINISHED'
77+
}
78+
79+
export enum IntegrationsTaskState {
80+
UPLOADED = "UPLOADED",
81+
AUTOMATICALLY_CREATED = "AUTOMATICALLY_CREATED",
82+
INITIAL = "INITIAL",
83+
RUNNING = "RUNNING",
84+
USABLE = "USABLE",
85+
FAILED = "FAILED",
86+
INITIALIZING = "INITIALIZING",
87+
WAITING = "WAITING",
88+
ENCODING = "ENCODING",
89+
FINISHED = "FINISHED",
90+
QUEUED = "QUEUED",
91+
}
92+
93+
export enum StrategyStepType {
94+
NONE = "NONE",
95+
PYTHON = "PYTHON",
96+
LLM = "LLM",
97+
SELECTION = "SELECTION",
98+
QUERY_REPHRASING = "QUERY_REPHRASING", // now more like a common llm, but changing the enum value would break the existing data
99+
// INFO: Websearch strategy deactivated until compliance investigation is finished
100+
// WEBSEARCH = "WEBSEARCH"
101+
TRUNCATE_CONTEXT = "TRUNCATE_CONTEXT",
102+
HEADER = "HEADER",
103+
TMP_DOC_RETRIEVAL = "TMP_DOC_RETRIEVAL",
104+
CALL_OTHER_AGENT = "CALL_OTHER_AGENT",
105+
NEURAL_SEARCH = "NEURAL_SEARCH",
106+
WEBHOOK = "WEBHOOK",
107+
GRAPHRAG_SEARCH = "GRAPHRAG_SEARCH",
108+
TEMPLATED = "TEMPLATED",
109+
RERANKER = "RERANKER",
110+
FULL_TEXT_SEARCH = "FULL_TEXT_SEARCH",
111+
CURRENT_TIME = "CURRENT_TIME",
112+
COMPLIANT_WEBSEARCH = "COMPLIANT_WEBSEARCH",
113+
114+
}

0 commit comments

Comments
 (0)