sunholo 0.114.2__py3-none-any.whl → 0.115.1__py3-none-any.whl

{sunholo-0.114.2.dist-info → sunholo-0.115.1.dist-info}/RECORD

@@ -1,5 +1,6 @@
-sunholo/__init__.py,sha256=LV-oCDt_Q01UM8N_dua8m6bD5IrrZL1mZLWKkYfjOeQ,1178
+sunholo/__init__.py,sha256=Ap2yX2ITBVt_vkloYipUM8OwW14g6aor2NX7LWp0-mI,1133
 sunholo/custom_logging.py,sha256=YfIN1oP3dOEkkYkyRBU8BGS3uJFGwUDsFCl8mIVbwvE,12225
+sunholo/types.py,sha256=xdNNb4bR7O4jY9NTNzeeiX5VYjVxCGzYWy8hUXemxr0,1764
 sunholo/agents/__init__.py,sha256=X2I3pPkGeKWjc3d0QgSpkTyqD8J8JtrEWqwrumf1MMc,391
 sunholo/agents/chat_history.py,sha256=Gph_CdlP2otYnNdR1q1Umyyyvcad2F6K3LxU5yBQ9l0,5387
 sunholo/agents/dispatch_to_qa.py,sha256=AwLS41oK6iS8xre-HuWjS4jj9dvU-evWI58EYfG65fg,8879
@@ -35,16 +36,16 @@ sunholo/chunker/doc_handling.py,sha256=t_lDazHfJbs4Q2Ruq2MvBBeJRfsjjQkzMxKuX8qQK
 sunholo/chunker/encode_metadata.py,sha256=hxxd9KU35Xi0Z_EL8kt_oD66pKfBLhEjBImC16ew-Eo,1919
 sunholo/chunker/images.py,sha256=id2PBu6XyGEOtgafq2v0c9_O6kxaC_pYFMnbsIitkSg,1868
 sunholo/chunker/loaders.py,sha256=5NXrMxV-WdbFpxeLhFzccw0_zhf1UQ7yKFFeaMkc9Bc,11105
-sunholo/chunker/message_data.py,sha256=-gnF9cKIuo-aA6jhCXrAm6U2K9tL5GbeCQ3nMJfaxUQ,10884
-sunholo/chunker/pdfs.py,sha256=njDPop751GMHi3cOwIKd2Yct-_lWR2gqcB7WykfHphs,2480
+sunholo/chunker/message_data.py,sha256=iPrYUQRjjCGoVDvnqGEudkXi1PdbOwxey6v08_-GF2g,10826
+sunholo/chunker/pdfs.py,sha256=DFZdsvAQt8qMvg7d-3dj29ULgKJZYXy4qGdORLmE790,2467
 sunholo/chunker/process_chunker_data.py,sha256=uO-YOEHIjAOy0ZMJ0vea9OMNsQBISHfhbtgoyuHiP6s,3598
-sunholo/chunker/publish.py,sha256=8TrvmX51aQPvht11myxm0G4tuI3iqAM2wL2x8MWkfX4,2990
+sunholo/chunker/publish.py,sha256=Y-4w59iVGoXCKaun9jMWVR7Cqprb9mTZGiaVk5y_zrY,2933
 sunholo/chunker/pubsub.py,sha256=48bhuAcszN7LGe3-ksPSLHHhq0uKxiXOrizck5qpcP0,1012
-sunholo/chunker/splitter.py,sha256=WltIA6LYELwG0FEtiDKclgRtKuw3rXI2myFOQM4LkIs,6826
+sunholo/chunker/splitter.py,sha256=RfekLPkjhCcNd1PFXIj_FxusJMJ8_3cyWl7bsYvtQ0g,7068
 sunholo/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sunholo/cli/chat_vac.py,sha256=sYPzUDwwwebJvIobv3GRW_xbQQ4BTy9G-WHdarGCHB0,23705
 sunholo/cli/cli.py,sha256=Bhyrs8GEtJTbsvPYufEY184ra13eusATXAnJClJ_LGY,4474
