From 7c693b8acf51226f8de0d75e4b5e8c4a3f18de19 Mon Sep 17 00:00:00 2001
From: Marcel Coetzee <marcel@mooncoon.com>
Date: Sun, 16 Jun 2024 00:10:50 +0200
Subject: [PATCH] Packaging

Signed-off-by: Marcel Coetzee <marcel@mooncoon.com>
---
 poetry.lock    | 20 ++++++++++++++++++--
 pyproject.toml |  3 ++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index bdb7f68503..3beeb74ac6 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1883,6 +1883,22 @@ lz4 = ["clickhouse-cityhash (>=1.0.2.1)", "lz4", "lz4 (<=3.0.1)"]
 numpy = ["numpy (>=1.12.0)", "pandas (>=0.24.0)"]
 zstd = ["clickhouse-cityhash (>=1.0.2.1)", "zstd"]
 
+[[package]]
+name = "cohere"
+version = "5.1.5"
+description = ""
+optional = true
+python-versions = "<4.0,>=3.8"
+files = [
+    {file = "cohere-5.1.5-py3-none-any.whl", hash = "sha256:10a2bc0aab8b3a03d5da84412a6309aeb2b267d852b969d8b6acd0a19176b5a9"},
+    {file = "cohere-5.1.5.tar.gz", hash = "sha256:2ed04ac3fb1b4e3a1e243cc94898158b3be41cd64165dd0910952d73207298fa"},
+]
+
+[package.dependencies]
+httpx = ">=0.21.2"
+pydantic = ">=1.9.2"
+typing_extensions = ">=4.0.0"
+
 [[package]]
 name = "colorama"
 version = "0.4.6"
@@ -9484,7 +9500,7 @@ duckdb = ["duckdb"]
 filesystem = ["botocore", "s3fs"]
 gcp = ["gcsfs", "google-cloud-bigquery", "grpcio"]
 gs = ["gcsfs"]
-lancedb = ["lancedb", "pyarrow"]
+lancedb = ["cohere", "lancedb", "pyarrow"]
 motherduck = ["duckdb", "pyarrow"]
 mssql = ["pyodbc"]
 parquet = ["pyarrow"]
@@ -9499,4 +9515,4 @@ weaviate = ["weaviate-client"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.8.1,<3.13"
-content-hash = "ccb23824d108abb5a2ccbae39864507f15e289c4a82b37db60c11aa2fa7d6489"
+content-hash = "80596bc72fb60fcd895511d366571f7c48f30ab3296dce7da78dd1e9abf774e7"
diff --git a/pyproject.toml b/pyproject.toml
index 4beb38a86e..f073ac95ee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -79,6 +79,7 @@ databricks-sql-connector = {version = ">=3", optional = true}
 clickhouse-driver = { version = ">=0.2.7", optional = true }
 clickhouse-connect = { version = ">=0.7.7", optional = true }
 lancedb = { version = ">=0.8.2", optional = true, markers = "python_version >= '3.9'", extras = ["emdeddings"] }
+cohere = { version = ">=5.0.0", optional = true }
 deltalake = { version = ">=0.17.4", optional = true }
 
 [tool.poetry.extras]
@@ -104,7 +105,7 @@ qdrant = ["qdrant-client"]
 databricks = ["databricks-sql-connector"]
 clickhouse = ["clickhouse-driver", "clickhouse-connect", "s3fs", "gcsfs", "adlfs", "pyarrow"]
 dremio = ["pyarrow"]
-lancedb = ["lancedb", "pyarrow"]
+lancedb = ["lancedb", "pyarrow", "cohere"]
 deltalake = ["deltalake", "pyarrow"]
 
 [tool.poetry.scripts]