Update insight_and_tasks/agents/task_extraction_agent.py

insight_and_tasks/agents/task_extraction_agent.py

@@ -8,7 +8,7 @@ from google.adk.runners import InMemoryRunner # Assuming this is used for direct
 from google.genai import types as genai_types # For constructing ADK agent inputs
 
 # Project-specific imports
-from data_models.tasks import (
+from insight_and_tasks.data_models.tasks import (
     TaskExtractionOutput, 
     OKR, 
     KeyResult, 
@@ -19,7 +19,7 @@ from data_models.tasks import (
     TaskType, 
     DataSubject # Ensure all are imported
 )
-from utils.retry_mechanism import RetryMechanism # If retries are needed for ADK calls
+from insight_and_tasks.utils.retry_mechanism import RetryMechanism # If retries are needed for ADK calls
 
 # Configure logger for this module
 logger = logging.getLogger(__name__)