diff --git a/src/pdl/pdl_interpreter.py b/src/pdl/pdl_interpreter.py index e5d8a719d..7dd55cb4c 100644 --- a/src/pdl/pdl_interpreter.py +++ b/src/pdl/pdl_interpreter.py @@ -1427,7 +1427,7 @@ def process_call_model( } ) - model_input = [{"role": m["role"], "content": m["content"]} for m in model_input] + model_input = [{k: v for k, v in m.items() if k != "defsite"} for m in model_input] # Execute model call try: litellm_params = {}