Package dev.enola.ai.adk.test

Class MockModel

java.lang.Object
com.google.adk.models.BaseLlm
dev.enola.ai.adk.test.MockModel
public class MockModel extends com.google.adk.models.BaseLlm

  • Field Summary

    Fields
    Modifier and Type
    Field
    Description
    protected Supplier<io.reactivex.rxjava3.core.Flowable<com.google.adk.models.LlmResponse>>
    responsesSupplier
     

  • Constructor Summary

    Constructors
    Constructor
    Description
    MockModel(String reply)
     

  • Method Summary

    Modifier and Type
    Method
    Description
    com.google.adk.models.BaseLlmConnection
    connect(com.google.adk.models.LlmRequest llmRequest)
     
    io.reactivex.rxjava3.core.Flowable<com.google.adk.models.LlmResponse>
    generateContent(com.google.adk.models.LlmRequest llmRequest, boolean stream)
     
    com.google.adk.models.LlmRequest
    getLastRequest()
     
    ImmutableList<com.google.adk.models.LlmRequest>
    getRequests()
     

    Methods inherited from class com.google.adk.models.BaseLlm

    model

    Methods inherited from class java.lang.Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

  • Field Details

    • responsesSupplier

      protected Supplier<io.reactivex.rxjava3.core.Flowable<com.google.adk.models.LlmResponse>> responsesSupplier

  • Constructor Details

  • Method Details

    • generateContent

      public io.reactivex.rxjava3.core.Flowable<com.google.adk.models.LlmResponse> generateContent(com.google.adk.models.LlmRequest llmRequest, boolean stream)
      Specified by:
      generateContent in class com.google.adk.models.BaseLlm

    • connect

      public com.google.adk.models.BaseLlmConnection connect(com.google.adk.models.LlmRequest llmRequest)
      Specified by:
      connect in class com.google.adk.models.BaseLlm

    • getRequests

      public ImmutableList<com.google.adk.models.LlmRequest> getRequests()

    • getLastRequest

      public com.google.adk.models.LlmRequest getLastRequest()