Method query

test_utils/pinecone_mocks.py:19–32

(self, vector, *args, **kwargs)

            self.kv[id] = vector

    def query(self, vector, *args, **kwargs):
        for id, v in self.kv.items():
            if vector == v:
                return {
                    "matches": [
                        {
                            "id": id,
                            "score": 0.9,
                        }
                    ],
                    "namespace": "",
                }
        else:
            return {}


class DummyEncoder(BaseLanguageModel):