codeflash-admin/codeflash-internal
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash-internal.git synced 2026-05-04 18:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refine prompt, remove hard rules and make them more fluid

Browse source
This commit is contained in:
Aseem Saxena 2025-10-17 16:00:02 -07:00
parent 5fe68a3b4b
commit e33c569365
1 changed files with 3 additions and 3 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

@ -15,7 +15,6 @@ from openai.types.chat import ChatCompletionSystemMessageParam, ChatCompletionUs
if TYPE_CHECKING:
from aiservice.models.aimodels import LLM
from django.core.handlers.wsgi import WSGIRequest
from aiservice.analytics.posthog import ph
optimization_impact_api = NinjaAPI(urls_namespace="optimization_impact")
@ -27,6 +26,7 @@ class OptimizationImpactErrorSchema(Schema):
class ImpactLevel(str, Enum):
HIGH = "high"
MEDIUM = "medium"
LOW = "low"
@ -163,7 +163,7 @@ async def get_optimization_impact(
impact_level = json.loads(match.group(2).lower().strip())
impact_explanation = match.group(1) + match.group(3)
impact = OptimizationImpactResponseSchema(
impact=impact_level["impact"], impact_explanation=impact_explanation
impact=impact_level["rating"], impact_explanation=impact_explanation
)
except Exception as e:
# invalid response
@ -191,7 +191,7 @@ async def get_optimization_impact(
},
)
async def optimization_impact(
request: WSGIRequest, data: OptimizationImpactSchema
request, data: OptimizationImpactSchema
) -> tuple[int, OptimizationImpactResponseSchema | OptimizationImpactErrorSchema]:
response_code, output = await get_optimization_impact(request, data)
try: