Skip to content

Commit b961a1e

Browse files
authored
fix(drivers-vector-store-astradb): pass correct client parameters (#1907)
1 parent e3d9f73 commit b961a1e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

griptape/drivers/vector/astradb_vector_store_driver.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
from attrs import define, field
66

77
from griptape.drivers.vector import BaseVectorStoreDriver
8-
from griptape.utils import import_optional_dependency
98
from griptape.utils.decorators import lazy_property
9+
from griptape.utils.import_utils import import_optional_dependency
1010

1111
if TYPE_CHECKING:
1212
import astrapy
@@ -49,9 +49,11 @@ class AstraDbVectorStoreDriver(BaseVectorStoreDriver):
4949

5050
@lazy_property()
5151
def client(self) -> astrapy.DataAPIClient:
52-
return import_optional_dependency("astrapy").DataAPIClient(
53-
caller_name=self.caller_name,
54-
environment=self.environment,
52+
astrapy = import_optional_dependency("astrapy")
53+
54+
return astrapy.DataAPIClient(
55+
callers=[(self.caller_name, None)],
56+
environment=self.environment or astrapy.utils.unset.UnsetType(),
5557
)
5658

5759
@lazy_property()

0 commit comments

Comments
 (0)