-sunholo/cli/cli_init.py,sha256=ITM-GeCRia1kXXQekzkYqYmfcaByCwr1ZWfAA3qCaVk,8507
+sunholo/cli/cli_init.py,sha256=u6BZFtUyFMOKrXZ46-DfET0IpH3Tl2PlOz386rADtrw,8549
 sunholo/cli/configs.py,sha256=QUM9DvKOdZmEQRM5uI3Nh887T0YDiSMr7O240zTLqws,4546
 sunholo/cli/deploy.py,sha256=zxdwUsRTRMC8U5vyRv0JiKBLFn84Ug_Tc88-_h9hJSs,1609
 sunholo/cli/embedder.py,sha256=v-FKiSPHaQzB6ctClclYueIf3bf3CqYtC1oRgPfT4dY,5566
@@ -54,9 +55,9 @@ sunholo/cli/sun_rich.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
 sunholo/cli/swagger.py,sha256=absYKAU-7Yd2eiVNUY-g_WLl2zJfeRUNdWQ0oH8M_HM,1564
 sunholo/cli/vertex.py,sha256=8130YCarxHL1UC3aqblNmUwGZTXbkdL4Y_FOnZJsWiI,2056
 sunholo/components/__init__.py,sha256=IDoylb74zFKo6NIS3RQqUl0PDFBGVxM1dfUmO7OJ44U,176
-sunholo/components/llm.py,sha256=8iyY6K1ZiiJx9MGL1fY5CHh8CD3YYhwEA6O8B44tkxE,13004
+sunholo/components/llm.py,sha256=xIO_UPYwlpI9Bve76pQVhPgT1DLqCCo1626uKrvwFI4,13001
 sunholo/components/retriever.py,sha256=Wmchv3huAM4w7DIS-a5Lp9Hi7M8pE6vZdxgseiT9SXk,7820
-sunholo/components/vectorstore.py,sha256=xKk7micTRwZckaI7U6PxvFz_ZSjCH48xPTDYiDcv2tc,5913
+sunholo/components/vectorstore.py,sha256=k7GS1Y5c6ZGXSDAJvyCes6dTjhDAi0fjGbVLqpyfzBc,5918
 sunholo/database/__init__.py,sha256=bpB5Nk21kwqYj-qdVnvNgXjLsbflnH4g-San7OHMqR4,283
 sunholo/database/alloydb.py,sha256=x1zUMB-EVWbE2Zvp4nAs2Z-tB_kOZmS45H2lwVHdYnk,11678
 sunholo/database/alloydb_client.py,sha256=q732tmRdSDutnUk7vRUPUPpi-yU5FK5rQko8co6yke0,19132
@@ -76,7 +77,7 @@ sunholo/discovery_engine/create_new.py,sha256=jWg5LW-QpFE8zq50ShaQJB3Wu8loiWB0P4
 sunholo/discovery_engine/discovery_engine_client.py,sha256=jfIayVUOPM4svGF1S5Kk60rIG-xSo_e3zOHtBRg0nZA,22002
 sunholo/discovery_engine/get_ai_search_chunks.py,sha256=hsFGOQugSeTMPEaQ16XTs_D45F8NABBm2IsAEdTk7kQ,4316
 sunholo/embedder/__init__.py,sha256=sI4N_CqgEVcrMDxXgxKp1FsfsB4FpjoXgPGkl4N_u4I,44
-sunholo/embedder/embed_chunk.py,sha256=sy--Gtf8x0N2KcGnPaDyS2GIVrZjxml4HZ9smlY-B5A,7037
+sunholo/embedder/embed_chunk.py,sha256=Vvvj3-H4pSb1a2sLik3-X3X459j2jrUq1dBNAsOcQLo,7156
 sunholo/excel/__init__.py,sha256=AqTMN9K4qJYi4maEgoORc5oxDVGO_eqmwzDaVP37JgY,56
 sunholo/excel/plugin.py,sha256=rl3FoECZ6Ts8KKExPrbPwr3u3CegZfsevmcjgUXAlhE,4033
 sunholo/gcs/__init__.py,sha256=SZvbsMFDko40sIRHTHppA37IijvJTae54vrhooEF5-4,90
