From 945f93509f757cab564f006688d8a3f6b068b6f6 Mon Sep 17 00:00:00 2001
From: Christopher Hertel <mail@christopher-hertel.de>
Date: Mon, 28 Apr 2025 19:35:47 +0200
Subject: [PATCH] fix: model name argument on bundle extension/injection

---
 src/DependencyInjection/LlmChainExtension.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/DependencyInjection/LlmChainExtension.php b/src/DependencyInjection/LlmChainExtension.php
index 1ed4ffb..ae2cce9 100644
--- a/src/DependencyInjection/LlmChainExtension.php
+++ b/src/DependencyInjection/LlmChainExtension.php
@@ -232,7 +232,7 @@ private function processChainConfig(string $name, array $config, ContainerBuilde
         };
         $llmDefinition = new Definition($llmClass);
         if (null !== $version) {
-            $llmDefinition->setArgument('$version', $version);
+            $llmDefinition->setArgument('$name', $version);
         }
         if (0 !== count($options)) {
             $llmDefinition->setArgument('$options', $options);
@@ -440,7 +440,7 @@ private function processEmbedderConfig(int|string $name, array $config, Containe
         };
         $modelDefinition = (new Definition($modelClass));
         if (null !== $version) {
-            $modelDefinition->setArgument('$version', $version);
+            $modelDefinition->setArgument('$name', $version);
         }
         if (0 !== count($options)) {
             $modelDefinition->setArgument('$options', $options);