File tree Expand file tree Collapse file tree 2 files changed +3
-14
lines changed
veadk/integrations/ve_tos Expand file tree Collapse file tree 2 files changed +3
-14
lines changed Original file line number Diff line number Diff line change @@ -29,13 +29,3 @@ async def test_knowledgebase():
2929 )
3030
3131 assert isinstance (kb ._backend , InMemoryKnowledgeBackend )
32-
33-
34- @pytest .mark .asyncio
35- async def test_viking_knowledgebase_add_texts ():
36- app_name = "kb_test_app"
37- kb = KnowledgeBase (
38- backend = "viking" ,
39- app_name = app_name ,
40- )
41- assert kb .add_from_text (text = "test text" , tag = "tag" ) is True
Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ def __init__(
3737 ak : str = "" ,
3838 sk : str = "" ,
3939 region : str = "cn-beijing" ,
40- bucket_name : str = "" ,
40+ bucket_name : str = DEFAULT_TOS_BUCKET_NAME ,
4141 ) -> None :
4242 self .ak = ak if ak else os .getenv ("VOLCENGINE_ACCESS_KEY" , "" )
4343 self .sk = sk if sk else os .getenv ("VOLCENGINE_SECRET_KEY" , "" )
@@ -49,10 +49,9 @@ def __init__(
4949 )
5050
5151 self .region = region
52- self .bucket_name = bucket_name or getenv (
53- "DATABASE_TOS_BUCKET " , DEFAULT_TOS_BUCKET_NAME
52+ self .bucket_name = (
53+ bucket_name if bucket_name else getenv ( " " , DEFAULT_TOS_BUCKET_NAME )
5454 )
55-
5655 self ._tos_module = None
5756
5857 try :
You can’t perform that action at this time.
0 commit comments