31 lines
1011 B
TypeScript
31 lines
1011 B
TypeScript
/**
|
|
* Mock provider implementation
|
|
*/
|
|
|
|
import { callMock, callMockCustom, type MockCallOptions } from '../mock/client.js';
|
|
import type { AgentResponse } from '../models/types.js';
|
|
import type { Provider, ProviderCallOptions } from './index.js';
|
|
|
|
/** Mock provider - wraps existing Mock client */
|
|
export class MockProvider implements Provider {
|
|
async call(agentName: string, prompt: string, options: ProviderCallOptions): Promise<AgentResponse> {
|
|
const callOptions: MockCallOptions = {
|
|
cwd: options.cwd,
|
|
sessionId: options.sessionId,
|
|
onStream: options.onStream,
|
|
};
|
|
|
|
return callMock(agentName, prompt, callOptions);
|
|
}
|
|
|
|
async callCustom(agentName: string, prompt: string, _systemPrompt: string, options: ProviderCallOptions): Promise<AgentResponse> {
|
|
const callOptions: MockCallOptions = {
|
|
cwd: options.cwd,
|
|
sessionId: options.sessionId,
|
|
onStream: options.onStream,
|
|
};
|
|
|
|
return callMockCustom(agentName, prompt, _systemPrompt, callOptions);
|
|
}
|
|
}
|