engineering/unrhodecode
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request !2834 from rhodecode-enterprise-ce feature/RCCE-325_AI-model-versions-auto-update

Browse source 
feature: implements possibility to update AI model versions for admin
This commit is contained in:
Andrii Verbytskyi 2025-10-10 18:22:37 +00:00
commit 7d2a9d1d48
14 changed files with 479 additions and 95 deletions
Download patch file Download diff file Expand all files Collapse all files

14
rhodecode/lib/celerylib/tasks.py
View file

@ -513,11 +513,6 @@ def start_ai_code_review(pull_request_id):
log.info("AI model or model version or API key is not set, review not possible.")
return
service = get_ai_service(
api_key=ai_api_key,
model_name=ai_model,
version=ai_model_version,
)
instructions = rc_settings.get_setting_by_name("ai_code_review_instructions")
if instructions:
instructions = instructions.app_settings_value.split("\r\n")
@ -535,6 +530,12 @@ def start_ai_code_review(pull_request_id):
)
try:
service = get_ai_service(
api_key=ai_api_key,
model_name=ai_model,
api_model_version_name=ai_model_version,
)
response = service.code_review(diffset, instructions=instructions)
_add_comments(response, pull_request, log, ai_user)
audit_logger.store(
@ -548,7 +549,6 @@ def start_ai_code_review(pull_request_id):
},
repo=pull_request.target_repo,
)
except AIServiceError as e:
log.error("AI service error: %s", e)
audit_logger.store(
@ -556,7 +556,7 @@ def start_ai_code_review(pull_request_id):
user=ai_user,
action_data={
audit_logger.PR_ID: pull_request_id,
audit_logger.AI_MODEL: service.get_model_name(),
audit_logger.AI_MODEL: service.get_api_model_name(),
"error": True,
"error_message": str(e),
},