File tree Expand file tree Collapse file tree 1 file changed +12
-16
lines changed
Expand file tree Collapse file tree 1 file changed +12
-16
lines changed Original file line number Diff line number Diff line change @@ -1277,25 +1277,21 @@ def process_call_model(
12771277 assert False
12781278 # evaluate input
12791279 model_input : ModelInput
1280- if concrete_block .input is not None : # If not implicit, then input must be a block
1281- model_input_future , _ , _ , input_trace = process_block_of (
1282- concrete_block ,
1283- "input" ,
1284- state .with_yield_result (False ).with_yield_background (False ),
1285- scope ,
1286- loc ,
1287- )
1288- model_input_result = model_input_future .result ()
1289- if isinstance (model_input_result , str ):
1290- model_input = [{"role" : state .role , "content" : model_input_result }]
1291- else :
1292- model_input = model_input_result
1280+ model_input_future , _ , _ , concrete_block = process_block_of (
1281+ concrete_block ,
1282+ "input" ,
1283+ state .with_yield_result (False ).with_yield_background (False ),
1284+ scope ,
1285+ loc ,
1286+ )
1287+ model_input_result = model_input_future .result ()
1288+ if isinstance (model_input_result , str ):
1289+ model_input = [{"role" : state .role , "content" : model_input_result }]
12931290 else :
1294- model_input = scope ["pdl_context" ] # pyright: ignore
1295- input_trace = None
1291+ model_input = model_input_result
12961292 concrete_block = concrete_block .model_copy (
12971293 update = {
1298- "input " : input_trace ,
1294+ "pdl__model_input " : model_input ,
12991295 }
13001296 )
13011297 # Execute model call
You can’t perform that action at this time.
0 commit comments