salman/igny8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sd

Browse Source
This commit is contained in:
alorig
2025-11-10 00:22:46 +05:00
parent 71487575bd
commit 3633ccb7ca
5 changed files with 27 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/igny8_core/ai/functions/__init__.py
View File

@@ -2,14 +2,16 @@
AI Function implementations AI Function implementations
""" """
from igny8_core.ai.functions.auto_cluster import AutoClusterFunction from igny8_core.ai.functions.auto_cluster import AutoClusterFunction
from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction, generate_ideas_core # REMOVED: generate_ideas function removed
# 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_content import GenerateContentFunction, generate_content_core
from igny8_core.ai.functions.generate_images import GenerateImagesFunction, generate_images_core from igny8_core.ai.functions.generate_images import GenerateImagesFunction, generate_images_core
__all__ = [ __all__ = [
'AutoClusterFunction', 'AutoClusterFunction',
'GenerateIdeasFunction', # REMOVED: generate_ideas function removed
'generate_ideas_core', # 'GenerateIdeasFunction',
# 'generate_ideas_core',
'GenerateContentFunction', 'GenerateContentFunction',
'generate_content_core', 'generate_content_core',
'GenerateImagesFunction', 'GenerateImagesFunction',

3
backend/igny8_core/ai/prompts.py
View File

@@ -117,7 +117,8 @@ Make sure each prompt is detailed enough for image generation, describing the vi
# Mapping from function names to prompt types # Mapping from function names to prompt types
FUNCTION_TO_PROMPT_TYPE = { FUNCTION_TO_PROMPT_TYPE = {
'auto_cluster': 'clustering', 'auto_cluster': 'clustering',
'generate_ideas': 'ideas', # REMOVED: generate_ideas function removed
# 'generate_ideas': 'ideas',
'generate_content': 'content_generation', 'generate_content': 'content_generation',
'generate_images': 'image_prompt_extraction', 'generate_images': 'image_prompt_extraction',
'extract_image_prompts': 'image_prompt_extraction', 'extract_image_prompts': 'image_prompt_extraction',

12
backend/igny8_core/ai/registry.py
View File

@@ -66,10 +66,11 @@ def _load_auto_cluster():
from igny8_core.ai.functions.auto_cluster import AutoClusterFunction from igny8_core.ai.functions.auto_cluster import AutoClusterFunction
return AutoClusterFunction return AutoClusterFunction
def _load_generate_ideas(): # REMOVED: generate_ideas function removed
"""Lazy loader for generate_ideas function""" # def _load_generate_ideas():
from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction # """Lazy loader for generate_ideas function"""
return GenerateIdeasFunction # from igny8_core.ai.functions.generate_ideas import GenerateIdeasFunction
# return GenerateIdeasFunction
def _load_generate_content(): def _load_generate_content():
"""Lazy loader for generate_content function""" """Lazy loader for generate_content function"""
@@ -82,7 +83,8 @@ def _load_generate_images():
return GenerateImagesFunction return GenerateImagesFunction
register_lazy_function('auto_cluster', _load_auto_cluster) register_lazy_function('auto_cluster', _load_auto_cluster)
register_lazy_function('generate_ideas', _load_generate_ideas) # REMOVED: generate_ideas function removed
# register_lazy_function('generate_ideas', _load_generate_ideas)
register_lazy_function('generate_content', _load_generate_content) register_lazy_function('generate_content', _load_generate_content)
register_lazy_function('generate_images', _load_generate_images) register_lazy_function('generate_images', _load_generate_images)

16
backend/igny8_core/ai/settings.py
View File

@@ -11,12 +11,13 @@ MODEL_CONFIG = {
"temperature": 0.7, "temperature": 0.7,
"response_format": {"type": "json_object"}, # Auto-enabled for JSON mode models "response_format": {"type": "json_object"}, # Auto-enabled for JSON mode models
}, },
"generate_ideas": { # REMOVED: generate_ideas function removed
"model": "gpt-4.1", # "generate_ideas": {
"max_tokens": 4000, # "model": "gpt-4.1",
"temperature": 0.7, # "max_tokens": 4000,
"response_format": {"type": "json_object"}, # "temperature": 0.7,
}, # "response_format": {"type": "json_object"},
# },
"generate_content": { "generate_content": {
"model": "gpt-4.1", "model": "gpt-4.1",
"max_tokens": 8000, "max_tokens": 8000,
@@ -40,7 +41,8 @@ MODEL_CONFIG = {
FUNCTION_ALIASES = { FUNCTION_ALIASES = {
"cluster_keywords": "auto_cluster", "cluster_keywords": "auto_cluster",
"auto_cluster_keywords": "auto_cluster", "auto_cluster_keywords": "auto_cluster",
"auto_generate_ideas": "generate_ideas", # REMOVED: generate_ideas function removed
# "auto_generate_ideas": "generate_ideas",
"auto_generate_content": "generate_content", "auto_generate_content": "generate_content",
"auto_generate_images": "generate_images", "auto_generate_images": "generate_images",
} }

6
backend/igny8_core/ai/tests/test_run.py
View File

@@ -12,7 +12,8 @@ os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'igny8.settings')
django.setup() django.setup()
from igny8_core.ai.functions.auto_cluster import AutoClusterFunction from igny8_core.ai.functions.auto_cluster import AutoClusterFunction
from igny8_core.ai.functions.generate_ideas import generate_ideas_core # REMOVED: generate_ideas function removed
# from igny8_core.ai.functions.generate_ideas import generate_ideas_core
from igny8_core.ai.functions.generate_content import generate_content_core from igny8_core.ai.functions.generate_content import generate_content_core
from igny8_core.ai.functions.generate_images import generate_images_core from igny8_core.ai.functions.generate_images import generate_images_core
from igny8_core.ai.ai_core import AICore from igny8_core.ai.ai_core import AICore
@@ -122,7 +123,8 @@ if __name__ == '__main__':
test_ai_core() test_ai_core()
test_json_extraction() test_json_extraction()
test_auto_cluster() test_auto_cluster()
test_generate_ideas() # REMOVED: generate_ideas function removed
# test_generate_ideas()
test_generate_content() test_generate_content()
test_generate_images() test_generate_images()