summaryrefslogtreecommitdiff
path: root/packages/cli/src/core/GeminiStream.ts
blob: 285683069395dfe0e5bec7c6ccc2700adf114547 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import { ToolCallEvent } from "../ui/types.js";

export enum GeminiEventType {
    Content,
    ToolCallInfo,
}

export interface GeminiContentEvent {
    type: GeminiEventType.Content;
    value: string;
}

export interface GeminiToolCallInfoEvent {
    type: GeminiEventType.ToolCallInfo;
    value: ToolCallEvent;
}

export type GeminiEvent =
    | GeminiContentEvent
    | GeminiToolCallInfoEvent;

export type GeminiStream = AsyncIterable<GeminiEvent>;