refac: memory

This commit is contained in:
Timothy Jaeryang Baek 2025-03-28 01:32:50 -07:00
parent 2d4b4756ba
commit 37d84e6a97

View File

@ -203,7 +203,7 @@ def add_observations(req: AddObservationsRequest):
results = [] results = []
for obs in req.observations: for obs in req.observations:
name = obs.entityName name = obs.entityName.lower()
contents = obs.contents contents = obs.contents
entity = next((e for e in graph.entities if e.name == name), None) entity = next((e for e in graph.entities if e.name == name), None)
if not entity: if not entity:
@ -234,7 +234,7 @@ def delete_observations(req: DeleteObservationsRequest):
graph = read_graph_file() graph = read_graph_file()
for deletion in req.deletions: for deletion in req.deletions:
name = deletion.entityName name = deletion.entityName.lower()
to_delete = deletion.observations to_delete = deletion.observations
entity = next((e for e in graph.entities if e.name == name), None) entity = next((e for e in graph.entities if e.name == name), None)
if entity: if entity: