@convex-dev/agent
Version:
A agent component for Convex.
42 lines • 2.01 kB
TypeScript
import type { LanguageModelV2, LanguageModelV2Content } from "@ai-sdk/provider";
import { type ProviderMetadata } from "ai";
export declare const DEFAULT_TEXT = "\nA A A A A A A A A A A A A A A\nB B B B B B B B B B B B B B B\nC C C C C C C C C C C C C C C\nD D D D D D D D D D D D D D D\n";
export type MockModelArgs = {
provider?: LanguageModelV2["provider"];
modelId?: LanguageModelV2["modelId"];
supportedUrls?: LanguageModelV2["supportedUrls"] | (() => LanguageModelV2["supportedUrls"]);
chunkDelayInMs?: number;
initialDelayInMs?: number;
/** A list of the responses for multiple steps.
* For tool calls, the first list would include a tool call part,
* then the next list would be after the tool response or another tool call.
* Tool responses come from actual tool calls!
*/
contentSteps?: LanguageModelV2Content[][];
/** A single list of content responded from each step.
* Provide contentSteps instead if you want to do multi-step responses with
* tool calls.
*/
content?: LanguageModelV2Content[];
doGenerate?: LanguageModelV2["doGenerate"];
doStream?: LanguageModelV2["doStream"];
providerMetadata?: ProviderMetadata;
fail?: boolean | {
probability?: number;
error?: string;
};
};
export declare function mockModel(args?: MockModelArgs): LanguageModelV2;
export declare class MockLanguageModel implements LanguageModelV2 {
readonly specificationVersion = "v2";
private _supportedUrls;
readonly provider: LanguageModelV2["provider"];
readonly modelId: LanguageModelV2["modelId"];
doGenerate: LanguageModelV2["doGenerate"];
doStream: LanguageModelV2["doStream"];
doGenerateCalls: Parameters<LanguageModelV2["doGenerate"]>[0][];
doStreamCalls: Parameters<LanguageModelV2["doStream"]>[0][];
constructor(args: MockModelArgs);
get supportedUrls(): Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>>;
}
//# sourceMappingURL=mockModel.d.ts.map