调用说明

{
  "model": "mengzi-lite",
  "messages": [
    {
      "role": "user",
      "content": "你知道牛顿吗"
    }
  ],
  "n": 1,
  "max_tokens": 1024,
  "user": ""
}

StringToSign =
    HTTP-Verb + "\n" +      //HTTP-Verb只支持POST
    Accept + “\n” +         //Accept为application/json
    Content-MD5 + "\n" +    //第1步中计算出来的MD5值
    Content-Type + "\n" +   //Content-Type值为application/json
    Date + "\n" +           //Date值为GMT时间
    x-langboat-signature-method + "\n" +  // 只支持 HMAC-SHA256
    x-langboat-signature-nonce + "\n";

POST
application/json
BeJNYUEsjNBAO+IG9kTo3A=
application/json
Wed, 20 Apr 2022 02:50:21 GMT
HMAC-SHA256
53371

a=b&c=d

stringToSign = headerStringToSign + queryToSign;

Signature = Base64(HMAC-SHA256( AccessSecret, UTF-8-Encoding-Of(stringToSign)))
Authorization = AccessKey + ":" + Signature

{
  "model": "mengzi-std",
  "messages": [
    {
      "role": "user",
      "content": "你知道牛顿吗"
    }
  ],
  "temperature": 0.1,
  "top_p": 0.05,
  "n": 1,
  "stream": false,
  "max_tokens": 4096,
  "user": ""
}

POST https://open.langboat.com/chat

Content-Type: application/json
Content-MD5: BeJNYUEsjNBAO+IG9kTo3A==
Date: Tue, 19 Apr 2022 10:03:46 GMT
Accept: application/json
x-langboat-signature-method: HMAC-SHA256
x-langboat-signature-nonce: 43785
Authorization: 7Bo9ByyiTWRC1Y8KJJQ9cWtNpZLmrgyb:po/vsPI0RcvY/eu4bohhxxADHyPj4/rcglLTQEBtHQM=

{
   "id": "chatcmpl-1135c2751ec9c45ec5d525303500d548",
   "object": "chat.completion",
   "created": 1692865424,
   "model": "mengzi-std",
   "choices": [
      {
         "index": 0,
         "message": {
            "role": "assistant",
            "content": "当然了！伊萨克·牛顿（Isaac Newton）是一位伟大的英国物理学家、数学家和自然哲学家，被认为是现代科学之父之一。他最著名的成就是发现了万有引力定律以及三大运动定律，这些理论对我们理解宇宙产生了深远影响。此外，他在光学、微积分等领域也做出了重要贡献。"
         },
         "finish_reason": "stop"
      }
   ],
   "usage": {
      "prompt_tokens": 4,
      "completion_tokens": 99,
      "total_tokens": 103
   }
}

{
  "error": {
    "code": 10403,
    "message": "AccessKey 不存在, 请检查 AccessKey 是否正确",
    "requestId": "6f233f3e276bc1ba2ffb0069df62cd1b"
  }
}

{
  "error": {
    "code": 10403,
    "message": "权限不足，查看是否开通服务",
    "requestId": "6f233f3e276bc1ba2ffb0069df62cd1b"
  }
}

curl 'http://open.langboat.com/chat' \
--header 'Content-MD5: WkwTeY7YOwSSeyB09sF6ow==' \
--header 'x-langboat-signature-method: HMAC-SHA256' \
--header 'x-langboat-signature-nonce: 43785' \
--header 'Authorization: mWve3ivAD1NljBkCj7Pk2eZQQB3PmMt6:MyJ/1mthTM51zHVXOr13+8RhjMAuGIhKttKFr+nb6hs=' \
--header 'Date: Thu, 16 Jun 2022 13:09:33 GMT' \
--header 'Content-Type: application/json' \
--data '{
    "model": "mengzi-std",
    "messages": [
        {
            "role":"user",
            "content":"你知道牛顿吗"
        }
    ],
    "temperature": 1,
    "top_p": 0,
    "n": 1,
    "stream": true,
    "max_tokens": 4096,
    "user": ""
}'

id: 0-0
data: {"id":"chatcmpl-dac45d8d2fd528f46afdda15be6a4e63","object":"chat.completion.chunk","created":1692865308,"model":"mengzi-std","choices":[{"index":0,"delta":{"content":"当然了！伊萨克·牛顿（Isaac Newton）是一位伟大的英国物理学家、数学家和自然哲学家，被认为是现代科学之父之一。"},"finish_reason":null}]}

