claude

LunaTranslator/LunaTranslator/translator/claude.py

@@ -0,0 +1,131 @@
+from traceback import print_exc
+import json
+from translator.basetranslator import basetrans
+
+
+class TS(basetrans):
+    def langmap(self):
+        return {
+            "zh": "Simplified Chinese",
+            "ja": "Japanese",
+            "en": "English",
+            "ru": "Russian",
+            "es": "Spanish",
+            "ko": "Korean",
+            "fr": "French",
+            "cht": "Traditional Chinese",
+            "vi": "Vietnamese",
+            "tr": "Turkish",
+            "pl": "Polish",
+            "uk": "Ukrainian",
+            "it": "Italian",
+            "ar": "Arabic",
+            "th": "Thai",
+        }
+
+    def __init__(self, typename):
+        self.context = []
+        super().__init__(typename)
+
+    def inittranslator(self):
+        self.api_key = None
+
+    def checkv1(self, api_url):
+        if api_url[-4:] == "/v1/":
+            api_url = api_url[:-1]
+        elif api_url[-3:] == "/v1":
+            pass
+        elif api_url[-1] == "/":
+            api_url += "v1"
+        else:
+            api_url += "/v1"
+        return api_url
+
+    def translate(self, query):
+        self.contextnum = int(self.config["附带上下文个数"])
+
+        try:
+            temperature = float(self.config["Temperature"])
+        except:
+            temperature = 0.3
+
+        if self.config["使用自定义promt"]:
+            system = self.config["自定义promt"]
+        else:
+            system = "You are a translator, translate from {} to {}".format(
+                self.srclang, self.tgtlang
+            )
+        message = []
+        for _i in range(min(len(self.context) // 2, self.contextnum)):
+            i = (
+                len(self.context) // 2
+                - min(len(self.context) // 2, self.contextnum)
+                + _i
+            )
+            message.append(self.context[i * 2])
+            message.append(self.context[i * 2 + 1])
+        message.append({"role": "user", "content": query})
+
+        headers = {
+            "anthropic-version": "2023-06-01",
+            "accept": "application/json",
+            "anthropic-version": "2023-06-01",
+            "content-type": "application/json",
+        }
+        if len(self.config["API_KEY"]):
+            headers.update({"X-Api-Key": self.config["API_KEY"]})
+        # elif len(self.config["AUTH_TOKEN"]):
+        #    headers.update({"Authorization": f'Bearer {self.config["AUTH_TOKEN"]}'})
+        else:
+            self.checkempty(["API_KEY"])  # , "AUTH_TOKEN"])
+
+        usingstream = self.config["流式输出"]
+        data = dict(
+            model=self.config["model"],
+            messages=message,
+            system=system,
+            max_tokens=self.config["max_tokens"],
+            temperature=temperature,
+            stream=usingstream,
+        )
+        response = self.session.post(
+            self.checkv1(self.config["BASE_URL"]) + "/messages",
+            headers=headers,
+            json=data,
+            stream=usingstream,
+        )
+        if usingstream:
+            # https://docs.anthropic.com/claude/reference/messages-streaming
+            message = ""
+            for chunk in response.iter_lines():
+                response_data = chunk.decode("utf-8").strip()
+                if not response_data:
+                    continue
+                if response_data.startswith("data: "):
+                    try:
+                        json_data = json.loads(response_data[6:])
+                        if json_data["type"] == "message_stop":
+                            break
+                        elif json_data["type"] == "content_block_delta":
+                            msg = json_data["delta"]["text"]
+                            yield msg
+                            message += msg
+                        elif json_data["type"] == "content_block_start":
+                            msg = json_data["content_block"]["text"]
+                            yield msg
+                            message += msg
+                    except:
+                        print_exc()
+                        raise Exception(response_data)
+
+        else:
+            # https://docs.anthropic.com/claude/reference/messages_post
+            try:
+                message = (
+                    response.json()["content"][0]["text"].replace("\n\n", "\n").strip()
+                )
+                yield message
+            except:
+                raise Exception(response.text)
+        self.context.append({"role": "user", "content": query})
+        self.context.append({"role": "assistant", "content": message})

LunaTranslator/files/defaultconfig/config.json

@@ -1105,6 +1105,12 @@
             "name": "ChatGPT",
             "is_gpt_like":true
         },
+        "claude": {
+            "type": "api",
+            "use": false,
+            "color": "blue",
+            "name": "claude"
+        },
         "chatgpt-3rd-party": {
             "type": "api",
             "use": false,

LunaTranslator/files/defaultconfig/translatorsetting.json

@@ -256,7 +256,46 @@
 				"step":0.1
 			}
 		}
-    },
+    },"claude": {
+		"args": {
+		  "BASE_URL":"https://api.anthropic.com",
+		  "model":"claude-3-opus-20240229",
+		  "API_KEY": "",
+		  "max_tokens":2048,
+		  "Temperature": 0.3,
+		  "附带上下文个数":0,
+		  "使用自定义promt":false,
+		  "自定义promt":"",
+		  "流式输出":false
+		}
+		,
+		  "argstype":{
+			  "流式输出":{
+				  "type": "switch"
+			  },
+			  "使用自定义promt":{
+				  "type":"switch"
+			  },
+			  "max_tokens":{
+				  "type":"intspin",
+				  "min":0,
+				  "max":1000000,
+				  "step":1
+			  },
+			  "附带上下文个数":{
+				  "type":"intspin",
+				  "min":0,
+				  "max":10,
+				  "step":1
+			  },
+			  "Temperature":{
+				  "type":"spin",
+				  "min":0,
+				  "max":1,
+				  "step":0.1
+			  }
+		  }
+	  },
 	"chatgpt-3rd-party": {
       "args": {
 		"使用说明":"oneapi/gptgod/...",