From 1fe59aa064cc1580ab39ae13a7ccf9621d818779 Mon Sep 17 00:00:00 2001 From: Benjamin McCormick Date: Wed, 11 Feb 2026 13:12:41 -0500 Subject: [PATCH] fix: comment out response model in vllm --- .gitignore | 1 + src/utils/clients.py | 26 +++++++++++++++----------- 2 files changed, 16 insertions(+), 11 deletions(-) diff --git a/.gitignore b/.gitignore index 9a31fc38..00c40fc1 100644 --- a/.gitignore +++ b/.gitignore @@ -191,3 +191,4 @@ metrics.jsonl AGENTS.md lancedb_data/ grafana-data/ +traces.jsonl diff --git a/src/utils/clients.py b/src/utils/clients.py index c3641b7b..7b89ac01 100644 --- a/src/utils/clients.py +++ b/src/utils/clients.py @@ -1936,17 +1936,17 @@ async def honcho_llm_call_inner( # NOTE: this is all specific to the Representation model. # Do not call with any other response model. if provider == "vllm" and response_model: - if response_model is not PromptRepresentation: - raise NotImplementedError( - "vLLM structured output currently supports only PromptRepresentation" - ) - openai_params["response_format"] = { - "type": "json_schema", - "json_schema": { - "name": response_model.__name__, - "schema": response_model.model_json_schema(), - }, - } + # if response_model is not PromptRepresentation: + # raise NotImplementedError( + # "vLLM structured output currently supports only PromptRepresentation" + # ) + # openai_params["response_format"] = { + # "type": "json_schema", + # "json_schema": { + # "name": response_model.__name__, + # "schema": response_model.model_json_schema(), + # }, + # } if stop_seqs: openai_params["stop"] = stop_seqs vllm_response: ChatCompletion = cast( @@ -1962,8 +1962,12 @@ async def honcho_llm_call_inner( if vllm_response.choices[0].message.content is not None: test_rep = vllm_response.choices[0].message.content + # logger.info(f"test_rep: {test_rep}") + final = validate_and_repair_json(test_rep) + # logger.info(f"final: {final}") + # Schema-aware repair: ensure deductive observations have required fields repaired_data = json.loads(final)