Skip to content

Commit e703bb5

Browse files
committed
Fix Refresh Models action for Gemini model family
1 parent a2ed004 commit e703bb5

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

src/main/java/com/didalgo/intellij/chatgpt/chat/models/GeminiModelFamily.java

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,15 @@
1111

1212
public class GeminiModelFamily implements ModelFamily {
1313

14-
private static final String DEFAULT_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai";
14+
public static final String STANDARD_BASE_URL = "https://generativelanguage.googleapis.com/v1beta";
15+
16+
public static final String OPENAI_COMPATIBLE_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai";
1517

1618
@Override
1719
public OpenAiChatModel createChatModel(GeneralSettings.AssistantOptions config) {
1820
var baseUrl = config.isEnableCustomApiEndpointUrl()? config.getApiEndpointUrl(): getDefaultApiEndpointUrl();
1921
if ("https://generativelanguage.googleapis.com".equals(baseUrl)) {
20-
baseUrl = DEFAULT_BASE_URL;
22+
baseUrl = OPENAI_COMPATIBLE_BASE_URL;
2123
}
2224

2325
var api = OpenAiApi.builder()
@@ -38,7 +40,7 @@ public OpenAiChatModel createChatModel(GeneralSettings.AssistantOptions config)
3840

3941
@Override
4042
public String getDefaultApiEndpointUrl() {
41-
return DEFAULT_BASE_URL;
43+
return OPENAI_COMPATIBLE_BASE_URL;
4244
}
4345

4446
@Override

src/main/java/com/didalgo/intellij/chatgpt/chat/models/api/ModelCatalogApi.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import java.util.function.Predicate;
1313

1414
import com.didalgo.intellij.chatgpt.chat.models.CustomModel;
15+
import com.didalgo.intellij.chatgpt.chat.models.GeminiModelFamily;
1516
import com.didalgo.intellij.chatgpt.chat.models.ModelFamily;
1617
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
1718
import com.fasterxml.jackson.annotation.JsonProperty;
@@ -66,7 +67,8 @@ public List<CustomModel> getAvailableModels(ModelFamily family, String baseUrl,
6667
private Function<UriBuilder, URI> getApiEndpointSpec(ModelFamily family) {
6768
String path;
6869
if (family == ModelFamily.GEMINI) {
69-
path = "/v1beta/models";
70+
//path = "/v1beta/models";
71+
return __ -> URI.create(GeminiModelFamily.STANDARD_BASE_URL + "/models");
7072
} else if (family == ModelFamily.OLLAMA) {
7173
path = "/api/tags";
7274
} else {

0 commit comments

Comments
 (0)