remove redundant endpoint (#383)

This commit is contained in:
Preston Rasmussen 2025-04-21 10:52:10 -04:00 committed by GitHub
parent 3bc8596657
commit c36652eb75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -265,31 +265,6 @@ class EpisodicNode(Node):
return episodes
@classmethod
async def get_by_entity_node_uuid(cls, driver: AsyncDriver, entity_node_uuid: str):
records, _, _ = await driver.execute_query(
"""
MATCH (e:Episodic)-[r:MENTIONS]->(n:Entity {uuid: $entity_node_uuid})
RETURN DISTINCT
e.content AS content,
e.created_at AS created_at,
e.valid_at AS valid_at,
e.uuid AS uuid,
e.name AS name,
e.group_id AS group_id,
e.source_description AS source_description,
e.source AS source,
e.entity_edges AS entity_edges
""",
entity_node_uuid=entity_node_uuid,
database_=DEFAULT_DATABASE,
routing_='r',
)
episodes = [get_episodic_node_from_record(record) for record in records]
return episodes
class EntityNode(Node):
name_embedding: list[float] | None = Field(default=None, description='embedding of the name')
@ -337,8 +312,8 @@ class EntityNode(Node):
async def get_by_uuid(cls, driver: AsyncDriver, uuid: str):
query = (
"""
MATCH (n:Entity {uuid: $uuid})
"""
MATCH (n:Entity {uuid: $uuid})
"""
+ ENTITY_NODE_RETURN
)
records, _, _ = await driver.execute_query(