@@ -106,10 +107,6 @@ sunholo/llamaindex/llamaindex_class.py,sha256=PnpPoc7LpP7xvKIXYu-UvI4ehj67pGhE1E
 sunholo/llamaindex/user_history.py,sha256=ZtkecWuF9ORduyGB8kF8gP66bm9DdvCI-ZiK6Kt-cSE,2265
 sunholo/lookup/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sunholo/lookup/model_lookup.yaml,sha256=O7o-jP53MLA06C8pI-ILwERShO-xf6z_258wtpZBv6A,739
-sunholo/patches/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sunholo/patches/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sunholo/patches/langchain/lancedb.py,sha256=KstVpYtI2E1s6_l_kq6js8ruIPJduO6nnvAXbqjoqCc,7507
-sunholo/patches/langchain/vertexai.py,sha256=kX1IvC2D1kMgM3SaSzP9HEosbD6CUymLJd7w9eXo3eE,17677
 sunholo/pubsub/__init__.py,sha256=DfTEk4zmCfqn6gFxRrqDO0pOrvXTDqH-medpgYO4PGw,117
 sunholo/pubsub/process_pubsub.py,sha256=rN2N4WM6PZkMKDrdT8pnEfTvsXACRyJFqIHJQCbuxLs,3088
 sunholo/pubsub/pubsub_manager.py,sha256=19w_N0LiG-wgVWvgJ13b8BUeN8ZzgSPXAhPmL1HRRSI,6966
@@ -117,14 +114,14 @@ sunholo/qna/__init__.py,sha256=F8q1uR_HreoSX0IfmKY1qoSwIgXhO2Q8kuDSxh9_-EE,28
 sunholo/qna/parsers.py,sha256=YpOaK5S_LxJ6FbliSYDc3AVOJ62RVduayoNnzi_p8CM,2494
 sunholo/qna/retry.py,sha256=yMw7RTkw-RXCzfENPJOt8c32mXlpvOR589EGkvK-6yI,2028
 sunholo/senses/__init__.py,sha256=fbWqVwwzkV5uRSb8lQzo4pn0ja_VYVWbUYapurSowBs,39
-sunholo/senses/stream_voice.py,sha256=VpCmooEKghBT1jPJe9mX7gKggGqY6qt-bpO7hwY4sPE,18122
+sunholo/senses/stream_voice.py,sha256=_r5rFZ30upLKzxPmgS0SHKxx0AZbbirn0gZfGUHVHtY,18696
 sunholo/streaming/__init__.py,sha256=MpbydI2UYo_adttPQFkxNM33b-QRyNEbrKJx0C2AGPc,241
-sunholo/streaming/content_buffer.py,sha256=pSAoajCIDm8KM4TWkR-XptI925m_hSPQuEy0ea9Uq-c,12863
+sunholo/streaming/content_buffer.py,sha256=bqPta3Q1tXI88Ngyj1kgPC-v4phhGm1nZURcuqQSGIQ,12537
 sunholo/streaming/langserve.py,sha256=hi7q8WY8DPKrALl9m_dOMxWOdE-iEuk7YW05SVDFIX8,6514
 sunholo/streaming/stream_lookup.py,sha256=hYg1DbdSE_QNJ8ZB-ynXJlWgvFjrGvwoUsGJu_E0pRQ,360
 sunholo/streaming/streaming.py,sha256=gSxLuwK-5-t5D1AjcHf838BY-L4jvdkdn_xePl-DK3o,16635
 sunholo/summarise/__init__.py,sha256=MZk3dblUMODcPb1crq4v-Z508NrFIpkSWNf9FIO8BcU,38
