Class TraceableModelClient<T>

java.lang.Object
ai.driftkit.common.domain.client.ModelClient<T>
ai.driftkit.clients.core.TraceableModelClient<T>
public class TraceableModelClient<T> extends ai.driftkit.common.domain.client.ModelClient<T>

  • Constructor Details

    • TraceableModelClient

      public TraceableModelClient(ai.driftkit.common.domain.client.ModelClient<T> delegate)

    • TraceableModelClient

      public TraceableModelClient(ai.driftkit.common.domain.client.ModelClient<T> delegate, ai.driftkit.common.utils.Tokenizer tokenizer)

  • Method Details

    • init

      public ai.driftkit.common.domain.client.ModelClient<T> init(ai.driftkit.config.EtlConfig.VaultConfig config)

    • getCapabilities

      public Set<ai.driftkit.common.domain.client.ModelClient.Capability> getCapabilities()
      Specified by:
      getCapabilities in class ai.driftkit.common.domain.client.ModelClient<T>

    • textToText

      public ai.driftkit.common.domain.client.ModelTextResponse textToText(ai.driftkit.common.domain.client.ModelTextRequest prompt) throws ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException
      Overrides:
      textToText in class ai.driftkit.common.domain.client.ModelClient<T>
      Throws:
      ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException

    • textToImage

      public ai.driftkit.common.domain.client.ModelImageResponse textToImage(ai.driftkit.common.domain.client.ModelImageRequest prompt) throws ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException
      Overrides:
      textToImage in class ai.driftkit.common.domain.client.ModelClient<T>
      Throws:
      ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException

    • imageToText

      public ai.driftkit.common.domain.client.ModelTextResponse imageToText(ai.driftkit.common.domain.client.ModelTextRequest prompt) throws ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException
      Overrides:
      imageToText in class ai.driftkit.common.domain.client.ModelClient<T>
      Throws:
      ai.driftkit.common.domain.client.ModelClient.UnsupportedCapabilityException

    • getWorkflow

      public T getWorkflow()
      Overrides:
      getWorkflow in class ai.driftkit.common.domain.client.ModelClient<T>

    • setWorkflow

      public void setWorkflow(T workflow)
      Overrides:
      setWorkflow in class ai.driftkit.common.domain.client.ModelClient<T>

    • getModel

      public String getModel()
      Overrides:
      getModel in class ai.driftkit.common.domain.client.ModelClient<T>

    • setModel

      public void setModel(String model)
      Overrides:
      setModel in class ai.driftkit.common.domain.client.ModelClient<T>

    • getSystemMessages

      public List<String> getSystemMessages()
      Overrides:
      getSystemMessages in class ai.driftkit.common.domain.client.ModelClient<T>

    • setSystemMessages

      public void setSystemMessages(List<String> systemMessages)
      Overrides:
      setSystemMessages in class ai.driftkit.common.domain.client.ModelClient<T>

    • getTemperature

      public Double getTemperature()
      Overrides:
      getTemperature in class ai.driftkit.common.domain.client.ModelClient<T>

    • setTemperature

      public void setTemperature(Double temperature)
      Overrides:
      setTemperature in class ai.driftkit.common.domain.client.ModelClient<T>

    • getTopP

      public Double getTopP()
      Overrides:
      getTopP in class ai.driftkit.common.domain.client.ModelClient<T>

    • setTopP

      public void setTopP(Double topP)
      Overrides:
      setTopP in class ai.driftkit.common.domain.client.ModelClient<T>

    • getStop

      public List<String> getStop()
      Overrides:
      getStop in class ai.driftkit.common.domain.client.ModelClient<T>

    • setStop

      public void setStop(List<String> stop)
      Overrides:
      setStop in class ai.driftkit.common.domain.client.ModelClient<T>

    • isJsonObjectSupport

      public boolean isJsonObjectSupport()
      Overrides:
      isJsonObjectSupport in class ai.driftkit.common.domain.client.ModelClient<T>

    • setJsonObjectSupport

      public void setJsonObjectSupport(boolean jsonObjectSupport)
      Overrides:
      setJsonObjectSupport in class ai.driftkit.common.domain.client.ModelClient<T>

    • getLogprobs

      public Boolean getLogprobs()
      Overrides:
      getLogprobs in class ai.driftkit.common.domain.client.ModelClient<T>

    • setLogprobs

      public void setLogprobs(Boolean logprobs)
      Overrides:
      setLogprobs in class ai.driftkit.common.domain.client.ModelClient<T>

    • getTopLogprobs

      public Integer getTopLogprobs()
      Overrides:
      getTopLogprobs in class ai.driftkit.common.domain.client.ModelClient<T>

    • setTopLogprobs

      public void setTopLogprobs(Integer topLogprobs)
      Overrides:
      setTopLogprobs in class ai.driftkit.common.domain.client.ModelClient<T>

    • getMaxTokens

      public Integer getMaxTokens()
      Overrides:
      getMaxTokens in class ai.driftkit.common.domain.client.ModelClient<T>

    • setMaxTokens

      public void setMaxTokens(Integer maxTokens)
      Overrides:
      setMaxTokens in class ai.driftkit.common.domain.client.ModelClient<T>

    • getMaxCompletionTokens

      public Integer getMaxCompletionTokens()
      Overrides:
      getMaxCompletionTokens in class ai.driftkit.common.domain.client.ModelClient<T>

    • setMaxCompletionTokens

      public void setMaxCompletionTokens(Integer maxCompletionTokens)
      Overrides:
      setMaxCompletionTokens in class ai.driftkit.common.domain.client.ModelClient<T>