feat: claudeclient adjustment (#850)

XiaoMi · Jun 27, 2024 · 048dcc6 · 048dcc6
2 parents 2dcd120 + 9e990c8
commit 048dcc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jcommon/ai/google/src/main/java/run/mone/ai/google/CloudeClient.java b/jcommon/ai/google/src/main/java/run/mone/ai/google/CloudeClient.java
@@ -40,9 +40,9 @@ public class CloudeClient {
 
 
     @SneakyThrows
-    public String token() {
+    public String token(String model) {
         GoogleCredentials credentials = GoogleCredentials.fromStream(
-                        new FileInputStream("/tmp/key.json"))
+                        new FileInputStream("/tmp/key-"+model+".json"))
                 .createScoped(Collections.singleton("https://" + googleUrl + "/auth/cloud-platform"));
         // Use the credentials to authenticate and generate an access token
         credentials.refreshIfExpired();