id: 0-1
data: {"id":"chatcmpl-dac45d8d2fd528f46afdda15be6a4e63","object":"chat.completion.chunk","created":1692865309,"model":"mengzi-std","choices":[{"index":0,"delta":{"content":"他最著名的成就是发现了万有引力定律以及三大运动定律，这些理论对我们理解宇宙产生了深远影响。"},"finish_reason":null}]}

id: 0-2
data: {"id":"chatcmpl-dac45d8d2fd528f46afdda15be6a4e63","object":"chat.completion.chunk","created":1692865309,"model":"mengzi-std","choices":[{"index":0,"delta":{"content":"此外，他在光学、微积分等领域也做出了重要贡献。"},"finish_reason":null}]}

data: [DONE]

curl 'http://open.langboat.com/chat' \
--header 'Content-MD5: WkwTeY7YOwSSeyB09sF6ow==' \
--header 'x-langboat-signature-method: HMAC-SHA256' \
--header 'x-langboat-signature-nonce: 43785' \
--header 'Authorization: mWve3ivAD1NljBkCj7Pk2eZQQB3PmMt6:MyJ/1mthTM51zHVXOr13+8RhjMAuGIhKttKFr+nb6hs=' \
--header 'Date: Thu, 16 Jun 2022 13:09:33 GMT' \
--header 'Content-Type: application/json' \
--data '{
    "model": "mengzi-std",
    "messages": [
        {
            "role":"user",
            "content":"你好"
        }
    ],
    "temperature": 0.1,
    "top_p": 0.05,
    "n": 1,
    "stream": true,
    "max_tokens": 4096,
    "user": ""
}'

id: 0-0
data: {"id":"chatcmpl-dac45d8d2fd528f46afdda15be6a4e63","object":"chat.completion.chunk","created":1692865308,"model":"mengzi-std","choices":[{"index":0,"delta":{"content":"你好，有什么需要帮助的吗？"},"finish_reason":null}]}

data: [DONE]

# -*- coding: utf-8 -*-
import base64
import datetime
import hashlib
import hmac
import json
import random
import requests


class LangboatOpenClient:
    """澜舟开放平台客户端"""

    def __init__(self,
                 access_key: str,
                 access_secret: str,
                 url: str = "https://open.langboat.com"):
        self.access_key = access_key
        self.access_secret = access_secret
        self.url = url

    def _build_header(self, query: dict, data: dict) -> dict:
        accept = "application/json"
        # 1. body MD5 加密
        content_md5 = base64.b64encode(
            hashlib.md5(
                json.dumps(data).encode("utf-8")
            ).digest()
        ).decode()
        content_type = "application/json"
        gmt_format = '%a, %d %b %Y %H:%M:%S GMT'
        date = datetime.datetime.utcnow().strftime(gmt_format)
        signature_method = "HMAC-SHA256"
        signature_nonce = str(random.randint(0, 65535))
        header_string = f"POST\n{accept}\n{content_md5}\n{content_type}\n" \
                        f"{date}\n{signature_method}\n{signature_nonce}\n"

        # 2. 计算 queryToSign
        queries_str = []
        for k, v in sorted(query.items(), key=lambda item: item[0]):
            if isinstance(v, list):
                for i in v:
                    queries_str.append(f"{k}={i}")
            else:
                queries_str.append(f"{k}={v}")
        queries_string = '&'.join(queries_str)
        # 3.计算 stringToSign
        sign_string = header_string + queries_string
        # 4.计算 HMAC-SHA256 + Base64
        secret_bytes = self.access_secret.encode("utf-8")
        # 5.计算签名
        signature = base64.b64encode(
            hmac.new(secret_bytes, sign_string.encode("utf-8"), hashlib.sha256).digest()
        ).decode()
        res = {
            "Content-Type": content_type,
            "Content-MD5": content_md5,
            "Date": date,
            "Accept": accept,
            "X-Langboat-Signature-Method": signature_method,
            "X-Langboat-Signature-Nonce": signature_nonce,
            "Authorization": f"{self.access_key}:{signature}"
        }
        return res

    def inference(self, queries: dict, data: dict) -> (int, dict):
        """
        服务调用: POST https://open.langboat.com/chat
        :param queries: query 参数
        :param data: request body 数据
        :return: response status, response body to json
        """
        headers = self._build_header(queries, data)
        response = requests.post(url=self.url + "/chat", 
                                 headers=headers, params=queries, json=data)
        return response.status_code, response.json()


