Skip to content

Commit b0e2c96

Browse files
committed
fix types
1 parent 385dd48 commit b0e2c96

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

agents-run-api/src/agents/Agent.ts

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,15 @@ import {
3434
TemplateEngine,
3535
} from '@inkeep/agents-core';
3636
import { type Span, SpanStatusCode, trace } from '@opentelemetry/api';
37-
import { generateText, Output, streamText, type Tool, type ToolSet, tool } from 'ai';
37+
import {
38+
generateText,
39+
Output,
40+
type StreamTextResult,
41+
streamText,
42+
type Tool,
43+
type ToolSet,
44+
tool,
45+
} from 'ai';
3846
import {
3947
AGENT_EXECUTION_MAX_GENERATION_STEPS,
4048
FUNCTION_TOOL_EXECUTION_TIMEOUT_MS_DEFAULT,
@@ -3259,7 +3267,10 @@ ${output}${structureHintsFormatted}`;
32593267
}
32603268
}
32613269

3262-
private async processStreamEvents(streamResult: any, parser: any) {
3270+
private async processStreamEvents(
3271+
streamResult: StreamTextResult<ToolSet, any>,
3272+
parser: IncrementalStreamParser
3273+
) {
32633274
for await (const event of streamResult.fullStream) {
32643275
switch (event.type) {
32653276
case 'text-delta':

0 commit comments

Comments
 (0)