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

fix demo call

Browse source
This commit is contained in:
Kevin Turcios 2025-10-17 13:21:40 -07:00
parent 4bb592541d
commit 423648b606
2 changed files with 4 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
django/aiservice/optimization_impact/optimization_impact.py
View file

@ -4,7 +4,6 @@ import json
import logging
import re
from enum import Enum
from json import JSONDecodeError
from typing import TYPE_CHECKING, cast
import sentry_sdk
@ -12,13 +11,12 @@ from ninja import NinjaAPI, Schema
from openai.types.chat import ChatCompletionSystemMessageParam, ChatCompletionUserMessageParam
from aiservice.env_specific import create_claude_client, debug_log_sensitive_data
from aiservice.models.aimodels import OPTIMIZATION_IMPACT_MODEL, calculate_llm_cost
from log_features.log_event import update_optimization_cost, update_metadata_optimization_event
from aiservice.models.aimodels import LLM, OPTIMIZATION_IMPACT_MODEL, calculate_llm_cost # noqa: TC001
from log_features.log_event import update_metadata_optimization_event, update_optimization_cost
if TYPE_CHECKING:
from django.core.handlers.wsgi import WSGIRequest
from aiservice.models.aimodels import LLM
from aiservice.analytics.posthog import ph
optimization_impact_api = NinjaAPI(urls_namespace="optimization_impact")

4
django/aiservice/optimizer/optimizer.py
View file

@ -64,7 +64,7 @@ optimizations_json = [
]
async def hack_for_demo(ctx: BaseOptimizerContext) -> list[OptimizeResponseSchema]:
async def hack_for_demo(ctx: BaseOptimizerContext) -> OptimizeResponseSchema:
response_list: list[OptimizeResponseItemSchema] = [
OptimizeResponseItemSchema(
explanation=optimization["explanation"],
@ -218,7 +218,7 @@ async def optimize(
return e.status_code, OptimizeErrorResponseSchema(error=e.message)
if should_hack_for_demo(request.user, ctx.source_code):
return hack_for_demo(ctx)
return 200, await hack_for_demo(ctx)
try:
async with asyncio.TaskGroup() as tg: