Skip to content

Commit

Permalink
Fix some references to moved/renamed types in the wrapper
Browse files Browse the repository at this point in the history
  • Loading branch information
twitchard committed Sep 27, 2024
1 parent b9a5621 commit 08cb6be
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion src/wrapper/expressionMeasurement/batch/BatchClient.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ import { Job } from "./Job";

export class BatchClient extends FernClient {
public async startInferenceJob(
request: Hume.expressionMeasurement.InferenceBaseRequest = {},
request: Hume.expressionMeasurement.batch.InferenceBaseRequest = {},
requestOptions?: FernClient.RequestOptions
): Promise<Job> {
const { jobId } = await super.startInferenceJob(request, requestOptions);
Expand Down
2 changes: 1 addition & 1 deletion src/wrapper/expressionMeasurement/batch/Job.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ import * as Hume from "../../../api";
import * as errors from "../../../errors";
import { BatchClient } from "./BatchClient";

export class Job implements Hume.expressionMeasurement.JobId {
export class Job implements Hume.expressionMeasurement.batch.JobId {
constructor(public readonly jobId: string, private readonly client: BatchClient) {}

public async awaitCompletion(timeoutInSeconds = 300): Promise<void> {
Expand Down
12 changes: 6 additions & 6 deletions src/wrapper/expressionMeasurement/streaming/StreamSocket.ts
Original file line number Diff line number Diff line change
Expand Up @@ -173,14 +173,14 @@ export class StreamSocket {

private async send(
payload: Hume.expressionMeasurement.stream.StreamModelsEndpointPayload
): Promise<Hume.expressionMeasurement.SubscribeEvent | void> {
): Promise<Hume.expressionMeasurement.stream.Config | void> {
await this.tillSocketOpen();
const jsonPayload = serializers.expressionMeasurement.StreamModelsEndpointPayload.jsonOrThrow(payload, {
const jsonPayload = serializers.expressionMeasurement.stream.StreamModelsEndpointPayload.jsonOrThrow(payload, {
unrecognizedObjectKeys: "strip",
});
this.websocket.send(JSON.stringify(jsonPayload));
const response = await new Promise<
Hume.expressionMeasurement.stream.Config | Hume.expressionMeasurement.StreamErrorMessage | undefined
Hume.expressionMeasurement.stream.Config | Hume.expressionMeasurement.stream.StreamErrorMessage | undefined
>((resolve) => {
this.websocket.addEventListener("message", (event) => {
const response = parse(event.data);
Expand Down Expand Up @@ -212,7 +212,7 @@ export class StreamSocket {
}

function isError(
response: Hume.expressionMeasurement.stream.Config | Hume.expressionMeasurement.StreamErrorMessage
): response is Hume.expressionMeasurement.StreamErrorMessage {
return (response as Hume.expressionMeasurement.StreamErrorMessage).error != null;
response: Hume.expressionMeasurement.stream.Config | Hume.expressionMeasurement.stream.StreamErrorMessage
): response is Hume.expressionMeasurement.stream.StreamErrorMessage {
return (response as Hume.expressionMeasurement.stream.StreamErrorMessage).error != null;
}
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ export declare namespace StreamClient {
streamWindowMs?: number;

onOpen?: (event: WebSocket.Event) => void;
onMessage?: (message: Hume.expressionMeasurement.Config) => void;
onMessage?: (message: Hume.expressionMeasurement.stream.Config) => void;
onError?: (error: Hume.expressionMeasurement.stream.StreamErrorMessage) => void;
onClose?: (event: WebSocket.Event) => void;
}
Expand Down

0 comments on commit 08cb6be

Please sign in to comment.