Explicit model input inherit the current role (#149)

mandel · web-flow · commit 7676870ccadc · 2024-10-15T13:34:45.000-04:00
diff --git a/src/pdl/pdl_interpreter.py b/src/pdl/pdl_interpreter.py
@@ -1036,7 +1036,7 @@ def step_call_model(
             loc,
         )
         if isinstance(model_input_result, str):
-            model_input = [{"role": None, "content": model_input_result}]
+            model_input = [{"role": state.role, "content": model_input_result}]
         else:
             model_input = model_input_result
     else:

Original file line number	Diff line number	Diff line change
`@@ -1036,7 +1036,7 @@ def step_call_model(`
`1036`	`1036`	`loc,`
`1037`	`1037`	`)`
`1038`	`1038`	`if isinstance(model_input_result, str):`
`1039`		`- model_input = [{"role": None, "content": model_input_result}]`
	`1039`	`+ model_input = [{"role": state.role, "content": model_input_result}]`
`1040`	`1040`	`else:`
`1041`	`1041`	`model_input = model_input_result`
`1042`	`1042`	`else:`