diff --git a/graphiti_core/edges.py b/graphiti_core/edges.py index a61880ca..8f657a1f 100644 --- a/graphiti_core/edges.py +++ b/graphiti_core/edges.py @@ -180,7 +180,7 @@ class EntityEdge(Edge): 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') diff --git a/graphiti_core/embedder/client.py b/graphiti_core/embedder/client.py index 950298e4..e120e203 100644 --- a/graphiti_core/embedder/client.py +++ b/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 diff --git a/graphiti_core/llm_client/utils.py b/graphiti_core/llm_client/utils.py index 2e367bad..e7639ae1 100644 --- a/graphiti_core/llm_client/utils.py +++ b/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') diff --git a/graphiti_core/nodes.py b/graphiti_core/nodes.py index 439aec80..01f79be2 100644 --- a/graphiti_core/nodes.py +++ b/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 @@ class CommunityNode(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') diff --git a/graphiti_core/search/search.py b/graphiti_core/search/search.py index a79c2354..47c3abb6 100644 --- a/graphiti_core/search/search.py +++ b/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 diff --git a/pyproject.toml b/pyproject.toml index c0c55640..3ef8eb54 100644 --- a/pyproject.toml +++ b/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 ",