diff --git a/src/curaflow_agent/activities.py b/src/curaflow_agent/activities.py
index 733ded1..e1f29c2 100644
--- a/src/curaflow_agent/activities.py
+++ b/src/curaflow_agent/activities.py
@@ -3,6 +3,7 @@ import json
 from temporalio import activity
 from langchain_openai import ChatOpenAI
 from langchain_core.prompts import PromptTemplate
+from pydantic import BaseModel, Field
 from .models import ClinicalNote, BillingCodesOutput
 from .lab_models import AnomalyReport