Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add custom retrieval query option in similarity search #3

Closed
wants to merge 3 commits into from

Conversation

willtai
Copy link
Contributor

@willtai willtai commented Mar 8, 2024

Description

Allow users to append a custom query to the original retrieval query. This is in its own method. This is as seen in langchain and llama-index:

Also the returned type of custom_similarity_search() is Any, as the returned type can be unstructured depending on the custom query.

Also allows custom parameters to be added for the query

Tests

  • Test for happy path
  • Test for CypherSyntaxError

@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from 4eaba67 to 2066c72 Compare March 11, 2024 17:18
@willtai willtai changed the base branch from main to jonbesga/tests March 11, 2024 17:19
@willtai willtai force-pushed the jonbesga/tests branch 3 times, most recently from a262c48 to d1685dc Compare March 13, 2024 16:44
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from 7048d56 to 4a5e749 Compare March 13, 2024 16:57
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch 5 times, most recently from f0dc1d6 to 86a98b1 Compare March 18, 2024 11:28
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from 5066a2a to c34b6ff Compare March 18, 2024 11:45
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from c34b6ff to 27ecffb Compare March 18, 2024 11:57
@willtai willtai marked this pull request as ready for review March 18, 2024 11:57
Base automatically changed from jonbesga/tests to main March 19, 2024 14:28
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from 27ecffb to 259c2d6 Compare March 26, 2024 10:50
@willtai willtai requested review from jonbesga and a team March 26, 2024 12:37
willtai added 3 commits March 28, 2024 15:57
…ent's _database_query()

Changed test_client.py to have mocks at execute_query level

Changed precommit config to include ruff linting and formatting

Update pre-commit

Corrected cypher query error test and dimensions constraint in CreateIndexModel data type

ruff formatting
@willtai willtai force-pushed the willtai/add-custom-retrieval-query branch from 89bdf87 to 779960f Compare March 28, 2024 15:59
@willtai willtai closed this Apr 3, 2024
@neo4j neo4j deleted a comment from oskarhane Apr 22, 2024
@willtai willtai deleted the willtai/add-custom-retrieval-query branch May 31, 2024 16:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant