After trying to update the allowed_comparators in my query_constructor I still get a ValueError: Received disallowed comparator Comparator.LIKE for Chroma RAG

[nebo333]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

## Modify the query parser

from langchain.chains.query_constructor.base import (
    StructuredQueryOutputParser,
    get_query_constructor_prompt,
    load_query_constructor_runnable
)

prompt = get_query_constructor_prompt(
    document_content_description,
    metadata_field_info,
)


chain = load_query_constructor_runnable(
    ChatOpenAI(model="gpt-4o-mini", temperature=0),
    document_content_description, 
    metadata_field_info
)

from langchain.chains.query_constructor.base import (
    StructuredQueryOutputParser,
    get_query_constructor_prompt,
)

# Define allowed comparators list
allowed_comparators = [
    "$eq",  # Equal to (number, string, boolean)
    "$ne",  # Not equal to (number, string, boolean)
    "$gt",  # Greater than (number)
    "$gte",  # Greater than or equal to (number)
    "$lt",  # Less than (number)
    "$lte",  # Less than or equal to (number)
    "$like",  # Like (string)
]

constructor_prompt = get_query_constructor_prompt(
    document_contents=document_content_description,
    attribute_info= metadata_field_info,
    allowed_comparators=allowed_comparators,
)


query_model = ChatOpenAI(
    model='gpt-4o-mini',
    temperature=0,
    streaming=True,
)

output_parser = StructuredQueryOutputParser.from_components()
query_constructor = constructor_prompt | query_model | output_parser

from langchain.retrievers.self_query.chroma import ChromaTranslator

retriever = SelfQueryRetriever(
    query_constructor=query_constructor,
    llm=llm_openai,
    vectorstore=chroma_vectorstore_saved,
    structured_query_translator=ChromaTranslator()

Description

I am trying to update the allowed_comparators to include "Like" by doing the following:

allowed_comparators = [
"$eq", # Equal to (number, string, boolean)
"$ne", # Not equal to (number, string, boolean)
"$gt", # Greater than (number)
"$gte", # Greater than or equal to (number)
"$lt", # Less than (number)
"$lte", # Less than or equal to (number)
"$like", # Like (string)
]

constructor_prompt = get_query_constructor_prompt(
document_contents=document_content_description,
attribute_info= metadata_field_info,
allowed_comparators=allowed_comparators,
)

output_parser = StructuredQueryOutputParser.from_components()
query_constructor = constructor_prompt | query_model | output_parser

However I am still getting the following error: ValueError: Received disallowed comparator Comparator.LIKE. Allowed comparators are [<Comparator.EQ: 'eq'>, <Comparator.NE: 'ne'>, <Comparator.GT: 'gt'>, <Comparator.GTE: 'gte'>, <Comparator.LT: 'lt'>, <Comparator.LTE: 'lte'>]

System Info

mac os, python 3.12

[ChellShort]

I have the same problem here, I defined a list of allowed comparators but the get_query_constructor_prompt does not seem to follow those rules