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

Unable to get score in Similarity search #5777

Open
Saikiranbonu1661 opened this issue Jan 8, 2025 · 0 comments · May be fixed by #5778
Open

Unable to get score in Similarity search #5777

Saikiranbonu1661 opened this issue Jan 8, 2025 · 0 comments · May be fixed by #5778

Comments

@Saikiranbonu1661
Copy link

Describe the bug

The method used to fetch vector similarity scores is returning only a list of records, rather than a list of tuples containing both the records and their corresponding similarity scores.

Stacktrace and code to create the bug

client = rg.Argilla(api_url, api_key)
dataset = client.datasets(name="datatset_name", workspace="workspace_name")
similar = rg.Similar(name="vector", value=vector)
records = dataset.records(query=rg.Query(similar=similar), with_vectors=True)

Iterate over the above records or just try to convert into to_list, to_json or to_dict methods, which doesnot return similarity score.

Expected behavior

The method should return a list of tuples, where each tuple contains a record and its corresponding similarity score when fetching vector similarity.

Environment

  • Argilla Version [e.g. 1.0.0]: 2.6.0
  • ElasticSearch Version [e.g. 7.10.2]: 8.15.1
  • Docker Image (optional) [e.g. argilla:v1.0.0]:

Additional context

No response

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 a pull request may close this issue.

1 participant