基础三大模型,需要先去官方注册获得key;后续可以使用下列代码调用
1.腾讯
安装:
pip install -i https://mirrors.tencent.com/pypi/simple/ --upgrade tencentcloud-sdk-python
实例:
import json
import types
from tencentcloud.common import credential
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.hunyuan.v20230901 import hunyuan_client, models
try:
cred = credential.Credential("你的SecretId", "你的SecretKey")
httpProfile = HttpProfile()
httpProfile.endpoint = "hunyuan.tencentcloudapi.com"
clientProfile = ClientProfile()
clientProfile.httpProfile = httpProfile
req = models.ChatCompletionsRequest()
params = {
"Model": "hunyuan-lite",
"Messages": [
{
"Role": "user",
"Content": "你好"
}
]
}
req.from_json_string(json.dumps(params))
resp = client.ChatCompletions(req)
if isinstance(resp, types.GeneratorType):
for event in resp:
print(event)
else:
print(resp)
except TencentCloudSDKException as err:
print(err)
2.讯飞
安装:
pip install spark-ai-python
实例:
from sparkai.llm.llm import ChatSparkLLM, ChunkPrintHandler
from sparkai.core.messages import ChatMessage
def get_response(text):
if text is not None:
SPARKAI_URL = 'wss://spark-api.xf-yun.com/v1.1/chat'
SPARKAI_APP_ID = ''
SPARKAI_API_SECRET = ''
SPARKAI_API_KEY = ''
SPARKAI_DOMAIN = 'general'
spark = ChatSparkLLM(spark_api_url=SPARKAI_URL, spark_app_id=SPARKAI_APP_ID, spark_api_key=SPARKAI_API_KEY,
spark_api_secret=SPARKAI_API_SECRET, spark_llm_domain=SPARKAI_DOMAIN, streaming=False)
messages = [ChatMessage(role="user", content=text)]
handler = ChunkPrintHandler()
st = spark.generate([messages], callbacks=[handler])
return st
response = get_response("你好")
print(response)
3.百度:
安装:
pip install requests
import requests
import json
def get_access_token():
url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=&client_secret="
payload = json.dumps("")
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
return response.json().get("access_token")
def get_response(text):
url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/yi_34b_chat?access_token=" + get_access_token()
payload = json.dumps({"messages": [{"role": "user","content": "hi"}]})
headers = {
'Content-Type': 'application/json'
}
response = requests.request("POST", url, headers=headers, data=payload)
print(response.json().get("result"))