diff --git a/src/FewShotPromptTemplate.php b/src/FewShotPromptTemplate.php index dcd401e..9961add 100644 --- a/src/FewShotPromptTemplate.php +++ b/src/FewShotPromptTemplate.php @@ -4,24 +4,23 @@ class FewShotPromptTemplate extends \Spatie\LaravelData\Data { - /** - * @param PromptTemplate[] $promptTemplates + * @param PromptTemplate[] $promptTemplates */ public function __construct( public array $promptTemplates, public ?string $prefix, public PromptTemplate $suffix, - ) - { + ) { } - public function format(string $input) { + public function format(string $input) + { $results = []; $results[] = $this->prefix; - foreach($this->promptTemplates as $index => $promptTemplate) { + foreach ($this->promptTemplates as $index => $promptTemplate) { $results[] = $promptTemplate->format(); } diff --git a/tests/FewShotPromptTemplateTest.php b/tests/FewShotPromptTemplateTest.php index 90cad93..e12cc88 100644 --- a/tests/FewShotPromptTemplateTest.php +++ b/tests/FewShotPromptTemplateTest.php @@ -1,14 +1,14 @@ format("big")); + assertEquals($expected, $dto->format('big')); }); - diff --git a/tests/PromptTemplateTest.php b/tests/PromptTemplateTest.php index 5adaa2a..7399ca5 100644 --- a/tests/PromptTemplateTest.php +++ b/tests/PromptTemplateTest.php @@ -55,7 +55,6 @@ What is a good name for a company that makes colorful socks, colorful hats and colorful headbands?', $dto->format()); }); - it('test no prompt', function () { $template = <<<'EOD' I want you to act as a naming consultant for new companies.