Skip to content

Commit 0b2b95a

Browse files
committed
Add usage validation to model information test
- Check for promptTokens and totalTokens in usage object - Verify token counts are numbers and greater than zero - Ensures comprehensive validation of API response
1 parent cf4875e commit 0b2b95a

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/e2e/embeddings.test.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,14 @@ describe("Embeddings E2E Tests", () => {
144144
if (typeof response === "object") {
145145
expect(response.model).toBeDefined();
146146
expect(typeof response.model).toBe("string");
147+
148+
if (response.usage) {
149+
expect(response.usage.promptTokens).toBeDefined();
150+
expect(response.usage.totalTokens).toBeDefined();
151+
expect(typeof response.usage.promptTokens).toBe("number");
152+
expect(typeof response.usage.totalTokens).toBe("number");
153+
expect(response.usage.totalTokens).toBeGreaterThan(0);
154+
}
147155
}
148156
});
149157
});

0 commit comments

Comments
 (0)