Skip to content

Commit b0c37c5

Browse files
authored
fix: remove defsite from messages in model inputs (#956)
Signed-off-by: Louis Mandel <[email protected]>
1 parent 7c79e73 commit b0c37c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pdl/pdl_interpreter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1427,7 +1427,7 @@ def process_call_model(
14271427
}
14281428
)
14291429

1430-
model_input = [{"role": m["role"], "content": m["content"]} for m in model_input]
1430+
model_input = [{k: v for k, v in m.items() if k != "defsite"} for m in model_input]
14311431
# Execute model call
14321432
try:
14331433
litellm_params = {}

0 commit comments

Comments
 (0)