Skip to content

Commit d683590

Browse files
committed
Ellipsis PR feedback
1 parent 75ef5e0 commit d683590

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

src/api/providers/openrouter.ts

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import OpenAI from "openai"
44
import { ApiHandler } from "../"
55
import { ApiHandlerOptions, ModelInfo, openRouterDefaultModelId, openRouterDefaultModelInfo } from "../../shared/api"
66
import { convertToOpenAiMessages } from "../transform/openai-format"
7-
import { ApiStreamChunk } from "../transform/stream"
7+
import { ApiStreamChunk, ApiStreamUsageChunk } from "../transform/stream"
88
import delay from "delay"
99

1010
// Add custom interface for OpenRouter params
@@ -13,11 +13,7 @@ interface OpenRouterChatCompletionParams extends OpenAI.Chat.ChatCompletionCreat
1313
}
1414

1515
// Add custom interface for OpenRouter usage chunk
16-
interface OpenRouterApiStreamUsageChunk {
17-
type: "usage";
18-
inputTokens: number;
19-
outputTokens: number;
20-
totalCost: number;
16+
interface OpenRouterApiStreamUsageChunk extends ApiStreamUsageChunk {
2117
fullResponseText: string;
2218
}
2319

0 commit comments

Comments
 (0)