add a start to few shot

alnutile · Apr 27, 2023 · 35c7422 · 35c7422
1 parent 36d3166
commit 35c7422
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 17 deletions.
diff --git 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
@@ -1,14 +1,14 @@
 <?php
 
-use Sundance\LarachainPromptTemplates\FewShotPromptTemplate;
 use function PHPUnit\Framework\assertEquals;
+use Sundance\LarachainPromptTemplates\FewShotPromptTemplate;
 use Sundance\LarachainPromptTemplates\Prompts\PromptToken;
 use Sundance\LarachainPromptTemplates\PromptTemplate;
 
 it('test few show', function () {
-    $prefix = "Give the antonym of every input";
+    $prefix = 'Give the antonym of every input';
 
-    $template = <<<EOD
+    $template = <<<'EOD'
 Word: {word}
 Antonym: {antonym}
 EOD;
@@ -45,7 +45,6 @@
         $template
     );
 
-
     $input_variables = [
         new PromptToken(
             'word',
@@ -57,9 +56,7 @@
         ),
     ];
 
-
-
-    $template = <<<EOD
+    $template = <<<'EOD'
 Word: {word}
 Antonym:
 EOD;
@@ -72,7 +69,7 @@
         $suffix
     );
 
-    $expected = <<<EOD
+    $expected = <<<'EOD'
 Give the antonym of every input
 Word: happy
 Antonym: sad
@@ -82,6 +79,5 @@
 Antonym:
 EOD;
 
-    assertEquals($expected, $dto->format("big"));
+    assertEquals($expected, $dto->format('big'));
 });
-
diff --git 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.