deepkoluguri
/
agents-runtime
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Enforce JSON mode for Ollama and add Temporal RetryPolicy
Build Agents Runtime / build (push) Successful in 2m6s Details

This commit is contained in:
Deep Koluguri 2026-05-23 09:50:58 -04:00
parent 3c5d08933e
commit 6df7a23d57
2 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/curaflow_agent/activities.py
View File

@ -17,7 +17,8 @@ def get_llm():
model=LLM_MODEL, model=LLM_MODEL,
api_key="ollama", # Ollama doesn't strictly need an API key api_key="ollama", # Ollama doesn't strictly need an API key
base_url=OLLAMA_BASE_URL, base_url=OLLAMA_BASE_URL,
temperature=0.1 temperature=0.1,
model_kwargs={"response_format": {"type": "json_object"}}
) )
@activity.defn @activity.defn

17
src/curaflow_agent/workflows.py
View File

@ -1,5 +1,6 @@
from datetime import timedelta from datetime import timedelta
from temporalio import workflow from temporalio import workflow
from temporalio.common import RetryPolicy
with workflow.unsafe.imports_passed_through(): with workflow.unsafe.imports_passed_through():
from .activities import ( from .activities import (
@ -23,14 +24,14 @@ class ClinicalIntakeWorkflow:
structured_note = await workflow.execute_activity( structured_note = await workflow.execute_activity(
structure_note_activity, structure_note_activity,
raw_dictation, raw_dictation,
start_to_close_timeout=timedelta(minutes=10) retry_policy=RetryPolicy(maximum_attempts=3), start_to_close_timeout=timedelta(minutes=10)
) )
# Step 2: Medical Coder (Generate ICD-10/CPT codes) # Step 2: Medical Coder (Generate ICD-10/CPT codes)
billing_codes = await workflow.execute_activity( billing_codes = await workflow.execute_activity(
generate_billing_codes_activity, generate_billing_codes_activity,
structured_note, structured_note,
start_to_close_timeout=timedelta(minutes=10) retry_policy=RetryPolicy(maximum_attempts=3), start_to_close_timeout=timedelta(minutes=10)
) )
# Step 3: Save to the client's database via ORM # Step 3: Save to the client's database via ORM
@ -43,7 +44,7 @@ class ClinicalIntakeWorkflow:
save_status = await workflow.execute_activity( save_status = await workflow.execute_activity(
save_clinical_record_activity, save_clinical_record_activity,
save_payload, save_payload,
start_to_close_timeout=timedelta(minutes=1) retry_policy=RetryPolicy(maximum_attempts=3), start_to_close_timeout=timedelta(minutes=1)
) )
return { return {
@ -99,14 +100,14 @@ class LabResultWatcherWorkflow:
anomaly_report = await workflow.execute_activity( anomaly_report = await workflow.execute_activity(
analyze_lab_anomaly_activity, analyze_lab_anomaly_activity,
input_data, input_data,
start_to_close_timeout=timedelta(minutes=5) retry_policy=RetryPolicy(maximum_attempts=3), start_to_close_timeout=timedelta(minutes=5)
) )
if anomaly_report.get("is_dangerous", False): if anomaly_report.get("is_dangerous", False):
await workflow.execute_activity( await workflow.execute_activity(
send_alert_activity, send_alert_activity,
anomaly_report, anomaly_report,
start_to_close_timeout=timedelta(minutes=1) retry_policy=RetryPolicy(maximum_attempts=3), start_to_close_timeout=timedelta(minutes=1)
) )
self._alerts.append(anomaly_report) self._alerts.append(anomaly_report)
alerts_triggered += 1 alerts_triggered += 1
@ -137,7 +138,7 @@ class AppointmentBookingWorkflow:
intent_data = await workflow.execute_activity( intent_data = await workflow.execute_activity(
"parse_booking_intent_activity", "parse_booking_intent_activity",
self._chat_history, self._chat_history,
schedule_to_close_timeout=timedelta(minutes=5) retry_policy=RetryPolicy(maximum_attempts=3), schedule_to_close_timeout=timedelta(minutes=5)
) )
reply_msg = intent_data.get("reply_message", "I can help with that.") reply_msg = intent_data.get("reply_message", "I can help with that.")
@ -147,7 +148,7 @@ class AppointmentBookingWorkflow:
await workflow.execute_activity( await workflow.execute_activity(
"send_whatsapp_reply_activity", "send_whatsapp_reply_activity",
{"phone_number": phone_number, "message": reply_msg}, {"phone_number": phone_number, "message": reply_msg},
schedule_to_close_timeout=timedelta(minutes=1) retry_policy=RetryPolicy(maximum_attempts=3), schedule_to_close_timeout=timedelta(minutes=1)
) )
# 3. Check if all details are gathered # 3. Check if all details are gathered
@ -162,7 +163,7 @@ class AppointmentBookingWorkflow:
"appointment_date": intent_data.get("appointment_date"), "appointment_date": intent_data.get("appointment_date"),
"appointment_time": intent_data.get("appointment_time") "appointment_time": intent_data.get("appointment_time")
}, },
schedule_to_close_timeout=timedelta(seconds=30) retry_policy=RetryPolicy(maximum_attempts=3), schedule_to_close_timeout=timedelta(seconds=30)
) )
return intent_data return intent_data