jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

test(astra): Refactor AstraDB imports in test_astra_component (#3413)

Browse source 
* test(astra): Refactor AstraDB imports in test_astra_component

* fix: Add AstraVectorizeComponent to AstraDB tests.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-20 16:49:12 -03:00 • committed by GitHub
commit 188708281a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/backend/tests/integration/astra/test_astra_component.py
View file

@ -1,7 +1,5 @@
import os import os
from langchain_astradb import AstraDBVectorStore, CollectionVectorServiceOptions
from langflow.components.embeddings.AstraVectorize import AstraVectorizeComponent
import pytest import pytest
from integration.utils import MockEmbeddings, check_env_vars from integration.utils import MockEmbeddings, check_env_vars
from langchain_core.documents import Document from langchain_core.documents import Document
@ -97,6 +95,10 @@ def test_astra_embeds_and_search(astra_fixture):
reason="missing astra env vars", reason="missing astra env vars",
) )
def test_astra_vectorize(): def test_astra_vectorize():
from langchain_astradb import AstraDBVectorStore, CollectionVectorServiceOptions
from langflow.components.embeddings.AstraVectorize import AstraVectorizeComponent
store = None store = None
try: try:
options = {"provider": "nvidia", "modelName": "NV-Embed-QA", "parameters": {}, "authentication": {}} options = {"provider": "nvidia", "modelName": "NV-Embed-QA", "parameters": {}, "authentication": {}}
@ -142,6 +144,10 @@ def test_astra_vectorize():
) )
def test_astra_vectorize_with_provider_api_key(): def test_astra_vectorize_with_provider_api_key():
"""tests vectorize using an openai api key""" """tests vectorize using an openai api key"""
from langchain_astradb import AstraDBVectorStore, CollectionVectorServiceOptions
from langflow.components.embeddings.AstraVectorize import AstraVectorizeComponent
store = None store = None
try: try:
application_token = os.getenv("ASTRA_DB_APPLICATION_TOKEN") application_token = os.getenv("ASTRA_DB_APPLICATION_TOKEN")
@ -186,6 +192,10 @@ def test_astra_vectorize_with_provider_api_key():
) )
def test_astra_vectorize_passes_authentication(): def test_astra_vectorize_passes_authentication():
"""tests vectorize using the authentication parameter""" """tests vectorize using the authentication parameter"""
from langchain_astradb import AstraDBVectorStore, CollectionVectorServiceOptions
from langflow.components.embeddings.AstraVectorize import AstraVectorizeComponent
store = None store = None
try: try:
application_token = os.getenv("ASTRA_DB_APPLICATION_TOKEN") application_token = os.getenv("ASTRA_DB_APPLICATION_TOKEN")