scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-05 08:10:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Use ApiStreamChunk type for proper type compatibility

Browse source 
- Replace custom StreamChunk interface with ApiStreamChunk import
- Update all generator method signatures to use ApiStreamChunk
- Fixes type compatibility issues with yield* delegation
- Resolves CI test failures related to type mismatches
This commit is contained in:
hannesrudolph 2025-06-09 11:06:28 -06:00
parent d9a6a0edb7
commit e7199c228b
1 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/api/providers/bedrock.ts
View file

@ -64,11 +64,8 @@ interface BedrockPayload {
}
}
// Define interface for stream chunks with proper typing
interface StreamChunk {
type: "reasoning" | "text"
text: string
}
// Import the proper stream chunk types
import { ApiStreamChunk } from "../transform/stream"
// Define types for stream events based on AWS SDK
export interface StreamEvent {
@ -170,7 +167,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
/**
* Handles thinking content block events
*/
private *handleThinkingContentBlock(contentBlock: any): Generator<StreamChunk, void, unknown> {
private *handleThinkingContentBlock(contentBlock: any): Generator<ApiStreamChunk, void, unknown> {
if (contentBlock?.type === "thinking" && contentBlock.thinking !== undefined) {
yield {
type: "reasoning",
@ -182,7 +179,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
/**
* Handles text content block events
*/
private *handleTextContentBlock(start: any, contentBlock: any): Generator<StreamChunk, void, unknown> {
private *handleTextContentBlock(start: any, contentBlock: any): Generator<ApiStreamChunk, void, unknown> {
const text = start?.text || contentBlock?.text
if (text !== undefined) {
yield {
@ -195,7 +192,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
/**
* Handles thinking delta events
*/
private *handleThinkingDelta(delta: any): Generator<StreamChunk, void, unknown> {
private *handleThinkingDelta(delta: any): Generator<ApiStreamChunk, void, unknown> {
if (delta.type === "thinking_delta" && delta.thinking) {
yield {
type: "reasoning",
@ -207,7 +204,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
/**
* Handles signature delta events (part of thinking)
*/
private *handleSignatureDelta(delta: any): Generator<StreamChunk, void, unknown> {
private *handleSignatureDelta(delta: any): Generator<ApiStreamChunk, void, unknown> {
if (delta.type === "signature_delta" && delta.signature) {
// Signature is part of the thinking process, treat it as reasoning
yield {
@ -220,7 +217,7 @@ export class AwsBedrockHandler extends BaseProvider implements SingleCompletionH
/**
* Handles text delta events
*/
private *handleTextDelta(delta: any): Generator<StreamChunk, void, unknown> {
private *handleTextDelta(delta: any): Generator<ApiStreamChunk, void, unknown> {
if (delta.text) {
yield {
type: "text",