From f3cd6cb09a7ddd03e669423e47c9a7818e579eeb Mon Sep 17 00:00:00 2001 From: ACBBZ Date: Tue, 7 May 2024 06:25:17 +0000 Subject: [PATCH] add chroma setting --- src/utils/chroma_setting.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 src/utils/chroma_setting.py diff --git a/src/utils/chroma_setting.py b/src/utils/chroma_setting.py new file mode 100644 index 0000000..5081134 --- /dev/null +++ b/src/utils/chroma_setting.py @@ -0,0 +1,25 @@ +from dataclasses import dataclass, field + +@dataclass +class ChromaSetting: + collection_ids: list[str]= field(default_factory=list) + string_ids: list[int] = field(default_factory=list) + file_ids: list[dict] = field(default_factory=list) + + +# c = ChromaSetting() +# # collection_name +# c.collection_ids = ["123", "collection_id2"] +# # 插入的字符串的id,从1开始 +# c.string_ids = [0, 0] +# # 插入的文件的文件名和ids +# c.file_ids = [ +# { +# "file_name1": ["file_name1", ["1","2","3","4"]], +# "file_name2": ["file_name2", ["1","2","3","4"]] +# }, +# {} +# ] + +# c.string_ids[0] +=1 +# print(c) \ No newline at end of file