@@ -213,28 +213,28 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
213
213
// MODEL
214
214
['name ' => $ modelName , 'version ' => $ version , 'options ' => $ options ] = $ config ['model ' ];
215
215
216
- $ llmClass = match (strtolower ((string ) $ modelName )) {
216
+ $ modelClass = match (strtolower ((string ) $ modelName )) {
217
217
'gpt ' => GPT ::class,
218
218
'claude ' => Claude::class,
219
219
'llama ' => Llama::class,
220
220
'gemini ' => Gemini::class,
221
221
default => throw new \InvalidArgumentException (sprintf ('Model "%s" is not supported. ' , $ modelName )),
222
222
};
223
- $ llmDefinition = new Definition ($ llmClass );
223
+ $ modelDefinition = new Definition ($ modelClass );
224
224
if (null !== $ version ) {
225
- $ llmDefinition ->setArgument ('$name ' , $ version );
225
+ $ modelDefinition ->setArgument ('$name ' , $ version );
226
226
}
227
227
if (0 !== count ($ options )) {
228
- $ llmDefinition ->setArgument ('$options ' , $ options );
228
+ $ modelDefinition ->setArgument ('$options ' , $ options );
229
229
}
230
- $ llmDefinition ->addTag ('llm_chain.model.language_model ' );
231
- $ container ->setDefinition ('llm_chain.chain. ' .$ name .'.llm ' , $ llmDefinition );
230
+ $ modelDefinition ->addTag ('llm_chain.model.language_model ' );
231
+ $ container ->setDefinition ('llm_chain.chain. ' .$ name .'.model ' , $ modelDefinition );
232
232
233
233
// CHAIN
234
234
$ chainDefinition = (new Definition (Chain::class))
235
235
->setAutowired (true )
236
236
->setArgument ('$platform ' , new Reference ($ config ['platform ' ]))
237
- ->setArgument ('$llm ' , new Reference ('llm_chain.chain. ' .$ name .'.llm ' ));
237
+ ->setArgument ('$model ' , new Reference ('llm_chain.chain. ' .$ name .'.model ' ));
238
238
239
239
$ inputProcessors = [];
240
240
$ outputProcessors = [];
@@ -266,7 +266,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
266
266
}
267
267
268
268
$ toolboxDefinition = (new ChildDefinition ('llm_chain.toolbox.abstract ' ))
269
- ->replaceArgument ('$metadataFactory ' , new Reference ('llm_chain.toolbox. ' .$ name .'.chain_factory ' ))
269
+ ->replaceArgument ('$toolFactory ' , new Reference ('llm_chain.toolbox. ' .$ name .'.chain_factory ' ))
270
270
->replaceArgument ('$tools ' , $ tools );
271
271
$ container ->setDefinition ('llm_chain.toolbox. ' .$ name , $ toolboxDefinition );
272
272
@@ -436,10 +436,10 @@ private function processEmbedderConfig(int|string $name, array $config, Containe
436
436
$ modelDefinition ->setArgument ('$options ' , $ options );
437
437
}
438
438
$ modelDefinition ->addTag ('llm_chain.model.embeddings_model ' );
439
- $ container ->setDefinition ('llm_chain.embedder. ' .$ name .'.embeddings ' , $ modelDefinition );
439
+ $ container ->setDefinition ('llm_chain.embedder. ' .$ name .'.model ' , $ modelDefinition );
440
440
441
441
$ definition = new Definition (Embedder::class, [
442
- '$embeddings ' => new Reference ('llm_chain.embedder. ' .$ name .'.embeddings ' ),
442
+ '$model ' => new Reference ('llm_chain.embedder. ' .$ name .'.model ' ),
443
443
'$platform ' => new Reference ($ config ['platform ' ]),
444
444
'$store ' => new Reference ($ config ['store ' ]),
445
445
]);
0 commit comments