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

FIX: proper parameters in Astra DB Vectorize options (#3901)

Browse source 
* FIX: proper parameters in vectorize options

* Update test_astra_component.py
This commit is contained in:
Eric Hare 2024-09-24 12:43:23 -07:00 • committed by GitHub
commit f403c17d10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -104,7 +104,7 @@ def test_astra_vectorize():
store = None
try:
options = {"provider": "nvidia", "modelName": "NV-Embed-QA"}
options_comp = {"provider": "nvidia", "z_01_model_name": "NV-Embed-QA"}
options_comp = {"provider": "nvidia", "z_00_model_name": "NV-Embed-QA"}
store = AstraDBVectorStore(
collection_name=VECTORIZE_COLLECTION,
@ -156,10 +156,10 @@ def test_astra_vectorize_with_provider_api_key():
options_comp = {
"provider": "openai",
"z_01_model_name": "text-embedding-3-small",
"z_04_model_parameters": {},
"z_02_authentication": {},
"z_00_model_name": "text-embedding-3-small",
"z_01_model_parameters": {},
"z_03_provider_api_key": "openai",
"z_04_authentication": {},
}
store = AstraDBVectorStore(
@ -212,9 +212,9 @@ def test_astra_vectorize_passes_authentication():
}
options_comp = {
"provider": "openai",
"z_01_model_name": "text-embedding-3-small",
"z_04_model_parameters": {},
"z_02_authentication": {"providerKey": "openai"},
"z_00_model_name": "text-embedding-3-small",
"z_01_model_parameters": {},
"z_04_authentication": {"providerKey": "openai"},
}
store = AstraDBVectorStore(