Refactor generate_ideas function integration and add auto_generate_ideas action for cluster idea generation
This commit is contained in:
@@ -2,16 +2,14 @@
|
||||
AI Function implementations
|
||||
"""
|
||||
from igny8_core.ai.functions.auto_cluster import AutoClusterFunction
|
||||
# REMOVED: generate_ideas function removed
|
||||
# from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction, generate_ideas_core
|
||||
from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction, generate_ideas_core
|
||||
from igny8_core.ai.functions.generate_content import GenerateContentFunction, generate_content_core
|
||||
from igny8_core.ai.functions.generate_images import GenerateImagesFunction, generate_images_core
|
||||
|
||||
__all__ = [
|
||||
'AutoClusterFunction',
|
||||
# REMOVED: generate_ideas function removed
|
||||
# 'GenerateIdeasFunction',
|
||||
# 'generate_ideas_core',
|
||||
'GenerateIdeasFunction',
|
||||
'generate_ideas_core',
|
||||
'GenerateContentFunction',
|
||||
'generate_content_core',
|
||||
'GenerateImagesFunction',
|
||||
|
||||
@@ -270,8 +270,7 @@ Make sure each prompt is detailed enough for image generation, describing the vi
|
||||
# Mapping from function names to prompt types
|
||||
FUNCTION_TO_PROMPT_TYPE = {
|
||||
'auto_cluster': 'clustering',
|
||||
# REMOVED: generate_ideas function removed
|
||||
# 'generate_ideas': 'ideas',
|
||||
'generate_ideas': 'ideas',
|
||||
'generate_content': 'content_generation',
|
||||
'generate_images': 'image_prompt_extraction',
|
||||
'extract_image_prompts': 'image_prompt_extraction',
|
||||
|
||||
@@ -66,11 +66,10 @@ def _load_auto_cluster():
|
||||
from igny8_core.ai.functions.auto_cluster import AutoClusterFunction
|
||||
return AutoClusterFunction
|
||||
|
||||
# REMOVED: generate_ideas function removed
|
||||
# def _load_generate_ideas():
|
||||
# """Lazy loader for generate_ideas function"""
|
||||
# from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction
|
||||
# return GenerateIdeasFunction
|
||||
def _load_generate_ideas():
|
||||
"""Lazy loader for generate_ideas function"""
|
||||
from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction
|
||||
return GenerateIdeasFunction
|
||||
|
||||
def _load_generate_content():
|
||||
"""Lazy loader for generate_content function"""
|
||||
@@ -83,8 +82,7 @@ def _load_generate_images():
|
||||
return GenerateImagesFunction
|
||||
|
||||
register_lazy_function('auto_cluster', _load_auto_cluster)
|
||||
# REMOVED: generate_ideas function removed
|
||||
# register_lazy_function('generate_ideas', _load_generate_ideas)
|
||||
register_lazy_function('generate_ideas', _load_generate_ideas)
|
||||
register_lazy_function('generate_content', _load_generate_content)
|
||||
register_lazy_function('generate_images', _load_generate_images)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user