fixed embedoc call

Signed-off-by: sharanshirodkar7 <[email protected]>

comps/embeddings/predictionguard/embedding_predictionguard.py

@@ -8,7 +8,7 @@
 from predictionguard import PredictionGuard
 
 from comps import (
-    EmbedDoc512,
+    EmbedDoc,
     ServiceType,
     TextDoc,
     opea_microservices,
@@ -28,15 +28,15 @@
     host="0.0.0.0",
     port=6000,
     input_datatype=TextDoc,
-    output_datatype=EmbedDoc512,
+    output_datatype=EmbedDoc,
 )
 @register_statistics(names=["opea_service@embedding_predictionguard"])
-def embedding(input: TextDoc) -> EmbedDoc512:
+def embedding(input: TextDoc) -> EmbedDoc:
     start = time.time()
     response = client.embeddings.create(model=pg_embedding_model_name, input=[{"text": input.text}])
     embed_vector = response["data"][0]["embedding"]
     embed_vector = embed_vector[:512]  # Keep only the first 512 elements
-    res = EmbedDoc512(text=input.text, embedding=embed_vector)
+    res = EmbedDoc(text=input.text, embedding=embed_vector)
     statistics_dict["opea_service@embedding_predictionguard"].append_latency(time.time() - start, None)
     return res