ai21: address breaking changes in sdk 2.14.0 [wip]

This commit is contained in:
Erick Friis
2024-08-23 10:35:07 -07:00
parent bc557a5663
commit ddfcdd39cb
2 changed files with 465 additions and 482 deletions

File diff suppressed because it is too large Load Diff

View File

@@ -17,9 +17,9 @@ _EXAMPLE_EMBEDDING_2 = [7.0, 8.0, 9.0]
_EXAMPLE_EMBEDDING_RESPONSE = EmbedResponse(
results=[
EmbedResult(_EXAMPLE_EMBEDDING_0),
EmbedResult(_EXAMPLE_EMBEDDING_1),
EmbedResult(_EXAMPLE_EMBEDDING_2),
EmbedResult(embedding=_EXAMPLE_EMBEDDING_0),
EmbedResult(embedding=_EXAMPLE_EMBEDDING_1),
EmbedResult(embedding=_EXAMPLE_EMBEDDING_2),
],
id="test_id",
)