Skip to content

Commit 7d19138

Browse files
committed
small tweak
1 parent d2b7dee commit 7d19138

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

.github/workflows/test.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ jobs:
4545
HF_REPLICATE_KEY: dummy
4646
HF_SAMBANOVA_KEY: dummy
4747
HF_TOGETHER_KEY: dummy
48-
HF_FIREWORKS_AI_KEY: dummy
48+
HF_FIREWORKS_KEY: dummy
4949

5050
browser:
5151
runs-on: ubuntu-latest
@@ -86,7 +86,7 @@ jobs:
8686
HF_REPLICATE_KEY: dummy
8787
HF_SAMBANOVA_KEY: dummy
8888
HF_TOGETHER_KEY: dummy
89-
HF_FIREWORKS_AI_KEY: dummy
89+
HF_FIREWORKS_KEY: dummy
9090

9191
e2e:
9292
runs-on: ubuntu-latest
@@ -154,4 +154,4 @@ jobs:
154154
HF_REPLICATE_KEY: dummy
155155
HF_SAMBANOVA_KEY: dummy
156156
HF_TOGETHER_KEY: dummy
157-
HF_FIREWORKS_AI_KEY: dummy
157+
HF_FIREWORKS_KEY: dummy

packages/inference/test/HfInference.spec.ts

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1081,7 +1081,7 @@ describe.concurrent("HfInference", () => {
10811081
describe.concurrent(
10821082
"Fireworks",
10831083
() => {
1084-
const client = new HfInference(env.HF_FIREWORKS_AI_KEY);
1084+
const client = new HfInference(env.HF_FIREWORKS_KEY);
10851085

10861086
it("chatCompletion", async () => {
10871087
const res = await client.chatCompletion({
@@ -1093,14 +1093,14 @@ describe.concurrent("HfInference", () => {
10931093
const completion = res.choices[0].message?.content;
10941094
expect(completion).toContain("two");
10951095
}
1096-
});
1096+
});
10971097

10981098
it("chatCompletion stream", async () => {
10991099
const stream = client.chatCompletionStream({
11001100
model: "deepseek-ai/DeepSeek-R1",
11011101
provider: "fireworks-ai",
11021102
messages: [{ role: "user", content: "Say this is a test" }],
1103-
stream: true
1103+
stream: true,
11041104
}) as AsyncGenerator<ChatCompletionStreamOutput>;
11051105

11061106
let fullResponse = "";
@@ -1112,12 +1112,12 @@ describe.concurrent("HfInference", () => {
11121112
}
11131113
}
11141114
}
1115-
1115+
11161116
// Verify we got a meaningful response
11171117
expect(fullResponse).toBeTruthy();
11181118
expect(fullResponse.length).toBeGreaterThan(0);
11191119
});
11201120
},
11211121
TIMEOUT
11221122
);
1123-
});
1123+
});

0 commit comments

Comments
 (0)