Skip to content

Commit 2772d2b

Browse files
author
dori
committed
feat: fix build
1 parent 07b1b50 commit 2772d2b

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

src/mcp_as_a_judge/server.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -120,11 +120,10 @@ async def build_workflow(
120120

121121
# STEP 3: Use messaging layer to get LLM evaluation with dynamic token limit
122122
model_name = await detect_model_name(ctx)
123-
dynamic_max_tokens = get_llm_output_limit(model_name)
124123
response_text = await llm_provider.send_message(
125124
messages=messages,
126125
ctx=ctx,
127-
max_tokens=dynamic_max_tokens,
126+
max_tokens=get_llm_output_limit(model_name),
128127
prefer_sampling=True,
129128
)
130129

@@ -497,11 +496,10 @@ async def _evaluate_coding_plan(
497496

498497
# Use dynamic token limit for response
499498
model_name = await detect_model_name(ctx)
500-
dynamic_max_tokens = get_llm_output_limit(model_name)
501499
response_text = await llm_provider.send_message(
502500
messages=messages,
503501
ctx=ctx,
504-
max_tokens=dynamic_max_tokens,
502+
max_tokens=get_llm_output_limit(model_name),
505503
prefer_sampling=True,
506504
)
507505

@@ -695,11 +693,10 @@ async def judge_code_change(
695693

696694
# STEP 3: Use messaging layer for LLM evaluation with dynamic token limit
697695
model_name = await detect_model_name(ctx)
698-
dynamic_max_tokens = get_llm_output_limit(model_name)
699696
response_text = await llm_provider.send_message(
700697
messages=messages,
701698
ctx=ctx,
702-
max_tokens=dynamic_max_tokens,
699+
max_tokens=get_llm_output_limit(model_name),
703700
prefer_sampling=True,
704701
)
705702

test_real_scenario.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,14 @@ async def test_real_scenario():
2323
identified_gaps=[
2424
"Required fields for profile updates",
2525
"Validation rules for each field",
26-
"Authentication requirements",
26+
"Authentication requirements"
2727
],
2828
specific_questions=[
2929
"What fields should be updatable?",
3030
"Should we validate email format?",
31-
"Is admin approval required?",
31+
"Is admin approval required?"
3232
],
33-
ctx=mock_ctx,
33+
ctx=mock_ctx
3434
)
3535

3636
print(f"Result type: {type(result)}")

0 commit comments

Comments
 (0)