Skip to content

Commit 88d6dba

Browse files
committed
add collection type
1 parent 3534acb commit 88d6dba

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

veadk/integrations/ve_viking_db_memory/ve_viking_db_memory.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# limitations under the License.
1414

1515
import json
16+
import os
1617
import threading
1718

1819
from volcengine.ApiInfo import ApiInfo
@@ -281,6 +282,9 @@ def create_collection(
281282
params = {
282283
"CollectionName": collection_name,
283284
"ProjectName": project,
285+
"CollectionType": os.getenv(
286+
"DATABASE_VIKINGMEM_COLLECTION_TYPE", "standard"
287+
),
284288
"Description": description,
285289
"CustomEventTypeSchemas": custom_event_type_schemas,
286290
"CustomEntityTypeSchemas": custom_entity_type_schemas,

0 commit comments

Comments
 (0)