update new names with input_data (#204)

graphiti_core/edges.py

@@ -180,7 +180,7 @@ async def generate_embedding(self, embedder: EmbedderClient):
         start = time()
 
         text = self.fact.replace('\n', ' ')
-        self.fact_embedding = await embedder.create(input=[text])
+        self.fact_embedding = await embedder.create(input_data=[text])
 
         end = time()
         logger.debug(f'embedded {text} in {end - start} ms')

graphiti_core/embedder/client.py

@@ -29,6 +29,6 @@ class EmbedderConfig(BaseModel):
 class EmbedderClient(ABC):
     @abstractmethod
     async def create(
-        self, input: str | List[str] | Iterable[int] | Iterable[Iterable[int]]
+        self, input_data: str | List[str] | Iterable[int] | Iterable[Iterable[int]]
     ) -> list[float]:
         pass

graphiti_core/llm_client/utils.py

@@ -26,7 +26,7 @@ async def generate_embedding(embedder: EmbedderClient, text: str):
     start = time()
 
     text = text.replace('\n', ' ')
-    embedding = await embedder.create(input=[text])
+    embedding = await embedder.create(input_data=[text])
 
     end = time()
     logger.debug(f'embedded text of length {len(text)} in {end - start} ms')

graphiti_core/nodes.py

@@ -222,7 +222,7 @@ class EntityNode(Node):
     async def generate_name_embedding(self, embedder: EmbedderClient):
         start = time()
         text = self.name.replace('\n', ' ')
-        self.name_embedding = await embedder.create(input=[text])
+        self.name_embedding = await embedder.create(input_data=[text])
         end = time()
         logger.debug(f'embedded {text} in {end - start} ms')
 
@@ -334,7 +334,7 @@ async def save(self, driver: AsyncDriver):
     async def generate_name_embedding(self, embedder: EmbedderClient):
         start = time()
         text = self.name.replace('\n', ' ')
-        self.name_embedding = await embedder.create(input=[text])
+        self.name_embedding = await embedder.create(input_data=[text])
         end = time()
         logger.debug(f'embedded {text} in {end - start} ms')
 

graphiti_core/search/search.py

@@ -66,7 +66,7 @@ async def search(
     bfs_origin_node_uuids: list[str] | None = None,
 ) -> SearchResults:
     start = time()
-    query_vector = await embedder.create(input=[query.replace('\n', ' ')])
+    query_vector = await embedder.create(input_data=[query.replace('\n', ' ')])
 
     # if group_ids is empty, set it to None
     group_ids = group_ids if group_ids else None

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphiti-core"
-version = "0.3.18"
+version = "0.3.19"
 description = "A temporal graph building library"
 authors = [
     "Paul Paliychuk <[email protected]>",