Skip to content

Commit a0ec42a

Browse files
committed
fix(tos): set default bucket name for VeTOS class
1 parent 1a2f87f commit a0ec42a

File tree

2 files changed

+3
-14
lines changed

2 files changed

+3
-14
lines changed

tests/test_knowledgebase.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff 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

veadk/integrations/ve_tos/ve_tos.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff 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:

0 commit comments

Comments
 (0)