GuglielmoTor commited on
Commit
1714a1c
·
verified ·
1 Parent(s): bf08717

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,