@@ -213,28 +213,28 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
213213 // MODEL
214214 ['name ' => $ modelName , 'version ' => $ version , 'options ' => $ options ] = $ config ['model ' ];
215215
216- $ llmClass = match (strtolower ((string ) $ modelName )) {
216+ $ modelClass = match (strtolower ((string ) $ modelName )) {
217217 'gpt ' => GPT ::class,
218218 'claude ' => Claude::class,
219219 'llama ' => Llama::class,
220220 'gemini ' => Gemini::class,
221221 default => throw new \InvalidArgumentException (sprintf ('Model "%s" is not supported. ' , $ modelName )),
222222 };
223- $ llmDefinition = new Definition ($ llmClass );
223+ $ modelDefinition = new Definition ($ modelClass );
224224 if (null !== $ version ) {
225- $ llmDefinition ->setArgument ('$name ' , $ version );
225+ $ modelDefinition ->setArgument ('$name ' , $ version );
226226 }
227227 if (0 !== count ($ options )) {
228- $ llmDefinition ->setArgument ('$options ' , $ options );
228+ $ modelDefinition ->setArgument ('$options ' , $ options );
229229 }
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 );
232232
233233 // CHAIN
234234 $ chainDefinition = (new Definition (Chain::class))
235235 ->setAutowired (true )
236236 ->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 ' ));
238238
239239 $ inputProcessors = [];
240240 $ outputProcessors = [];
@@ -266,7 +266,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
266266 }
267267
268268 $ 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 ' ))
270270 ->replaceArgument ('$tools ' , $ tools );
271271 $ container ->setDefinition ('llm_chain.toolbox. ' .$ name , $ toolboxDefinition );
272272
@@ -436,10 +436,10 @@ private function processEmbedderConfig(int|string $name, array $config, Containe
436436 $ modelDefinition ->setArgument ('$options ' , $ options );
437437 }
438438 $ 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 );
440440
441441 $ definition = new Definition (Embedder::class, [
442- '$embeddings ' => new Reference ('llm_chain.embedder. ' .$ name .'.embeddings ' ),
442+ '$model ' => new Reference ('llm_chain.embedder. ' .$ name .'.model ' ),
443443 '$platform ' => new Reference ($ config ['platform ' ]),
444444 '$store ' => new Reference ($ config ['store ' ]),
445445 ]);
0 commit comments