versionhq 1.1.11.5__py3-none-any.whl → 1.1.11.6__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- versionhq/__init__.py +1 -1
- versionhq/knowledge/storage.py +12 -1
- {versionhq-1.1.11.5.dist-info → versionhq-1.1.11.6.dist-info}/METADATA +1 -1
- {versionhq-1.1.11.5.dist-info → versionhq-1.1.11.6.dist-info}/RECORD +7 -7
- {versionhq-1.1.11.5.dist-info → versionhq-1.1.11.6.dist-info}/LICENSE +0 -0
- {versionhq-1.1.11.5.dist-info → versionhq-1.1.11.6.dist-info}/WHEEL +0 -0
- {versionhq-1.1.11.5.dist-info → versionhq-1.1.11.6.dist-info}/top_level.txt +0 -0
versionhq/__init__.py
CHANGED
versionhq/knowledge/storage.py
CHANGED
@@ -73,11 +73,21 @@ class KnowledgeStorage(BaseKnowledgeStorage):
|
|
73
73
|
|
74
74
|
|
75
75
|
def __init__(self, embedder_config: Optional[Dict[str, Any]] = None, collection_name: Optional[str] = None):
|
76
|
-
self.collection_name = collection_name if collection_name else "knowledge"
|
76
|
+
self.collection_name = self._validate_collection_name(collection_name) if collection_name else "knowledge"
|
77
77
|
self.embedder_config = embedder_config
|
78
78
|
self.initialize_knowledge_storage()
|
79
79
|
|
80
80
|
|
81
|
+
def _validate_collection_name(self, collection_name: str = None) -> str:
|
82
|
+
"""
|
83
|
+
Return a valid collection name from the given collection name.
|
84
|
+
Expected collection name (1) contains 3-63 characters, (2) starts and ends with an alphanumeric character, (3) otherwise contains only alphanumeric characters, underscores or hyphens (-), (4) contains no two consecutive periods (..) and (5) is not a valid IPv4 address.
|
85
|
+
"""
|
86
|
+
collection_name = collection_name if collection_name else self.collection_name
|
87
|
+
valid_collection_name = collection_name.replace(' ', "-").replace("(", "-").replace(")", "").replace("..", "")
|
88
|
+
return valid_collection_name
|
89
|
+
|
90
|
+
|
81
91
|
def _create_default_embedding_function(self) -> Any:
|
82
92
|
from chromadb.utils.embedding_functions.openai_embedding_function import OpenAIEmbeddingFunction
|
83
93
|
|
@@ -101,6 +111,7 @@ class KnowledgeStorage(BaseKnowledgeStorage):
|
|
101
111
|
chroma_client = chromadb.PersistentClient(path=base_path, settings=Settings(allow_reset=True))
|
102
112
|
self.app = chroma_client
|
103
113
|
self._set_embedding_function(embedder_config=self.embedder_config)
|
114
|
+
self.collection_name = self.collection_name if self.collection_name else "knowledge"
|
104
115
|
|
105
116
|
try:
|
106
117
|
if self.app:
|
@@ -1,4 +1,4 @@
|
|
1
|
-
versionhq/__init__.py,sha256=
|
1
|
+
versionhq/__init__.py,sha256=khEpIbv1gBkv7txiWOdTPlq7MYqe15UUyLe63UaP-ro,863
|
2
2
|
versionhq/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
3
3
|
versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
|
4
4
|
versionhq/_utils/logger.py,sha256=U-MpeGueA6YS8Ptfy0VnU_ePsZP-8Pvkvi0tZ4s_UMg,1438
|
@@ -26,7 +26,7 @@ versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3
|
|
26
26
|
versionhq/knowledge/model.py,sha256=n7kU4jQ24BUIxwosSVRK8tYhAFYhgc4yf7e4Q-bq4bk,1832
|
27
27
|
versionhq/knowledge/source.py,sha256=WOARChmm_cNtBD-xGo4RoYmcuodzdalctXI-gDBCW6k,13610
|
28
28
|
versionhq/knowledge/source_docling.py,sha256=Lv7PDE97pwV5_3SPcIgTzEu3H4obRbPfY3NALvfkgX8,5364
|
29
|
-
versionhq/knowledge/storage.py,sha256=
|
29
|
+
versionhq/knowledge/storage.py,sha256=7oxCg3W9mFjYH1YmuH9kFtTbNxquzYFjuUjd_TlsB9E,8170
|
30
30
|
versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
31
31
|
versionhq/llm/llm_vars.py,sha256=PO__b-h5e-6oQ-uoIgXx3lPSAUPUwXYfdVRW73fvX14,8761
|
32
32
|
versionhq/llm/model.py,sha256=1uaBxT10GIlUl-BtE8Mfux-ZRcScp4HUIas_fD_cdWQ,14471
|
@@ -57,8 +57,8 @@ versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtg
|
|
57
57
|
versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
|
58
58
|
versionhq/tool/model.py,sha256=7ccEnje_8LuxLVeog6pL38nToArXQXk4KY7A9hfprDo,12239
|
59
59
|
versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
|
60
|
-
versionhq-1.1.11.
|
61
|
-
versionhq-1.1.11.
|
62
|
-
versionhq-1.1.11.
|
63
|
-
versionhq-1.1.11.
|
64
|
-
versionhq-1.1.11.
|
60
|
+
versionhq-1.1.11.6.dist-info/LICENSE,sha256=7CCXuMrAjPVsUvZrsBq9DsxI2rLDUSYXR_qj4yO_ZII,1077
|
61
|
+
versionhq-1.1.11.6.dist-info/METADATA,sha256=6ApakYLq7bQHnNPx9kwvjTctXrXf7lYL4CJ-xTdPPgM,18638
|
62
|
+
versionhq-1.1.11.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
63
|
+
versionhq-1.1.11.6.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
|
64
|
+
versionhq-1.1.11.6.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|