Skip to content

Commit

Permalink
Fix snap
Browse files Browse the repository at this point in the history
  • Loading branch information
billytrend-cohere committed Jan 14, 2025
1 parent 8745e19 commit 3f5449c
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions src/test/tests.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ describe("test sdk", () => {
test.skip("check that no emojis get chopped up", async () => {
const chat = await cohere.chatStream({
model: "command-r",
message: "generate 2000 emojis"
message: "generate 2000 emojis",
});

let finalChunk: StreamedChatResponse;
Expand All @@ -99,7 +99,6 @@ describe("test sdk", () => {
}
});


test.concurrent("classify works", async () => {
const classify = await cohere.classify({
examples: [
Expand Down Expand Up @@ -240,7 +239,7 @@ describe("test sdk", () => {
tools,
toolResults: toolResults,
model: "command-nightly",
forceSingleStep: true
forceSingleStep: true,
});

expect(citedResponse.documents).toMatchInlineSnapshot(`
Expand All @@ -250,7 +249,6 @@ describe("test sdk", () => {
"date": "2023-09-29",
"id": "sales_database:0:0",
"numberOfSales": "120",
"tool_name": "sales_database",
"totalRevenue": "48500",
},
]
Expand Down

0 comments on commit 3f5449c

Please sign in to comment.