@agent-labs/ag-ui-server
Version:
AG-UI Node.js server with OpenAI integration
78 lines (77 loc) • 1.85 kB
text/typescript
import { EventType } from "@ag-ui/core";
//#region src/agent/types.d.ts
interface StreamContext {
messageId: string;
toolCallId: string;
isMessageStarted: boolean;
isToolCallStarted: boolean;
fullResponse: string;
toolCallArgs: string;
toolCallName: string;
getSnapshot(): StateSnapshot;
}
interface StateSnapshot {
last_response: string;
last_tool_call: {
name: string;
arguments: string;
} | null;
usage: {
prompt_tokens: number;
completion_tokens: number;
total_tokens: number;
};
}
interface StreamHandler {
handle(chunk: any, context: StreamContext): AsyncGenerator<string, void, unknown>;
finalize(context: StreamContext): AsyncGenerator<string, void, unknown>;
}
interface EventData {
type: EventType;
threadId?: string;
runId?: string;
messageId?: string;
role?: string;
delta?: string;
toolCallId?: string;
toolCallName?: string;
toolCallArgs?: string;
snapshot?: StateSnapshot;
content?: string;
name?: string;
toolCalls?: Array<{
id: string;
type: 'function';
function: {
name: string;
arguments: string;
};
}>;
messages?: Array<{
id: string;
role: string;
content?: string;
name?: string;
toolCalls?: Array<{
id: string;
type: 'function';
function: {
name: string;
arguments: string;
};
}>;
}>;
error?: {
message: string;
code?: string;
};
}
interface StreamProcessor {
process(stream: AsyncIterable<any>): AsyncGenerator<string, void, unknown>;
handleError(error: Error): AsyncGenerator<string, void, unknown>;
addHandler(type: string, handler: StreamHandler): void;
}
//# sourceMappingURL=types.d.ts.map
//#endregion
export { EventData, StateSnapshot, StreamContext, StreamHandler, StreamProcessor };
//# sourceMappingURL=types-CAxEI5am.d.mts.map