dev.langchain4j.model.huggingface

Class HuggingFaceLanguageModel

java.lang.Object

dev.langchain4j.model.huggingface.HuggingFaceLanguageModel

All Implemented Interfaces:

LanguageModel

public class HuggingFaceLanguageModel
extends Object
implements LanguageModel

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	HuggingFaceLanguageModel.Builder

Constructor Summary

Constructors

Constructor and Description
HuggingFaceLanguageModel(HuggingFaceLanguageModel.Builder builder)
HuggingFaceLanguageModel(String accessToken, String modelId, Duration timeout, Double temperature, Integer maxNewTokens, Boolean returnFullText, Boolean waitForModel)

Method Summary

Modifier and Type	Method and Description
static HuggingFaceLanguageModel.Builder	builder()
Response<String>	generate(String prompt)
static HuggingFaceLanguageModel	withAccessToken(String accessToken)

Methods inherited from class java.lang.Object

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

Methods inherited from interface dev.langchain4j.model.language.LanguageModel

generate

Constructor Detail

HuggingFaceLanguageModel

public HuggingFaceLanguageModel(String accessToken,
                                String modelId,
                                Duration timeout,
                                Double temperature,
                                Integer maxNewTokens,
                                Boolean returnFullText,
                                Boolean waitForModel)

HuggingFaceLanguageModel

public HuggingFaceLanguageModel(HuggingFaceLanguageModel.Builder builder)

Method Detail

generate

public Response<String> generate(String prompt)

Specified by:

generate in interface LanguageModel

builder

public static HuggingFaceLanguageModel.Builder builder()

withAccessToken

public static HuggingFaceLanguageModel withAccessToken(String accessToken)