-sunholo/summarise/summarise.py,sha256=XYOdBrTYRSinEOvbwfGKayk-5ELdQFucNuzZ7XSmXeQ,4028
+sunholo/summarise/summarise.py,sha256=UnycBVLLEXK1HitCOG2zW3XIyxMrw47xoVf6e2OC9A0,4150
 sunholo/terraform/__init__.py,sha256=yixxEltc3n9UpZaVi05GlgS-YRq_DVGjUc37I9ajeP4,76
 sunholo/terraform/tfvars_editor.py,sha256=-TBBWbALYb5HLFYwD2s70Kp27ys6fzIyreBFOT5kqqY,13142
 sunholo/tools/__init__.py,sha256=5NuYpwwTX81qGUWvgwfItoSLXteNnp7KjgD7IPZUFjI,53
@@ -132,10 +129,10 @@ sunholo/tools/web_browser.py,sha256=8Gdf02F4zCOeSnijnfaL6jzk4oaSI0cj48o-esoWzwE,
 sunholo/utils/__init__.py,sha256=Hv02T5L2zYWvCso5hzzwm8FQogwBq0OgtUbN_7Quzqc,89
 sunholo/utils/api_key.py,sha256=Ct4bIAQZxzPEw14hP586LpVxBAVi_W9Serpy0BK-7KI,244
 sunholo/utils/big_context.py,sha256=gJIP7_ZL-YSLhOMq8jmFTMqH1wq8eB1NK7oKPeZAq2s,5578
-sunholo/utils/config.py,sha256=uXD9cCLjZzdNpH5qZ2oVEo5Zb5EDkP9KxxjtriG9LgY,9074
-sunholo/utils/config_class.py,sha256=aNhxhEHIxRvqQNq_9wFJt1yM9U1ypahCVZ3NzPWuay4,9843
+sunholo/utils/config.py,sha256=bz0ODJyqnoHQIsk4pmNpVxxq5WvwS0SfOq4cnCjQPJk,9105
+sunholo/utils/config_class.py,sha256=7xp9jKzxej2SyFmtc74nf_rYjdRhGFBtY2cV9-4JFGY,9795
 sunholo/utils/config_schema.py,sha256=Wv-ncitzljOhgbDaq9qnFqH5LCuxNv59dTGDWgd1qdk,4189
-sunholo/utils/gcp.py,sha256=uueODEpA-P6O15-t0hmcGC9dONLO_hLfzSsSoQnkUss,4854
+sunholo/utils/gcp.py,sha256=lus1HH8YhFInw6QRKwfvKZq-Lz-2KQg4ips9v1I_3zE,4783
 sunholo/utils/gcp_project.py,sha256=Fa0IhCX12bZ1ctF_PKN8PNYd7hihEUfb90kilBfUDjg,1411
 sunholo/utils/mime.py,sha256=7_J1PnWOlvAPRoHWKESAncdRVVldVwRdKvuDvi9sRfE,2020
 sunholo/utils/parsers.py,sha256=wES0fRn3GONoymRXOXt-z62HCoOiUvvFXa-MfKfjCls,6421
@@ -144,15 +141,15 @@ sunholo/utils/user_ids.py,sha256=SQd5_H7FE7vcTZp9AQuQDWBXd4FEEd7TeVMQe1H4Ny8,292
 sunholo/utils/version.py,sha256=P1QAJQdZfT2cMqdTSmXmcxrD2PssMPEGM-WI6083Fck,237
 sunholo/vertex/__init__.py,sha256=tMd7ysJ1uwBjfFSn8JL0uS3-s6h_X4GAUBz8AArZEF0,339
 sunholo/vertex/extensions_call.py,sha256=QeQbL3aAHlc4_-SynOzooZ_3xkQWAlcgNmFBSwLNtN8,13816
