How to modify QdrantClient for custom indexing and queries in Kernel Memory?

[michelschep]

Hello everyone,

I'm looking for a way to put an index on a Qdrant collection while continuing to use QdrantClient. To achieve this, I need to add a custom property XYZ to the DefaultQdrantPayload.

Currently, the code base is somewhat generic, as in QdrantClient<T>. However, I can only use DefaultQdrantPayload as the type for T. My question is: is there a way to use a derived class of DefaultQdrantPayload where I can add the additional XYZ property?

The goal is to create an index on the XYZ property in Qdrant.

That's the first part of my issue.

In addition, I need to modify the Qdrant query generated by QdrantClient in GetSimilarListAsync. The query is built using the SearchVectorsRequest query builder, but I'd like to specify the XYZ property and its value. Currently, the builder only supports the use of tags, which are part of the inner payload of the Qdrant payload. Unfortunately, it's not possible to create an index on specific tags.

To work around this, I copied the Qdrant folder from the Kernel Memory, added the XYZ property to DefaultQdrantPayload, and introduced a HavingXYZ builder method in SearchVectorsRequest. This solution works, but it's not ideal. I'd prefer a cleaner approach, such as configuring Kernel Memory or overriding specific components, rather than maintaining my own version of the Qdrant classes.

Has anyone here encountered a similar issue or found a clean way to add index fields and use them in queries with QdrantClient? Am I missing some capability in the Kernel Memory that would allow me to achieve this?

Thanks in advance for any advice!