Update langchain-community version to 0.3.21 in requirements.txt and clean up unused imports in test.py

requirements.txt

@@ -84,7 +84,7 @@ jupyterlab_server==2.27.2
 keras==3.4.1
 kiwisolver==1.4.5
 langchain==0.2.14
-langchain-community==0.2.12
+langchain-community==0.3.21
 langchain-core==0.2.38
 langchain-text-splitters==0.2.2
 langchain-openai==0.1.23

test.py

@@ -3,11 +3,6 @@ import os
 import logging
 
 from langchain_astradb import AstraDBVectorStore
-from langchain_openai import AzureOpenAIEmbeddings
-from astrapy import DataAPIClient
-from astrapy.info import VectorServiceOptions
-# from astrapy.exceptions import CollectionAlreadyExistsException
-# from astrapy.core.api import APIRequestError
 
 logging.basicConfig(
     format='%(asctime)s - %(levelname)s - %(funcName)s - %(message)s',
@@ -16,59 +11,11 @@ logging.basicConfig(
 
 logging.info("Initializing AstraDB client...")
 
-# from astrapy import AstraClient
 ASTRA_DB_APPLICATION_TOKEN = os.environ['ASTRA_DB_APPLICATION_TOKEN']
 ASTRA_DB_API_ENDPOINT = os.environ['ASTRA_DB_API_ENDPOINT']
 COLLECTION_NAME = "article"
-# VECTOR_OPTIONS = VectorServiceOptions(
-#     provider="azureOpenAI",
-#     model_name="text-embedding-3-small",
-#     authentication={"providerKey": "AZURE_OPENAI_API_KEY"},
-#     parameters={
-#         "resourceName": "openai-oe",
-#         "deploymentId": "text-embedding-3-small",
-#     },
-# )
-
-# client = DataAPIClient(token=ASTRA_DB_APPLICATION_TOKEN)
-# database = client.get_database(ASTRA_DB_API_ENDPOINT)
-
-# embedding = AzureOpenAIEmbeddings(
-#     api_version="2024-07-01-preview",
-#     azure_endpoint="https://openai-oe.openai.azure.com/")
-
-# try:
-#     # Try to create the collection
-#     database.create_collection(
-#         name = COLLECTION_NAME,
-#         dimension=1536,  # Default dimension for text-embedding-3-small
-#         metric="cosine",
-#         service=VECTOR_OPTIONS
-#     )
-#     logging.info("Created new collection '%s'", COLLECTION_NAME)
-# except (CollectionAlreadyExistsException, APIRequestError) as e:
-#     logging.info("Collection '%s' already exists. Error Message: %s", COLLECTION_NAME, e)
-#     collection = database.get_collection(COLLECTION_NAME)
-
-# vstore = AstraDBVectorStore(
-#     collection_name=COLLECTION_NAME,
-#     namespace="default_keyspace",
-#     embedding=embedding,
-#     token=ASTRA_DB_APPLICATION_TOKEN,
-#     api_endpoint=ASTRA_DB_API_ENDPOINT)
 
 vstore = AstraDBVectorStore(
-    # collection_vector_service_options=VectorServiceOptions(
-    #     provider="azureOpenAI",
-    #     model_name="text-embedding-3-small",
-    #     authentication={
-    #         "providerKey": "AZURE_OPENAI_API_KEY",
-    #     },
-    #     parameters={
-    #         "resourceName": "openai-oe",
-    #         "deploymentId": "text-embedding-3-small",
-    #     },
-    # ),
     namespace="default_keyspace",
     collection_name="article",
     token=os.environ["ASTRA_DB_APPLICATION_TOKEN"],