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
Labels
type: enhancement Indicates new feature requests

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

@frascuchon frascuchon added the type: enhancement Indicates new feature requests label Jan 10, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
type: enhancement Indicates new feature requests
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants