참고소스 수정본

참고/neo4j-graphrag-python-main/examples/customize/embeddings/custom_embeddings.py

@@ -0,0 +1,18 @@
+import random
+from typing import Any
+
+from neo4j_graphrag.embeddings import Embedder
+
+
+class CustomEmbeddings(Embedder):
+    def __init__(self, dimension: int = 10, **kwargs: Any):
+        super().__init__(**kwargs)
+        self.dimension = dimension
+
+    def embed_query(self, input: str) -> list[float]:
+        return [random.random() for _ in range(self.dimension)]
+
+
+llm = CustomEmbeddings(dimensions=1024)
+res = llm.embed_query("text")
+print(res[:10])