Skip to content

Commit c206fd4

Browse files
authored
fix(ai): remove deprecated Hugging Face client (#14442)
1 parent 1d0a3d1 commit c206fd4

File tree

3 files changed

+4
-14
lines changed

3 files changed

+4
-14
lines changed

jablib/build.gradle.kts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,6 @@ dependencies {
166166
implementation("dev.langchain4j:langchain4j-open-ai")
167167
implementation("dev.langchain4j:langchain4j-mistral-ai")
168168
implementation("dev.langchain4j:langchain4j-google-ai-gemini")
169-
implementation("dev.langchain4j:langchain4j-hugging-face")
170169
implementation("dev.langchain4j:langchain4j-http-client")
171170
implementation("dev.langchain4j:langchain4j-http-client-jdk")
172171

jablib/src/main/java/org/jabref/logic/ai/chatting/model/JabRefChatLanguageModel.java

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
import dev.langchain4j.model.chat.ChatModel;
1919
import dev.langchain4j.model.chat.response.ChatResponse;
2020
import dev.langchain4j.model.googleai.GoogleAiGeminiChatModel;
21-
import dev.langchain4j.model.huggingface.HuggingFaceChatModel;
2221
import dev.langchain4j.model.mistralai.MistralAiChatModel;
2322

2423
/**
@@ -60,7 +59,9 @@ private void rebuild() {
6059
}
6160

6261
switch (aiPreferences.getAiProvider()) {
63-
case OPEN_AI ->
62+
// Hugging Face uses OpenAI API.
63+
case OPEN_AI,
64+
HUGGING_FACE ->
6465
langchainChatModel = Optional.of(new JvmOpenAiChatLanguageModel(aiPreferences, httpClient));
6566

6667
case GPT4ALL ->
@@ -87,16 +88,6 @@ private void rebuild() {
8788
.logRequestsAndResponses(true)
8889
.build()
8990
);
90-
91-
case HUGGING_FACE -> // NOTE: {@link HuggingFaceChatModel} doesn't support API base url.
92-
langchainChatModel = Optional.of(HuggingFaceChatModel
93-
.builder()
94-
.accessToken(apiKey)
95-
.modelId(aiPreferences.getSelectedChatModel())
96-
.temperature(aiPreferences.getTemperature())
97-
.timeout(Duration.ofMinutes(2))
98-
.build()
99-
);
10091
}
10192
}
10293

jablib/src/main/java/org/jabref/model/ai/AiProvider.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ public enum AiProvider implements Serializable {
66
OPEN_AI("OpenAI (or API compatible)", "https://api.openai.com/v1", "https://openai.com/policies/privacy-policy/"),
77
MISTRAL_AI("Mistral AI", "https://api.mistral.ai/v1", "https://mistral.ai/terms/#privacy-policy"),
88
GEMINI("Gemini", "https://generativelanguage.googleapis.com/v1beta/", "https://ai.google.dev/gemini-api/terms"),
9-
HUGGING_FACE("Hugging Face", "https://huggingface.co/api", "https://huggingface.co/privacy"),
9+
HUGGING_FACE("Hugging Face", "https://router.huggingface.co/v1", "https://huggingface.co/privacy"),
1010
GPT4ALL("GPT4All", "http://localhost:4891/v1", "https://www.nomic.ai/gpt4all/legal/privacy-policy");
1111

1212
private final String label;

0 commit comments

Comments
 (0)