mirror of
https://github.com/codeflash-ai/codeflash-internal.git
synced 2026-05-04 18:25:18 +00:00
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:
parent
6bba3a3093
commit
7703c35cc1
2 changed files with 3 additions and 3 deletions
|
|
@ -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")
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
)
|
||||
|
|
|
|||
Loading…
Reference in a new issue