forked from Archives/langchain
d368c43648
Quick bug fix for semantic similarity vector injection
10 lines
358 B
Python
10 lines
358 B
Python
"""Test functionality related to prompt utils."""
|
|
from langchain.prompts.example_selector.semantic_similarity import sorted_values
|
|
|
|
|
|
def test_sorted_vals() -> None:
|
|
"""Test sorted values from dictionary."""
|
|
test_dict = {"key2": "val2", "key1": "val1"}
|
|
expected_response = ["val1", "val2"]
|
|
assert sorted_values(test_dict) == expected_response
|