Package ai.protify.core.provider.mock
Class MockProvider
- java.lang.Object
-
- ai.protify.core.provider.mock.MockProvider
-
- All Implemented Interfaces:
AIProvider
public class MockProvider extends java.lang.Object implements AIProvider
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classMockProvider.Builder
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static MockProvider.Builderbuilder()voidclearRecordedRequests()voidenqueueResponse(AIResponse response)voidenqueueResponse(java.lang.String text)java.lang.StringgetApiKeyVarName()java.util.Map<java.lang.String,java.lang.String>getHeaders(java.lang.String credential)MockProviderRequestgetLastRequest()java.lang.StringgetName()java.lang.Class<? extends AIProviderClient<?>>getProviderClientType()java.util.List<MockProviderRequest>getRecordedRequests()intgetRequestCount()booleanisMimeTypeSupported(MimeType mimeType)voidreset()voidsetDefaultResponse(AIResponse response)voidsetDefaultResponse(java.lang.String text)voidsetResponseFunction(java.util.function.Function<MockProviderRequest,AIResponse> responseFunction)voidsetStreamTokenDelayMillis(long millis)java.lang.StringtoString()static MockProviderwithResponse(AIResponse response)static MockProviderwithResponse(java.lang.String text)static MockProviderwithResponseFunction(java.util.function.Function<MockProviderRequest,AIResponse> responseFunction)
-
-
-
Method Detail
-
builder
public static MockProvider.Builder builder()
-
withResponse
public static MockProvider withResponse(java.lang.String text)
-
withResponse
public static MockProvider withResponse(AIResponse response)
-
withResponseFunction
public static MockProvider withResponseFunction(java.util.function.Function<MockProviderRequest,AIResponse> responseFunction)
-
getName
public java.lang.String getName()
- Specified by:
getNamein interfaceAIProvider
-
getHeaders
public java.util.Map<java.lang.String,java.lang.String> getHeaders(java.lang.String credential)
- Specified by:
getHeadersin interfaceAIProvider
-
getProviderClientType
public java.lang.Class<? extends AIProviderClient<?>> getProviderClientType()
- Specified by:
getProviderClientTypein interfaceAIProvider
-
isMimeTypeSupported
public boolean isMimeTypeSupported(MimeType mimeType)
- Specified by:
isMimeTypeSupportedin interfaceAIProvider
-
getApiKeyVarName
public java.lang.String getApiKeyVarName()
- Specified by:
getApiKeyVarNamein interfaceAIProvider
-
enqueueResponse
public void enqueueResponse(java.lang.String text)
-
enqueueResponse
public void enqueueResponse(AIResponse response)
-
setDefaultResponse
public void setDefaultResponse(java.lang.String text)
-
setDefaultResponse
public void setDefaultResponse(AIResponse response)
-
setResponseFunction
public void setResponseFunction(java.util.function.Function<MockProviderRequest,AIResponse> responseFunction)
-
setStreamTokenDelayMillis
public void setStreamTokenDelayMillis(long millis)
-
getRecordedRequests
public java.util.List<MockProviderRequest> getRecordedRequests()
-
getLastRequest
public MockProviderRequest getLastRequest()
-
getRequestCount
public int getRequestCount()
-
clearRecordedRequests
public void clearRecordedRequests()
-
reset
public void reset()
-
toString
public java.lang.String toString()
- Overrides:
toStringin classjava.lang.Object
-
-