add original flag to cli

README.md

@@ -12,12 +12,14 @@ A simple translate api that proxies to openai written by Golang.
 
 2. Build by `make build`.
 3. Start server by `./translate`. Try it with:
-  ```bash
-  curl -X POST \
-  -d '{"content": "Hi, how are you?", "lang": "Chinese", "model": "gpt-3.5-turbo"}' \
-  http://localhost:8080/translate
-  ```
-4. translate-cli is a way to test translate endpoint, start translate server before using the cli tool.
+
+```bash
+curl -X POST \
+-d '{"content": "Hi, how are you?", "lang": "Chinese", "model": "gpt-3.5-turbo"}' \
+http://localhost:8080/translate
+```
+
+4. translate-cli is a way to test translate endpoints, start translate server before using the cli tool.
    Try it like `./translate-cli -c "content" -l "Chinese" -m "gpt-3.5-turbo"`, -l and -m are optional.
    The endpoint is http://localhost:8080 by default, customize it by setting TRANSLATE_ENDPOINT env var.
 

translate-cli.go

@@ -13,6 +13,8 @@ var endpoint = String(os.Getenv("TRANSLATE_ENDPOINT")).
 	orElse("http://localhost:8080/translate").
 	get()
 
+var original = false
+
 func parseFlags() Ask {
 	var (
 		content string
@@ -22,6 +24,7 @@ func parseFlags() Ask {
 	flag.StringVar(&content, "c", "", "The content to be translated")
 	flag.StringVar(&lang, "l", "English", "The language to be translated")
 	flag.StringVar(&model, "m", "gpt-3.5-turbo", "The chatGPT model to be chose")
+	flag.BoolVar(&original, "o", false, "Set to get original OpenAI response")
 	flag.Parse()
 
 	payload := Ask{
@@ -34,6 +37,10 @@ func parseFlags() Ask {
 
 func translate(ask Ask) Answer {
 	_bytes, _ := json.Marshal(ask)
+	fmt.Println("original", original)
+	if original {
+		endpoint = "http://localhost:8080/translate/openai"
+	}
 	resp, _ := http.Post(endpoint, "application/json", bytes.NewBuffer(_bytes))
 	defer resp.Body.Close()