Skip to content

Commit cc0ab83

Browse files
fix: Added dimensions, instruction, query params to EmbeddingV2Params
1 parent 48ca7dc commit cc0ab83

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

jigsawstack/embedding_v2.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,9 @@ class EmbeddingV2Params(TypedDict):
1515
url: NotRequired[str]
1616
file_store_key: NotRequired[str]
1717
token_overflow_mode: NotRequired[Literal["truncate", "error"]]
18+
dimensions: NotRequired[int]
19+
instruction: NotRequired[str]
20+
query: NotRequired[bool]
1821
speaker_fingerprint: NotRequired[bool]
1922

2023

@@ -44,7 +47,9 @@ def __init__(
4447
@overload
4548
def execute(self, params: EmbeddingV2Params) -> EmbeddingV2Response: ...
4649
@overload
47-
def execute(self, blob: bytes, options: EmbeddingV2Params = None) -> EmbeddingV2Response: ...
50+
def execute(
51+
self, blob: bytes, options: EmbeddingV2Params = None
52+
) -> EmbeddingV2Response: ...
4853

4954
def execute(
5055
self,

0 commit comments

Comments
 (0)