GuglielmoTor commited on
Commit
27a3988
·
verified ·
1 Parent(s): 4a82e4b

Update features/insight_and_tasks/agents/task_extraction_agent.py

Browse files
features/insight_and_tasks/agents/task_extraction_agent.py CHANGED
@@ -8,7 +8,7 @@ from google.adk.runners import InMemoryRunner # Assuming this is used for direct
8
  from google.genai import types as genai_types # For constructing ADK agent inputs
9
 
10
  # Project-specific imports
11
- from insight_and_tasks.data_models.tasks import (
12
  TaskExtractionOutput,
13
  OKR,
14
  KeyResult,
@@ -19,7 +19,7 @@ from insight_and_tasks.data_models.tasks import (
19
  TaskType,
20
  DataSubject # Ensure all are imported
21
  )
22
- from insight_and_tasks.utils.retry_mechanism import RetryMechanism # If retries are needed for ADK calls
23
 
24
  # Configure logger for this module
25
  logger = logging.getLogger(__name__)
 
8
  from google.genai import types as genai_types # For constructing ADK agent inputs
9
 
10
  # Project-specific imports
11
+ from features.insight_and_tasks.data_models.tasks import (
12
  TaskExtractionOutput,
13
  OKR,
14
  KeyResult,
 
19
  TaskType,
20
  DataSubject # Ensure all are imported
21
  )
22
+ from features.insight_and_tasks.utils.retry_mechanism import RetryMechanism # If retries are needed for ADK calls
23
 
24
  # Configure logger for this module
25
  logger = logging.getLogger(__name__)