codeflash-ai/codeflash-internal
1
0
Fork 0
Code Issues Pull requests 39 Projects Releases Packages Wiki Activity Actions 2.0k

Merge branch 'main' into update-prompts-mini-5

Browse source
This commit is contained in:
Kevin Turcios 2026-01-09 01:56:37 -05:00 committed by GitHub
commit 93519bbcbe
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
django/aiservice/optimization_review/optimization_review.py
View file

@ -8,17 +8,21 @@ from pathlib import Path
from typing import TYPE_CHECKING
import sentry_sdk
from aiservice.env_specific import debug_log_sensitive_data
from aiservice.models.aimodels import OPTIMIZATION_REVIEW_MODEL, calculate_llm_cost
from log_features.log_event import update_optimization_cost, update_optimization_features_review
from ninja import NinjaAPI, Schema
from openai.types.chat import ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam
from packaging import version
from aiservice.analytics.posthog import ph
from aiservice.env_specific import debug_log_sensitive_data
from aiservice.llm import OPTIMIZATION_REVIEW_MODEL, calculate_llm_cost, call_llm
from authapp.auth import AuthenticatedRequest
from log_features.log_event import update_optimization_cost, update_optimization_features_review
if TYPE_CHECKING:
from aiservice.llm import LLM
from openai.types.chat import ChatCompletionMessageParam
from aiservice.llm import LLM
optimization_review_api = NinjaAPI(urls_namespace="optimization_review")
current_dir = Path(__file__).parent
@ -93,7 +97,9 @@ def _build_optimization_review_messages(data: OptimizationReviewSchema) -> list[
async def get_optimization_review(
request, data: OptimizationReviewSchema, optimization_review_model: LLM = OPTIMIZATION_REVIEW_MODEL
request: AuthenticatedRequest,
data: OptimizationReviewSchema,
optimization_review_model: LLM = OPTIMIZATION_REVIEW_MODEL,
) -> tuple[int, OptimizationReviewResponseSchema | OptimizationReviewErrorSchema]:
"""Compute optimization review via Claude."""
ph(request.user, "aiservice-optimization-review-called")
@ -186,7 +192,7 @@ async def get_optimization_review(
},
)
async def optimization_review(
request, data: OptimizationReviewSchema
request: AuthenticatedRequest, data: OptimizationReviewSchema
) -> tuple[int, OptimizationReviewResponseSchema | OptimizationReviewErrorSchema]:
response_code, output = await get_optimization_review(request, data)
try: