INTPYTHON-445 Enable integration tests (#24)

Author: blink1073

.github/workflows/_release.yml

@@ -167,7 +167,8 @@ jobs:
 
       - name: Run integration tests
         env:
-          MONGODB_ATLAS_URI: ${{ secrets.MONGODB_ATLAS_URI }}
+          MONGODB_URI: ${{ secrets.MONGODB_ATLAS_URI }}
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
         run: make integration_tests
         working-directory: ${{ inputs.working-directory }}
 

.github/workflows/_test.yml

@@ -46,6 +46,14 @@ jobs:
         run: |
           make test
 
+      - name: Run integration tests
+        env:
+          MONGODB_URI: ${{ secrets.MONGODB_ATLAS_URI }}
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        run: |
+          [[ "${{ github.repository_owner }}" == "langchain-ai" ]] && make integration_tests || true
+        working-directory: ${{ inputs.working-directory }}
+
       - name: Ensure the tests did not create any additional files
         shell: bash
         run: |

libs/mongodb/tests/integration_tests/test_cache.py

@@ -13,7 +13,7 @@
 
 from ..utils import ConsistentFakeEmbeddings, FakeChatModel, FakeLLM
 
-CONN_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONN_STRING = os.environ.get("MONGODB_URI")
 INDEX_NAME = "langchain-test-index-semantic-cache"
 DATABASE = "langchain_test_db"
 COLLECTION = "langchain_test_cache"

libs/mongodb/tests/integration_tests/test_chain_example.py

@@ -16,7 +16,7 @@
 
 from ..utils import PatchedMongoDBAtlasVectorSearch
 
-CONNECTION_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONNECTION_STRING = os.environ.get("MONGODB_URI")
 DB_NAME = "langchain_test_db"
 COLLECTION_NAME = "langchain_test_chain_example"
 INDEX_NAME = "langchain-test-chain-example-vector-index"
@@ -61,7 +61,7 @@ def test_chain(
     Follows example in the docs: https://python.langchain.com/docs/how_to/hybrid/
 
     Requires OpenAI_API_KEY for embedding and chat model.
-    Requires INDEX_NAME to have been set up on MONGODB_ATLAS_URI
+    Requires INDEX_NAME to have been set up on MONGODB_URI
     """
 
     from langchain_openai import ChatOpenAI, OpenAIEmbeddings

libs/mongodb/tests/integration_tests/test_chat_message_histories.py

@@ -10,7 +10,7 @@
 COLLECTION = "langchain_test_chat"
 
 # Replace these with your mongodb connection string
-connection_string = os.environ.get("MONGODB_ATLAS_URI", "")
+connection_string = os.environ.get("MONGODB_URI", "")
 
 
 def test_memory_with_message_store() -> None:

libs/mongodb/tests/integration_tests/test_docstore.py

@@ -6,7 +6,7 @@
 
 from langchain_mongodb.docstores import MongoDBDocStore
 
-CONNECTION_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONNECTION_STRING = os.environ.get("MONGODB_URI")
 DB_NAME = "langchain_test_db"
 COLLECTION_NAME = "langchain_test_docstore"
 

libs/mongodb/tests/integration_tests/test_index.py

@@ -20,7 +20,7 @@
 @pytest.fixture
 def collection() -> Generator:
     """Depending on uri, this could point to any type of cluster."""
-    uri = os.environ.get("MONGODB_ATLAS_URI")
+    uri = os.environ.get("MONGODB_URI")
     client: MongoClient = MongoClient(uri)
     client[DB_NAME].create_collection(COLLECTION_NAME)
     clxn = client[DB_NAME][COLLECTION_NAME]

libs/mongodb/tests/integration_tests/test_mmr.py

@@ -11,7 +11,7 @@
 
 from ..utils import ConsistentFakeEmbeddings, PatchedMongoDBAtlasVectorSearch
 
-CONNECTION_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONNECTION_STRING = os.environ.get("MONGODB_URI")
 DB_NAME = "langchain_test_db"
 COLLECTION_NAME = "langchain_test_vectorstores"
 INDEX_NAME = "langchain-test-index-vectorstores"

libs/mongodb/tests/integration_tests/test_parent_document.py

@@ -17,7 +17,7 @@
 
 from ..utils import ConsistentFakeEmbeddings, PatchedMongoDBAtlasVectorSearch
 
-CONNECTION_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONNECTION_STRING = os.environ.get("MONGODB_URI")
 DB_NAME = "langchain_test_db"
 COLLECTION_NAME = "langchain_test_parent_document_combined"
 VECTOR_INDEX_NAME = "langchain-test-parent-document-vector-index"

libs/mongodb/tests/integration_tests/test_retrievers.py

@@ -20,7 +20,7 @@
 
 from ..utils import PatchedMongoDBAtlasVectorSearch
 
-CONNECTION_STRING = os.environ.get("MONGODB_ATLAS_URI")
+CONNECTION_STRING = os.environ.get("MONGODB_URI")
 DB_NAME = "langchain_test_db"
 COLLECTION_NAME = "langchain_test_retrievers"
 VECTOR_INDEX_NAME = "vector_index"