Skip to content

Commit 01e5d8e

Browse files
committed
feat: make runtime config on model optional with single runtime
1 parent 770bc12 commit 01e5d8e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/DependencyInjection/LlmChainExtension.php

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,18 +85,20 @@ private function processRuntimeConfig(string $name, array $runtime, ContainerBui
8585

8686
private function processLlmConfig(string $name, array $llm, ContainerBuilder $container): void
8787
{
88+
$runtime = isset($llm['runtime']) ? 'llm_chain.runtime.'.$llm['runtime'] : Runtime::class;
89+
8890
$definition = new ChildDefinition(Gpt::class);
89-
$definition
90-
->replaceArgument('$runtime', new Reference('llm_chain.runtime.'.$llm['runtime']));
91+
$definition->replaceArgument('$runtime', new Reference($runtime));
9192

9293
$container->setDefinition('llm_chain.llm.'.$name, $definition);
9394
}
9495

9596
private function processEmbeddingsConfig(string $name, mixed $embeddings, ContainerBuilder $container): void
9697
{
98+
$runtime = isset($embeddings['runtime']) ? 'llm_chain.runtime.'.$embeddings['runtime'] : Runtime::class;
99+
97100
$definition = new ChildDefinition(Embeddings::class);
98-
$definition
99-
->replaceArgument('$runtime', new Reference('llm_chain.runtime.'.$embeddings['runtime']));
101+
$definition->replaceArgument('$runtime', new Reference($runtime));
100102

101103
$container->setDefinition('llm_chain.embeddings.'.$name, $definition);
102104
}

0 commit comments

Comments
 (0)