-sunholo/vertex/extensions_class.py,sha256=2QGW28lNjoMEnaoVb3QcqEDwphclIsZthnpLUi5_Ivo,21033
+sunholo/vertex/extensions_class.py,sha256=DqUovqIJgDhXRjCThzaVWM56u0cZHV0NRC6135Tllso,21032
 sunholo/vertex/genai_functions.py,sha256=2z6grM9H0Z79Yzx88l8mE1wXck3bRa0TWvnqZZ9ifDc,2051
 sunholo/vertex/init.py,sha256=1OQwcPBKZYBTDPdyU7IM4X4OmiXLdsNV30C-fee2scQ,2875
 sunholo/vertex/memory_tools.py,sha256=tBZxqVZ4InTmdBvLlOYwoSEWu4-kGquc-gxDwZCC4FA,7667
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
 sunholo/vertex/type_dict_to_json.py,sha256=uTzL4o9tJRao4u-gJOFcACgWGkBOtqACmb6ihvCErL8,4694
-sunholo-0.114.2.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.114.2.dist-info/METADATA,sha256=GEofig2vXtd-UPdm1fph5sd3JEGOART5NHqLkwDRRSA,9056
-sunholo-0.114.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-sunholo-0.114.2.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.114.2.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.114.2.dist-info/RECORD,,
+sunholo-0.115.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.115.1.dist-info/METADATA,sha256=FcfcUP6qrlqE6kyj6uUVWyWMfYvZ03zRyJGBkllThS4,9134
+sunholo-0.115.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+sunholo-0.115.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.115.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.115.1.dist-info/RECORD,,

sunholo/patches/langchain/lancedb.py

