sf-vector-sdk 0.2.0__py3-none-any.whl → 0.2.3__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.
- {sf_vector_sdk-0.2.0.dist-info → sf_vector_sdk-0.2.3.dist-info}/METADATA +38 -4
- {sf_vector_sdk-0.2.0.dist-info → sf_vector_sdk-0.2.3.dist-info}/RECORD +17 -17
- vector_sdk/__init__.py +17 -1
- vector_sdk/generated/embedding_pipeline/content_types/v1/content_types_pb2.py +2 -2
- vector_sdk/generated/embedding_pipeline/db/vectors/v1/vectors_pb2.py +2 -2
- vector_sdk/generated/embedding_pipeline/query/v1/query_pb2.py +2 -2
- vector_sdk/generated/embedding_pipeline/tools/v1/tools_pb2.py +13 -7
- vector_sdk/generated/embedding_pipeline/tools/v1/tools_pb2.pyi +23 -1
- vector_sdk/hash/__init__.py +2 -0
- vector_sdk/hash/hasher.py +28 -2
- vector_sdk/hash/types.py +10 -1
- vector_sdk/namespaces/embeddings.py +31 -57
- vector_sdk/namespaces/search.py +38 -60
- vector_sdk/structured/__init__.py +13 -0
- vector_sdk/structured/structured_embeddings.py +785 -0
- vector_sdk/structured/tool_config.py +23 -4
- {sf_vector_sdk-0.2.0.dist-info → sf_vector_sdk-0.2.3.dist-info}/WHEEL +0 -0
|
@@ -144,6 +144,25 @@ TOOL_CONFIGS: dict[ToolCollection, ToolConfig] = {
|
|
|
144
144
|
namespace_pattern="audiorecapv2section",
|
|
145
145
|
),
|
|
146
146
|
),
|
|
147
|
+
"Topic": ToolConfig(
|
|
148
|
+
tool_collection="Topic",
|
|
149
|
+
model="gemini-embedding-001",
|
|
150
|
+
dimensions=3072,
|
|
151
|
+
default_priority=PRIORITY_NORMAL,
|
|
152
|
+
turbopuffer=TurboPufferToolConfig(
|
|
153
|
+
enabled=True,
|
|
154
|
+
id_field="contentHash",
|
|
155
|
+
metadata_fields=_DEFAULT_METADATA_FIELDS,
|
|
156
|
+
namespace_pattern="topic_vectors",
|
|
157
|
+
),
|
|
158
|
+
pinecone=PineconeToolConfig(
|
|
159
|
+
enabled=False,
|
|
160
|
+
index_name="tool-vectors",
|
|
161
|
+
id_field="contentHash",
|
|
162
|
+
metadata_fields=_DEFAULT_METADATA_FIELDS,
|
|
163
|
+
namespace_pattern="topic_vectors",
|
|
164
|
+
),
|
|
165
|
+
),
|
|
147
166
|
}
|
|
148
167
|
|
|
149
168
|
|
|
@@ -198,8 +217,8 @@ def get_turbopuffer_namespace(
|
|
|
198
217
|
config = TOOL_CONFIGS[tool_collection]
|
|
199
218
|
pattern = config.turbopuffer.namespace_pattern
|
|
200
219
|
|
|
201
|
-
# AudioRecapV2Section
|
|
202
|
-
if tool_collection
|
|
220
|
+
# AudioRecapV2Section and Topic don't have sub-types
|
|
221
|
+
if tool_collection in ("AudioRecapV2Section", "Topic"):
|
|
203
222
|
return pattern
|
|
204
223
|
|
|
205
224
|
# Derive the type suffix
|
|
@@ -228,8 +247,8 @@ def get_pinecone_namespace(
|
|
|
228
247
|
config = TOOL_CONFIGS[tool_collection]
|
|
229
248
|
pattern = config.pinecone.namespace_pattern
|
|
230
249
|
|
|
231
|
-
# AudioRecapV2Section
|
|
232
|
-
if tool_collection
|
|
250
|
+
# AudioRecapV2Section and Topic don't have sub-types
|
|
251
|
+
if tool_collection in ("AudioRecapV2Section", "Topic"):
|
|
233
252
|
return pattern
|
|
234
253
|
|
|
235
254
|
# Derive the type suffix
|
|
File without changes
|