Skip to content

Commit 2f62978

Browse files
fix: skip logprobs for reasoning models in map_handler
Reasoning models (gpt-5.x, o-series) also reject logprobs parameter. Conditionally skip logprobs/top_logprobs when targeting these models. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent b833c70 commit 2f62978

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

src/ContentProcessor/src/libs/pipeline/handlers/map_handler.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from agent_framework import Content, Message
1818
from pdf2image import convert_from_bytes
1919

20-
from libs.agent_framework.agent_builder import AgentBuilder
20+
from libs.agent_framework.agent_builder import AgentBuilder, _is_reasoning_model, _resolve_model_name
2121
from libs.agent_framework.agent_framework_helper import AgentFrameworkHelper
2222
from libs.agent_framework.azure_openai_response_retry import ContextTrimConfig
2323
from libs.application.application_context import AppContext
@@ -255,12 +255,19 @@ async def execute(self, context: MessageContext) -> StepResult:
255255
.build()
256256
)
257257

258+
# logprobs is not supported by reasoning models (o1, o3, gpt-5.x)
259+
model_name = _resolve_model_name(agent_client)
260+
if model_name and _is_reasoning_model(model_name):
261+
run_options = {}
262+
else:
263+
run_options = {"logprobs": True, "top_logprobs": 5}
264+
258265
gpt_response = await agent.run(
259266
messages=Message(
260267
"user",
261268
contents=self._to_agent_framework_contents(user_content),
262269
),
263-
options={"logprobs": True, "top_logprobs": 5},
270+
options=run_options,
264271
)
265272

266273
response_content = gpt_response.text # Json format string

0 commit comments

Comments
 (0)