Skip to content

Commit c4cba4c

Browse files
committed
passing in top_p and temperature for agentic
1 parent 365c88a commit c4cba4c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

lib/idp_common_pkg/idp_common/extraction/agentic_idp.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1004,7 +1004,11 @@ async def structured_output_async(
10041004
# Track token usage
10051005
token_usage = _initialize_token_usage()
10061006
agent = Agent(
1007-
model=BedrockModel(**model_config), # pyright: ignore[reportArgumentType]
1007+
model=BedrockModel(
1008+
**model_config,
1009+
temperature=config.extraction.temperature,
1010+
top_p=config.extraction.top_p,
1011+
), # pyright: ignore[reportArgumentType]
10081012
tools=tools,
10091013
system_prompt=final_system_prompt,
10101014
state={

0 commit comments

Comments
 (0)