datawhalechina/hugging-llm

相似匹配代码 print(res) 没数据

josangmi opened this issue · 1 comments

params_dict = {"query_embedding": np.array(embed_query).astype(dtype=np.float32).tobytes()}

print(params_dict)

k = 3

base_query = f"* => [KNN {k} @Embedding $query_embedding AS similarity]"
return_fields = ["question", "answer", "similarity"]
query = (
Query(base_query)
.return_fields(*return_fields)
.sort_by("similarity")
.paging(0, k)
.dialect(2)
)

res = index.search(query, params_dict)

print(res) 这里的res是0条数据. 用的是教程一摸一样的代码和数据。 这是为啥啊。

用这个 print(r.hget(f"{INDEX_NAME}-{v.Index}", 'embedding')) 也能打印出来数据,说明redis里面也有数据才对。

很抱歉,补充一下,我上面的代码使用azure的接口,我替换成Openai的接口就没问题了。