Skip to content

Commit 4320ab0

Browse files
committed
job: repeat_min param
1 parent 399a2f6 commit 4320ab0

File tree

2 files changed

+16
-5
lines changed

2 files changed

+16
-5
lines changed

mindsdb_sdk/jobs.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -98,9 +98,15 @@ def get(self, name: str) -> Job:
9898
else:
9999
raise RuntimeError("Several jobs with the same name")
100100

101-
def create(self, name: str, query_str: str,
102-
start_at: dt.datetime = None, end_at: dt.datetime = None,
103-
repeat_str: str = None) -> Union[Job, None]:
101+
def create(
102+
self,
103+
name: str,
104+
query_str: str,
105+
start_at: dt.datetime = None,
106+
end_at: dt.datetime = None,
107+
repeat_str: str = None,
108+
repeat_min: int = None,
109+
) -> Union[Job, None]:
104110
"""
105111
Create new job in project and return it.
106112
@@ -114,6 +120,7 @@ def create(self, name: str, query_str: str,
114120
:param start_at: datetime, first start of job,
115121
:param end_at: datetime, when job have to be stopped,
116122
:param repeat_str: str, optional, how to repeat job (e.g. '1 hour', '2 weeks', '3 min')
123+
:param repeat_min: int, optional, period to repeat the job in minutes
117124
:return: Job object or None
118125
"""
119126

@@ -126,6 +133,10 @@ def create(self, name: str, query_str: str,
126133
end_str = end_at.strftime("%Y-%m-%d %H:%M:%S")
127134
else:
128135
end_str = None
136+
137+
if repeat_min is not None:
138+
repeat_str = f'{repeat_min} minutes'
139+
129140
ast_query = CreateJob(
130141
name=Identifier(name),
131142
query_str=query_str,

tests/test_sdk.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1044,14 +1044,14 @@ def check_project_jobs(self, project, mock_post):
10441044
project.jobs.create(
10451045
name='job2',
10461046
query_str='retrain m1',
1047-
repeat_str='1 min',
1047+
repeat_min=1,
10481048
start_at=dt.datetime(2025, 2, 5, 11, 22),
10491049
end_at=dt.date(2030, 1, 2)
10501050
)
10511051

10521052
check_sql_call(
10531053
mock_post,
1054-
f"CREATE JOB job2 (retrain m1) START '2025-02-05 11:22:00' END '2030-01-02 00:00:00' EVERY 1 min",
1054+
f"CREATE JOB job2 (retrain m1) START '2025-02-05 11:22:00' END '2030-01-02 00:00:00' EVERY 1 minutes",
10551055
call_stack_num=-2
10561056
)
10571057

0 commit comments

Comments
 (0)