updated cohere, removed langchain, updated tests

ContextData · Aug 5, 2024 · 8dbcf45 · 8dbcf45
1 parent 7cf7eed
commit 8dbcf45
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
 boto3
 botocore
+cohere
 cffi
 openai
 psycopg2-binary
-langchain
 pinecone-client
 requests
 tiktoken
@@ -16,7 +16,6 @@ qdrant-client
 singlestoredb
 weaviate-client
 azure-storage-blob
-langchain-community
 google-cloud-storage
 snowflake-connector-python
 stripe

diff --git a/setup.py b/setup.py
@@ -27,10 +27,10 @@
     install_requires=[
         "boto3",
         "botocore",
+        "cohere"
         "cffi",
         "openai",
         "psycopg2-binary",
-        "langchain",
         "pinecone-client",
         "requests",
         "tiktoken",
@@ -43,7 +43,6 @@
         "singlestoredb",
         "weaviate-client",
         "azure-storage-blob",
-        "langchain-community",
         "google-cloud-storage",
         "snowflake-connector-python",
         "stripe",

diff --git a/tests/test_source_mods.py b/tests/test_source_mods.py
@@ -2,8 +2,8 @@
 from unittest.mock import Mock, patch
 import pandas as pd
 from io import BytesIO
-from vector_etl.source_mods.s3 import S3Source
-from vector_etl.source_mods.database import DatabaseSource
+from vector_etl.source_mods.s3_loader import S3Source
+from vector_etl.source_mods.database_loader import DatabaseSource
 from vector_etl.source_mods.local_file import LocalFileSource
 
 @pytest.fixture