Package 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	Description
static final class	HuggingFaceLanguageModel.Builder

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
static HuggingFaceLanguageModel.Builder	builder()
Response<String>	generate(String prompt)	Generate a response to the given 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 Details

HuggingFaceLanguageModel

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

HuggingFaceLanguageModel

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

HuggingFaceLanguageModel

public HuggingFaceLanguageModel(HuggingFaceLanguageModel.Builder builder)

Method Details

generate

public Response<String> generate(String prompt)

Description copied from interface: LanguageModel

Generate a response to the given prompt.

Specified by:

generate in interface LanguageModel

Parameters:

prompt - the prompt.

Returns:

the response.

builder

public static HuggingFaceLanguageModel.Builder builder()

withAccessToken

public static HuggingFaceLanguageModel withAccessToken(String accessToken)