Fix on max score (#1821)

- **Connected #743 in cli**
- **fix to inline i cli and max check**

---------

Co-authored-by: saga4 <saga4@codeflashs-MacBook-Air.local>
This commit is contained in:
Sarthak Agarwal 2025-09-22 18:22:10 -07:00 committed by GitHub
parent 6bba3a3093
commit 7703c35cc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions

View file

@ -144,7 +144,7 @@ class OptimizeSchema(Schema):
async def optimize(request, data: OptimizeSchema) -> tuple[int, OptimizeResponseSchema | OptimizeErrorResponseSchema]:
system_prompt = ASYNC_SYSTEM_PROMPT if data.is_async else SYSTEM_PROMPT
user_prompt = ASYNC_USER_PROMPT if data.is_async else USER_PROMPT
ctx: BaseOptimizerContext = BaseOptimizerContext.get_dynamic_context(system_prompt, user_prompt, data.source_code)
ph(request.user, "aiservice-optimize-called")
try:
@ -162,7 +162,7 @@ async def optimize(request, data: OptimizeSchema) -> tuple[int, OptimizeResponse
if not validate_trace_id(data.trace_id):
return 400, OptimizeErrorResponseSchema(error="Invalid trace ID. Please provide a valid UUIDv4.")
optimization_response_items, llm_cost = await optimize_python_code(
request.user, ctx, data.dependency_code, n=data.n_candidates or 5, python_version=python_version
request.user, ctx, data.dependency_code, n=min(data.n_candidates or 5, 5), python_version=python_version
)
if len(optimization_response_items) == 0:
ph(request.user, "aiservice-optimize-no-optimizations-found")

View file

@ -162,7 +162,7 @@ async def optimize(request, data: OptimizeSchemaLP) -> tuple[int, OptimizeRespon
ctx=ctx,
dependency_code=data.dependency_code,
line_profiler_results=data.line_profiler_results,
n=data.n_candidates_lp or 6,
n=min(data.n_candidates_lp or 6, 8),
lsp_mode=data.lsp_mode,
python_version=python_version,
)