Spaces:
Running
Running
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__)
|