Base Service (trustgraph-base/trustgraph/base/embeddings_service.py):

- Changed on_request to use request.texts

FastEmbed Processor
(trustgraph-flow/trustgraph/embeddings/fastembed/processor.py):
- on_embeddings(texts, model=None) now processes full batch efficiently
- Returns [[v.tolist()] for v in vecs] - list of vector sets

Ollama Processor (trustgraph-flow/trustgraph/embeddings/ollama/processor.py):
- on_embeddings(texts, model=None) passes list directly to Ollama
- Returns [[embedding] for embedding in embeds.embeddings]

EmbeddingsClient (trustgraph-base/trustgraph/base/embeddings_client.py):
- embed(texts, timeout=300) accepts list of texts

Tests Updated:
- test_fastembed_dynamic_model.py - 4 tests updated for new interface
- test_ollama_dynamic_model.py - 4 tests updated for new interface

trustgraph-base/trustgraph/base/embeddings_client.py

@@ -3,11 +3,11 @@ from . request_response_spec import RequestResponse, RequestResponseSpec
 from .. schema import EmbeddingsRequest, EmbeddingsResponse
 
 class EmbeddingsClient(RequestResponse):
-    async def embed(self, text, timeout=30):
+    async def embed(self, texts, timeout=300):
 
         resp = await self.request(
             EmbeddingsRequest(
-                text = text
+                texts = texts
             ),
             timeout=timeout
         )

trustgraph-base/trustgraph/base/embeddings_service.py

@@ -65,7 +65,7 @@ class EmbeddingsService(FlowProcessor):
 
             # Pass model from request if specified (non-empty), otherwise use default
             model = flow("model")
-            vectors = await self.on_embeddings(request.text, model=model)
+            vectors = await self.on_embeddings(request.texts, model=model)
 
             await flow("response").send(
                 EmbeddingsResponse(
@@ -94,7 +94,7 @@ class EmbeddingsService(FlowProcessor):
                         type = "embeddings-error",
                         message = str(e),
                     ),
-                    vectors=None,
+                    vectors=[],
                 ),
                 properties={"id": id}
             )

trustgraph-base/trustgraph/schema/services/llm.py

@@ -29,12 +29,12 @@ class TextCompletionResponse:
 
 @dataclass
 class EmbeddingsRequest:
-    text: str = ""
+    texts: list[str] = field(default_factory=list)
 
 @dataclass
 class EmbeddingsResponse:
     error: Error | None = None
-    vectors: list[list[float]] = field(default_factory=list)
+    vectors: list[list[list[float]]] = field(default_factory=list)
 
 ############################################################################