if __name__ == '__main__':
    _access_key = '<Your access_key>'
    _access_secret = '<Your access_secret>'

    client = LangboatOpenClient(
        access_key=_access_key,
        access_secret=_access_secret
    )

    # 孟子GPT
    _queries = {}
    _data = {
        "model": "mengzi-std",
        "messages": [
            {
                "role": "user",
                "content": "你知道牛顿吗"
            }
        ],
        "n": 1,
        "max_tokens": 1024,
        "user": ""
    }
    status_code, result = client.inference(_queries, _data)
    print("response status:", status_code)
    print("response json:", json.dumps(result, ensure_ascii=False, indent=2))

package main

import (
   "crypto/hmac"
   "crypto/md5"
   "crypto/sha256"
   "encoding/base64"
   "fmt"
   "io/ioutil"
   "log"
   "math/rand"
   "net/http"
   "net/url"
   "sort"
   "strings"
   "time"
)

func main() {
   client := OpenClient{
      baseURL:      "https://open.langboat.com",
      accessKey:    "Your_Access_Key",
      accessSecret: "Your_Access_Secret",
   }
   reqBody := `{"model":"mengzi-std","messages":[{"role":"user","content":"你知道牛顿吗"}],"temperature":1,"top_p":0,"n":1,"stream":false,"max_tokens":1024,"user":""}`
   resp := client.inference(reqBody)
   response, ok := resp.(*http.Response)
   if !ok {
      log.Fatal("fail to convert response")
   }
   body, err := ioutil.ReadAll(response.Body)
   if err != nil {
      log.Fatal("fail to read response body")
   }
   log.Println(string(body))
}

type OpenClient struct {
   baseURL      string
   accessKey    string
   accessSecret string
}

func (c *OpenClient) inference(body string) interface{} {
   targetURL := c.baseURL + "/chat"
   client := &http.Client{
      Timeout: 60 * time.Second,
   }
   // 构造header
   var (
      payload = strings.NewReader(body)
      date    = time.Now().UTC().Format(http.TimeFormat)
      nonce   = fmt.Sprint(10000 + rand.Intn(89999))
   )
   //  签名
   signParam := SignParam{
      Body:    body,
      DateGMT: date,
      Nonce:   nonce,
   }
   contentMD5, signature := GenSignature(signParam, c.accessSecret)
   // 设置header
   headers := map[string]string{
      "Authorization":               c.accessKey + ":" + signature,
      "Content-Type":                "application/json",
      "Accept":                      "application/json",
      "Date":                        date,
      "Content-MD5":                 contentMD5,
      "x-langboat-signature-method": "HMAC-SHA256",
      "x-langboat-signature-nonce":  nonce,
   }

   req, _ := http.NewRequest("POST", targetURL, payload)
   for k, v := range headers {
      req.Header.Add(k, v)
   }
   resp, err := client.Do(req)
   if err != nil {
      log.Println(err.Error())
   }

   return resp
}

// SignParam 生成签名需要的参数
type SignParam struct {
   Body    string // body数据
   Query   string // 原始query
   DateGMT string // GTM时间
   Nonce   string // 随机数
}

func getMD5(str string) []byte {
   h := md5.New()
   h.Write([]byte(str))
   return h.Sum(nil)
}

func hmacSha256(data string, secret string) []byte {
   h := hmac.New(sha256.New, []byte(secret))
   h.Write([]byte(data))
   return h.Sum(nil)
}

func resortQuery(src string) string {
   queries, _ := url.ParseQuery(src)
   keys := make([]string, 0)
   for k := range queries {
      keys = append(keys, k)
   }
   sort.Strings(keys)
   newQuery := url.Values{}
   for _, k := range keys {
      for _, value := range queries[k] {
         newQuery.Add(k, value)
      }
   }
   return newQuery.Encode()
}

// GenSignature 生成签名
func GenSignature(src SignParam, accessSecret string) (string, string) {
   // 计算body的md5值
   md5str := getMD5(src.Body)
   // base64后得到contentMD5
   contentMD5 := base64.StdEncoding.EncodeToString(md5str)
   // query解析，并按照字典序重新排列
   query := resortQuery(src.Query)
   query, _ = url.QueryUnescape(query)
   // 需要做签名的字符串结构
   stringToSign := `POST
application/json
%s
application/json
%s
HMAC-SHA256
%s
%s`
   stringToSign = fmt.Sprintf(stringToSign, contentMD5, src.DateGMT, src.Nonce, query)
   hmac256 := hmacSha256(stringToSign, accessSecret)
   signature := base64.StdEncoding.EncodeToString(hmac256)
   return contentMD5, signature
}