remove redundant endpoint (#383)
This commit is contained in:
parent
3bc8596657
commit
c36652eb75
1 changed files with 2 additions and 27 deletions
|
|
@ -265,31 +265,6 @@ class EpisodicNode(Node):
|
||||||
|
|
||||||
return episodes
|
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):
|
class EntityNode(Node):
|
||||||
name_embedding: list[float] | None = Field(default=None, description='embedding of the name')
|
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):
|
async def get_by_uuid(cls, driver: AsyncDriver, uuid: str):
|
||||||
query = (
|
query = (
|
||||||
"""
|
"""
|
||||||
MATCH (n:Entity {uuid: $uuid})
|
MATCH (n:Entity {uuid: $uuid})
|
||||||
"""
|
"""
|
||||||
+ ENTITY_NODE_RETURN
|
+ ENTITY_NODE_RETURN
|
||||||
)
|
)
|
||||||
records, _, _ = await driver.execute_query(
|
records, _, _ = await driver.execute_query(
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue