99use PhpLlm \LlmChain \Document \TextDocument ;
1010use PhpLlm \LlmChain \Document \Vector ;
1111use PhpLlm \LlmChain \Document \VectorDocument ;
12- use PhpLlm \LlmChain \EmbeddingProvider ;
12+ use PhpLlm \LlmChain \Embedder ;
1313use PhpLlm \LlmChain \Model \Message \ToolCallMessage ;
1414use PhpLlm \LlmChain \Model \Response \AsyncResponse ;
1515use PhpLlm \LlmChain \Model \Response \ToolCall ;
2626use Symfony \Component \Clock \MockClock ;
2727use Symfony \Component \Uid \Uuid ;
2828
29- #[CoversClass(EmbeddingProvider ::class)]
29+ #[CoversClass(Embedder ::class)]
3030#[Medium]
3131#[UsesClass(TextDocument::class)]
3232#[UsesClass(Vector::class)]
3737#[UsesClass(Platform::class)]
3838#[UsesClass(AsyncResponse::class)]
3939#[UsesClass(VectorResponse::class)]
40- final class EmbeddingProviderTest extends TestCase
40+ final class EmbedderTest extends TestCase
4141{
4242 #[Test]
4343 public function embedSingleDocument (): void
4444 {
4545 $ document = new TextDocument ($ id = Uuid::v4 (), 'Test content ' );
4646 $ vector = new Vector ([0.1 , 0.2 , 0.3 ]);
4747
48- $ embeddingProvider = new EmbeddingProvider (
48+ $ embedder = new Embedder (
4949 PlatformTestHandler::createPlatform (new VectorResponse ($ vector )),
5050 new Embeddings (),
5151 $ store = new TestStore (),
5252 new MockClock (),
5353 );
5454
55- $ embeddingProvider ->embed ($ document );
55+ $ embedder ->embed ($ document );
5656
5757 self ::assertCount (1 , $ store ->documents );
5858 self ::assertInstanceOf (VectorDocument::class, $ store ->documents [0 ]);
@@ -66,15 +66,15 @@ public function embedEmptyDocumentList(): void
6666 $ logger = $ this ->createMock (LoggerInterface::class);
6767 $ logger ->expects ($ this ->once ())->method ('debug ' )->with ('No documents to embed ' );
6868
69- $ embeddingProvider = new EmbeddingProvider (
69+ $ embedder = new Embedder (
7070 PlatformTestHandler::createPlatform (),
7171 new Embeddings (),
7272 $ store = new TestStore (),
7373 new MockClock (),
7474 $ logger ,
7575 );
7676
77- $ embeddingProvider ->embed ([]);
77+ $ embedder ->embed ([]);
7878
7979 self ::assertSame ([], $ store ->documents );
8080 }
@@ -86,14 +86,14 @@ public function embedDocumentWithMetadata(): void
8686 $ document = new TextDocument ($ id = Uuid::v4 (), 'Test content ' , $ metadata );
8787 $ vector = new Vector ([0.1 , 0.2 , 0.3 ]);
8888
89- $ embeddingProvider = new EmbeddingProvider (
89+ $ embedder = new Embedder (
9090 PlatformTestHandler::createPlatform (new VectorResponse ($ vector )),
9191 new Embeddings (),
9292 $ store = new TestStore (),
9393 new MockClock (),
9494 );
9595
96- $ embeddingProvider ->embed ($ document );
96+ $ embedder ->embed ($ document );
9797
9898 self ::assertSame (1 , $ store ->addCalls );
9999 self ::assertCount (1 , $ store ->documents );
@@ -112,14 +112,14 @@ public function embedWithSleep(): void
112112 $ document1 = new TextDocument (Uuid::v4 (), 'Test content 1 ' );
113113 $ document2 = new TextDocument (Uuid::v4 (), 'Test content 2 ' );
114114
115- $ embeddingProvider = new EmbeddingProvider (
115+ $ embedder = new Embedder (
116116 PlatformTestHandler::createPlatform (new VectorResponse ($ vector1 , $ vector2 )),
117117 new Embeddings (),
118118 $ store = new TestStore (),
119119 $ clock = new MockClock ('2024-01-01 00:00:00 ' ),
120120 );
121121
122- $ embeddingProvider ->embed (
122+ $ embedder ->embed (
123123 documents: [$ document1 , $ document2 ],
124124 sleep: 3
125125 );
0 commit comments