Remove cohere dependency and add embeddings extra to lancedb

Signed-off-by: Marcel Coetzee <[email protected]>
dlt-hub · May 16, 2024 · dafd24f · dafd24f
1 parent a86cd8d
commit dafd24f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 21 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -82,8 +82,7 @@ databricks-sql-connector = {version = ">=2.9.3,<3.0.0", optional = true}
 dbt-databricks = {version = ">=1.7.3", optional = true}
 clickhouse-driver = { version = ">=0.2.7", optional = true }
 clickhouse-connect = { version = ">=0.7.7", optional = true }
-lancedb = { version = ">=0.6.13", optional = true }
-cohere = { version = ">=3.0.0", optional = true }
+lancedb = { version = ">=0.6.13", optional = true, extras = ["emdeddings"] }
 
 [tool.poetry.extras]
 dbt = ["dbt-core", "dbt-redshift", "dbt-bigquery", "dbt-duckdb", "dbt-snowflake", "dbt-athena-community", "dbt-databricks"]
@@ -109,7 +108,7 @@ qdrant = ["qdrant-client"]
 databricks = ["databricks-sql-connector"]
 clickhouse = ["clickhouse-driver", "clickhouse-connect", "s3fs", "gcsfs", "adlfs", "pyarrow"]
 dremio = ["pyarrow"]
-lancedb = ["lancedb", "cohere"]
+lancedb = ["lancedb"]
 
 [tool.poetry.scripts]
 dlt = "dlt.cli._dlt:_main"