Update 'train_pfi_scheduler.py'

main
perdanamain 10 months ago
parent 19481c7cf6
commit 8ecae78f74

@ -41,11 +41,11 @@ def fire_and_forget_api_call(**context):
Fungsi ini mengirim permintaan API dan tidak menunggu respons (fire and forget)
"""
# Menggunakan HttpHook untuk mendapatkan konfigurasi koneksi
http = HttpHook(method='GET', http_conn_id='train-pfi-app')
http = HttpHook(method='GET', http_conn_id='pfi-api')
# Mendapatkan session dan base URL
session = http.get_conn()
endpoint = '/train'
endpoint = '/train/train'
# Generate request ID untuk tracking
request_id = str(uuid.uuid4())
@ -101,8 +101,8 @@ train_async = PythonOperator(
# Ini hanya untuk memulai request tanpa menunggu respons lengkap
train_alt = HttpOperator(
task_id='train_alt',
http_conn_id='train-pfi-app',
endpoint='/train',
http_conn_id='pfi-api',
endpoint='/train/train',
method='GET',
headers={
'Content-Type': 'application/json',

Loading…
Cancel
Save