import { AIMessage, DefaultToolCall, ToolCallWithResult } from "../../types.messages.js";
import { BagTemplate } from "../../types.template.js";
import { UseStreamOptions } from "../types.js";
import { BaseStream } from "./base.js";

//#region src/ui/stream/agent.d.ts
/**
 * Stream interface for ReactAgent instances created with `createAgent`.
 *
 * Extends {@link UseGraphStream} with tool calling capabilities. Tool calls are
 * automatically typed based on the agent's tools configuration.
 *
 * Use this interface when streaming from an agent created with `createAgent`.
 * For subagent streaming capabilities, use {@link UseDeepAgentStream} with `createDeepAgent`.
 *
 * @experimental This interface is subject to change.
 *
 * @template StateType - The agent's state type (base + middleware states)
 * @template ToolCall - Tool call type inferred from agent's tools
 * @template Bag - Type configuration bag
 *
 * @example
 * ```typescript
 * import { createAgent, tool } from "@langchain/langgraph";
 * import { useStream } from "@langchain/langgraph-sdk/react";
 * import { z } from "zod";
 *
 * // Define tools with typed schemas
 * const searchTool = tool(
 *   async ({ query }) => `Results for: ${query}`,
 *   { name: "search", schema: z.object({ query: z.string() }) }
 * );
 *
 * const calculatorTool = tool(
 *   async ({ expression }) => eval(expression).toString(),
 *   { name: "calculator", schema: z.object({ expression: z.string() }) }
 * );
 *
 * // Create the agent
 * const agent = createAgent({
 *   model: "gpt-4",
 *   tools: [searchTool, calculatorTool],
 * });
 *
 * // In React component:
 * function Chat() {
 *   const stream = useStream<typeof agent>({
 *     assistantId: "my-agent",
 *     apiUrl: "http://localhost:2024",
 *   });
 *
 *   // Tool calls are typed!
 *   stream.toolCalls.forEach(tc => {
 *     if (tc.call.name === "search") {
 *       // tc.call.args is typed as { query: string }
 *       console.log("Searching for:", tc.call.args.query);
 *     } else if (tc.call.name === "calculator") {
 *       // tc.call.args is typed as { expression: string }
 *       console.log("Calculating:", tc.call.args.expression);
 *     }
 *   });
 * }
 * ```
 *
 * @remarks
 * This interface adds tool calling on top of {@link UseGraphStream}:
 * - `toolCalls` - Array of tool calls paired with their results
 * - `getToolCalls(message)` - Get tool calls for a specific AI message
 *
 * It does NOT include subagent streaming features. For those, use
 * {@link UseDeepAgentStream} with `createDeepAgent`.
 */
interface UseAgentStream<StateType extends object = Record<string, unknown>, ToolCall = DefaultToolCall, Bag extends BagTemplate = BagTemplate> extends BaseStream<StateType, ToolCall, Bag> {
  /**
   * Tool calls paired with their results.
   *
   * Each entry contains the tool call request and its corresponding result.
   * Useful for rendering tool invocations and their outputs together.
   *
   * @example
   * ```typescript
   * stream.toolCalls.map(({ call, result }) => (
   *   <ToolCallCard
   *     name={call.name}
   *     args={call.args}
   *     result={result}
   *   />
   * ));
   * ```
   */
  toolCalls: ToolCallWithResult<ToolCall>[];
  /**
   * Get tool calls for a specific AI message.
   *
   * Use this to find which tool calls were initiated by a particular
   * assistant message, useful for rendering tool calls inline with messages.
   *
   * @param message - The AI message to get tool calls for
   * @returns Array of tool calls initiated by the message
   *
   * @example
   * ```typescript
   * messages.map(message => {
   *   if (message.type === "ai") {
   *     const calls = stream.getToolCalls(message);
   *     return (
   *       <>
   *         <MessageBubble message={message} />
   *         {calls.map(tc => <ToolCallCard key={tc.call.id} {...tc} />)}
   *       </>
   *     );
   *   }
   *   return <MessageBubble message={message} />;
   * });
   * ```
   */
  getToolCalls: (message: AIMessage<ToolCall>) => ToolCallWithResult<ToolCall>[];
}
/**
 * Options for configuring an agent stream.
 *
 * Use this options interface when calling `useStream` with a ReactAgent
 * created via `createAgent`.
 *
 * @experimental This interface is subject to change.
 *
 * @template StateType - The agent's state type
 * @template Bag - Type configuration bag
 *
 * @example
 * ```typescript
 * const stream = useStream<typeof agent>({
 *   assistantId: "my-agent",
 *   apiUrl: "http://localhost:2024",
 *   onError: (error) => console.error(error),
 * });
 * ```
 */
interface UseAgentStreamOptions<StateType extends Record<string, unknown> = Record<string, unknown>, Bag extends BagTemplate = BagTemplate> extends UseStreamOptions<StateType, Bag> {}
//#endregion
export { UseAgentStream, UseAgentStreamOptions };
//# sourceMappingURL=agent.d.ts.map