File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change 5
5
from attrs import define , field
6
6
7
7
from griptape .drivers .vector import BaseVectorStoreDriver
8
- from griptape .utils import import_optional_dependency
9
8
from griptape .utils .decorators import lazy_property
9
+ from griptape .utils .import_utils import import_optional_dependency
10
10
11
11
if TYPE_CHECKING :
12
12
import astrapy
@@ -49,9 +49,11 @@ class AstraDbVectorStoreDriver(BaseVectorStoreDriver):
49
49
50
50
@lazy_property ()
51
51
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 (),
55
57
)
56
58
57
59
@lazy_property ()
You can’t perform that action at this time.
0 commit comments