Spaces:
Sleeping
Sleeping
Update features/insight_and_tasks/agents/task_extraction_agent.py
Browse files
features/insight_and_tasks/agents/task_extraction_agent.py
CHANGED
@@ -26,6 +26,10 @@ logger = logging.getLogger(__name__)
|
|
26 |
|
27 |
DEFAULT_AGENT_MODEL = "gemini-2.5-flash-preview-05-20" # Or your specific model
|
28 |
|
|
|
|
|
|
|
|
|
29 |
class TaskExtractionAgent:
|
30 |
"""
|
31 |
Agent specialized in extracting actionable tasks and OKRs from analysis insights,
|
|
|
26 |
|
27 |
DEFAULT_AGENT_MODEL = "gemini-2.5-flash-preview-05-20" # Or your specific model
|
28 |
|
29 |
+
def _get_enum_values_str(enum_class: type[Enum]) -> str:
|
30 |
+
"""Helper to format enum values for inclusion in the prompt."""
|
31 |
+
return ", ".join([f"'{e.value}'" for e in enum_class])
|
32 |
+
|
33 |
class TaskExtractionAgent:
|
34 |
"""
|
35 |
Agent specialized in extracting actionable tasks and OKRs from analysis insights,
|