Class MockIndex

test_utils/pinecone_mocks.py:11–32

class MockIndex:
    def __init__(self):
        self.kv = {}

    def upsert(self, id_vectors, *args, **kwargs):
        for id, vector in id_vectors:
            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):