@@ -1,219 +0,0 @@
-# from https://github.com/langchain-ai/langchain/blob/6c18f73ca56bb72cb964aaa668c3f8ac14237619/libs/community/langchain_community/vectorstores/lancedb.py
-from __future__ import annotations
-
-import uuid, time
-from typing import Any, Iterable, List, Optional
-
-from langchain_core.documents import Document
-from langchain_core.embeddings import Embeddings
-from langchain_core.vectorstores import VectorStore
-
-
-class LanceDB(VectorStore):
-    """`LanceDB` vector store.
-
-    To use, you should have ``lancedb`` python package installed.
-
-    Example:
-        .. code-block:: python
-
-            db = lancedb.connect('./lancedb')
-            table = db.open_table('my_table')
-            vectorstore = LanceDB(table, embedding_function)
-            vectorstore.add_texts(['text1', 'text2'])
-            result = vectorstore.similarity_search('text1')
-    """
-
-    def __init__(
-        self,
-        connection: Optional[Any] = None,
-        embedding: Optional[Embeddings] = None,
-        vector_key: Optional[str] = "vector",
-        id_key: Optional[str] = "id",
-        text_key: Optional[str] = "text",
-        table_name: Optional[str] = "vectorstore",
-    ):
-        """Initialize with Lance DB connection"""
-        try:
-            import lancedb
-        except ImportError:
-            raise ImportError(
-                "Could not import lancedb python package. "
-                "Please install it with `pip install lancedb`."
-            )
-        if not isinstance(connection, lancedb.db.LanceTable):
-            raise ValueError(
-                "connection should be an instance of lancedb.db.LanceTable, ",
-                f"got {type(connection)}",
-            )
-        self.lancedb = lancedb
-        self._embedding = embedding
-        self._vector_key = vector_key
-        self._id_key = id_key
-        self._text_key = text_key
-        self._table_name = table_name
-
-        if self._embedding is None:
-            raise ValueError("embedding should be provided")
-
-        if connection is not None:
-            if not isinstance(connection, lancedb.db.LanceTable):
-                raise ValueError(
-                    "connection should be an instance of lancedb.db.LanceTable, ",
-                    f"got {type(connection)}",
-                )
-            self._connection = connection
-        else:
-            self._connection = self._init_table()
-
-    @property
-    def embeddings(self) -> Embeddings:
-        return self._embedding
-
-    def add_texts(
-        self,
-        texts: Iterable[str],
-        metadatas: Optional[List[dict]] = None,
-        ids: Optional[List[str]] = None,
-        **kwargs: Any,
-    ) -> List[str]:
-        """Turn texts into embedding and add it to the database
-
-        Args:
-            texts: Iterable of strings to add to the vectorstore.
-            metadatas: Optional list of metadatas associated with the texts.
-            ids: Optional list of ids to associate with the texts.
-
-        Returns:
-            List of ids of the added texts.
-        """
-        # Embed texts and create documents
-        docs = []
-        ids = ids or [str(uuid.uuid4()) for _ in texts]
-        embeddings = self._embedding.embed_documents(list(texts))
-        for idx, text in enumerate(texts):
-            embedding = embeddings[idx]
-            metadata = metadatas[idx] if metadatas else {}
-            docs.append(
-                {
-                    self._vector_key: embedding,
-                    self._id_key: ids[idx],
-                    self._text_key: text,
-                    **metadata,
-                }
-            )
-
-        max_retries = 5  
-        retry_delay = 1 
-        for attempt in range(max_retries):
-            try:
-                self._connection.add(docs)
-                return ids  # If success, return immediately
-            except OSError as e:
-                if "429 Too Many Requests" in str(e):
-                    if attempt < max_retries - 1:
-                        time.sleep(retry_delay)
-                        retry_delay *= 2  # Exponential backoff
-                        continue
-                    else:
-                        raise  # Re-raise the exception if max retries are reached
-                else:
-                    raise  # Re-raise the exception if it's not a rate limit error
-
-        return ids
-
-    def similarity_search(self, query: str, k: int = 4, **kwargs: Any) -> List[Document]:
-        """Return documents most similar to the query
-
-        Args:
-            query: String to query the vectorstore with.
-            k: Number of documents to return.
-
-        Returns:
-            List of documents most similar to the query.
-        """
-        query_type = kwargs.get('query_type', None)
-        where_clause = kwargs.get('where', None)
-
-        if query_type == "hybrid":
-            # Hybrid search logic - requires table created with embedding function
-            # https://lancedb.github.io/lancedb/hybrid_search/hybrid_search/
-            search_query = self._connection.search(query, query_type="hybrid")
-        elif query_type == "text":
-            # requires import tantivy to be installed
-            try:
-                import tantivy  # noqa: F401
-            except ImportError:
-                raise ImportError(
-                    "Could not import tantivy python package. "
-                    "Please install it with `pip install tantivy`."
-                )
-            search_query = self._connection.search(query)
-        else:
-            # Original search logic
-            embedding = self._embedding.embed_query(query)
-            search_query = self._connection.search(embedding, vector_column_name=self._vector_key)
-
-        if where_clause:
-            # Apply the where condition if specified
-            search_query = search_query.where(where_clause)        
-
-        docs = search_query.limit(k).to_arrow()
-        
-        columns = docs.schema.names
-        return [
-            Document(
-                page_content=docs[self._text_key][idx].as_py(),
-                metadata={
-                    col: docs[col][idx].as_py()
-                    for col in columns
-                    if col != self._text_key
-                },
-            )
-            for idx in range(len(docs))
-        ]
-
-
-    @classmethod
-    def from_texts(
-        cls,
-        texts: List[str],
-        embedding: Embeddings,
-        metadatas: Optional[List[dict]] = None,
-        connection: Any = None,
-        vector_key: Optional[str] = "vector",
-        id_key: Optional[str] = "id",
-        text_key: Optional[str] = "text",
-        **kwargs: Any,
-    ) -> LanceDB:
-        instance = LanceDB(
-            connection,
-            embedding,
-            vector_key,
-            id_key,
-            text_key,
-        )
-        instance.add_texts(texts, metadatas=metadatas, **kwargs)
-
-        return instance
-
-    def _init_table(self) -> Any:
-        import pyarrow as pa
-
-        schema = pa.schema(
-            [
-                pa.field(
-                    self._vector_key,
-                    pa.list_(
-                        pa.float32(),
-                        len(self.embeddings.embed_query("test")),  # type: ignore
-                    ),
-                ),
-                pa.field(self._id_key, pa.string()),
-                pa.field(self._text_key, pa.string()),
-            ]
-        )
-        db = self.lancedb.connect("/tmp/lancedb")
-        tbl = db.create_table(self._table_name, schema=schema, mode="overwrite")
-        return tbl
-