9
9
use PhpLlm \LlmChain \Document \TextDocument ;
10
10
use PhpLlm \LlmChain \Document \Vector ;
11
11
use PhpLlm \LlmChain \Document \VectorDocument ;
12
- use PhpLlm \LlmChain \EmbeddingProvider ;
12
+ use PhpLlm \LlmChain \Embedder ;
13
13
use PhpLlm \LlmChain \Model \Message \ToolCallMessage ;
14
14
use PhpLlm \LlmChain \Model \Response \AsyncResponse ;
15
15
use PhpLlm \LlmChain \Model \Response \ToolCall ;
26
26
use Symfony \Component \Clock \MockClock ;
27
27
use Symfony \Component \Uid \Uuid ;
28
28
29
- #[CoversClass(EmbeddingProvider ::class)]
29
+ #[CoversClass(Embedder ::class)]
30
30
#[Medium]
31
31
#[UsesClass(TextDocument::class)]
32
32
#[UsesClass(Vector::class)]
37
37
#[UsesClass(Platform::class)]
38
38
#[UsesClass(AsyncResponse::class)]
39
39
#[UsesClass(VectorResponse::class)]
40
- final class EmbeddingProviderTest extends TestCase
40
+ final class EmbedderTest extends TestCase
41
41
{
42
42
#[Test]
43
43
public function embedSingleDocument (): void
44
44
{
45
45
$ document = new TextDocument ($ id = Uuid::v4 (), 'Test content ' );
46
46
$ vector = new Vector ([0.1 , 0.2 , 0.3 ]);
47
47
48
- $ embeddingProvider = new EmbeddingProvider (
48
+ $ embedder = new Embedder (
49
49
PlatformTestHandler::createPlatform (new VectorResponse ($ vector )),
50
50
new Embeddings (),
51
51
$ store = new TestStore (),
52
52
new MockClock (),
53
53
);
54
54
55
- $ embeddingProvider ->embed ($ document );
55
+ $ embedder ->embed ($ document );
56
56
57
57
self ::assertCount (1 , $ store ->documents );
58
58
self ::assertInstanceOf (VectorDocument::class, $ store ->documents [0 ]);
@@ -66,15 +66,15 @@ public function embedEmptyDocumentList(): void
66
66
$ logger = $ this ->createMock (LoggerInterface::class);
67
67
$ logger ->expects ($ this ->once ())->method ('debug ' )->with ('No documents to embed ' );
68
68
69
- $ embeddingProvider = new EmbeddingProvider (
69
+ $ embedder = new Embedder (
70
70
PlatformTestHandler::createPlatform (),
71
71
new Embeddings (),
72
72
$ store = new TestStore (),
73
73
new MockClock (),
74
74
$ logger ,
75
75
);
76
76
77
- $ embeddingProvider ->embed ([]);
77
+ $ embedder ->embed ([]);
78
78
79
79
self ::assertSame ([], $ store ->documents );
80
80
}
@@ -86,14 +86,14 @@ public function embedDocumentWithMetadata(): void
86
86
$ document = new TextDocument ($ id = Uuid::v4 (), 'Test content ' , $ metadata );
87
87
$ vector = new Vector ([0.1 , 0.2 , 0.3 ]);
88
88
89
- $ embeddingProvider = new EmbeddingProvider (
89
+ $ embedder = new Embedder (
90
90
PlatformTestHandler::createPlatform (new VectorResponse ($ vector )),
91
91
new Embeddings (),
92
92
$ store = new TestStore (),
93
93
new MockClock (),
94
94
);
95
95
96
- $ embeddingProvider ->embed ($ document );
96
+ $ embedder ->embed ($ document );
97
97
98
98
self ::assertSame (1 , $ store ->addCalls );
99
99
self ::assertCount (1 , $ store ->documents );
@@ -112,14 +112,14 @@ public function embedWithSleep(): void
112
112
$ document1 = new TextDocument (Uuid::v4 (), 'Test content 1 ' );
113
113
$ document2 = new TextDocument (Uuid::v4 (), 'Test content 2 ' );
114
114
115
- $ embeddingProvider = new EmbeddingProvider (
115
+ $ embedder = new Embedder (
116
116
PlatformTestHandler::createPlatform (new VectorResponse ($ vector1 , $ vector2 )),
117
117
new Embeddings (),
118
118
$ store = new TestStore (),
119
119
$ clock = new MockClock ('2024-01-01 00:00:00 ' ),
120
120
);
121
121
122
- $ embeddingProvider ->embed (
122
+ $ embedder ->embed (
123
123
documents: [$ document1 , $ document2 ],
124
124
sleep: 3
125
125
);
0 commit comments