Update search method to return EntityEdge objects

[danielchalef]

---

For more details, open the Copilot Workspace session.

---

🚀	This description was created by Ellipsis for commit d15b597

Summary:

The search method in graphiti_core/graphiti.py is updated to return EntityEdge objects, and corresponding test cases in tests/test_graphiti_int.py are modified to handle these objects.

Key points:

• Update graphiti_core/graphiti.py search method to return EntityEdge objects instead of strings.
• Modify tests/test_graphiti_int.py to handle EntityEdge objects by extracting fact attribute for logging.
• Ensure search method returns relevant EntityEdge objects for a given query.

---

Generated with ❤️ by ellipsis.dev

[ellipsis-dev]

👍 Looks good to me! Reviewed everything up to d15b597 in 11 seconds

More details

• Looked at 54 lines of code in 2 files
• Skipped 0 files when reviewing.
• Skipped posting 2 drafted comments based on config settings.

1. tests/test_graphiti_int.py:81

• Draft comment:
  Consider renaming 'FACTS' to 'EDGES' in the format_context function to better reflect the data being logged.
• Reason this comment was not posted:
  Confidence changes required: 50%
  The PR changes the return type of the search method from a list of facts to a list of EntityEdge objects. The test cases have been updated accordingly to extract facts from the EntityEdge objects. However, the format_context function still uses the term 'FACTS', which might be misleading now that the function deals with edges.

2. tests/test_graphiti_int.py:81

• Draft comment:
  Avoid logging sensitive data. Ensure that edge.fact does not contain sensitive information before logging.
• Reason this comment was not posted:
  Confidence changes required: 80%
  The code in graphiti_core/graphiti.py and tests/test_graphiti_int.py is mostly compliant with the rules. However, there is a potential issue with logging sensitive data in the test file.

Workflow ID: wflow_tMpMno7BNEdfHv3B

---

You can customize Ellipsis with 👍 / 👎 feedback, review rules, user-specific overrides, quiet mode, and more.