Add function_id tracking and enable JSON mode for all AI functions

This commit is contained in:
Gitea Deploy
2025-11-10 06:28:34 +00:00
parent c4c953fae2
commit 38324aedcd
17 changed files with 544 additions and 113 deletions

View File

@@ -299,6 +299,9 @@ def generate_content_core(task_ids: List[int], account_id: int = None, progress_
# Get model config from settings
model_config = get_model_config('generate_content')
# Generate function_id for tracking (ai-generate-content-02 for legacy path)
function_id = "ai-generate-content-02"
# Call AI using centralized request handler
ai_core = AICore(account=account)
result = ai_core.run_ai_request(
@@ -307,7 +310,8 @@ def generate_content_core(task_ids: List[int], account_id: int = None, progress_
max_tokens=model_config.get('max_tokens'),
temperature=model_config.get('temperature'),
response_format=model_config.get('response_format'),
function_name='generate_content'
function_name='generate_content',
function_id=function_id # Pass function_id for tracking
)
if result.get('error'):

View File

@@ -282,6 +282,9 @@ def generate_ideas_core(cluster_id: int, account_id: int = None, progress_callba
# Get model config from settings
model_config = get_model_config('generate_ideas')
# Generate function_id for tracking (ai-generate-ideas-02 for legacy path)
function_id = "ai-generate-ideas-02"
# Call AI using centralized request handler
ai_core = AICore(account=account)
result = ai_core.run_ai_request(
@@ -291,6 +294,7 @@ def generate_ideas_core(cluster_id: int, account_id: int = None, progress_callba
temperature=model_config.get('temperature'),
response_format=model_config.get('response_format'),
function_name='generate_ideas',
function_id=function_id, # Pass function_id for tracking
tracker=tracker
)