目录
调用GPT-4进行单轮对话 调用GPT3.5 调用llama2-70B（百度接口） 调用chatGML 调用ERNIE-BOT 4.0关于如何开通GPT-API服务，请参考这一篇文章
目前仅使用到单轮对话，记录一下，之后拓展需求继续在该篇博客补充
调用GPT-4进行单轮对话
（调用API需科学上网，科学上网之后还是报connectionerror之类的，就是使用的节点的问题，网络问题）
import openai import openai.error openai.log = None openai.api_key = "sk-******" def get_gpt4_response(content, max_retries=10): messages = [{"role": "user", "content": content}] for attempt in range(max_retries): try: response = openai.ChatCompletion.create( model="gpt-4-1106-preview", messages=messages, temperature = 0.95, ) return response.choices[0].message.content except openai.error.APIConnectionError as e: print("再次请求中") continue return "请求失败：超过最大重试次数" get_gpt4_response(content='山东的简称是啥，一个字回答我') 输出：鲁
调用GPT3.5
（调用API需科学上网，科学上网之后还是报connectionerror之类的，就是使用的节点的问题，网络问题）
import openai import openai.error def query_gpt3(prompt): openai.api_key = 'sk-**********' response = openai.ChatCompletion.create( model="gpt-3.5-turbo", model messages=[{"role": "user", "content": prompt}] ) return response.choices[0].message['content'].strip() # 要询问的问题 prompt = "你是gpt3.5还是gpt4？" # 获取GPT-3的回答 answer = query_gpt3(prompt) print("GPT-3 Answer:", answer) 调用llama2-70B（百度接口）
申请入口：Llama-2-70b-chat
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 main(): url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/llama_2_7b?access_token=" + get_access_token() payload = json.dumps({ "messages": [ { "role": "user", "content": "please introduce yourself" } ] }) headers = { 'Content-Type': 'application/json' } response = requests.request("POST", url, headers=headers, data=payload) print(response.text) if __name__ == '__main__': main() 调用chatGML
申请入口：智谱AI大模型MaaS开放平台
import zhipuai zhipuai.api_key = "********" response = zhipuai.model_api.sse_invoke( model="chatglm_turbo", prompt=[ {"role": "user", "content": "请介绍一下你自己"} ], temperature=0.95, top_p=0.7, incremental=True ) for event in response.events(): if event.event == "add": print(event.data) elif event.event == "error" or event.event == "interrupted": print(event.data) elif event.event == "finish": print(event.data) print(event.meta) else: print(event.data) 调用ERNIE-BOT 4.0
申请入口：ERNIE-BOT 4.0
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=****" response = requests.post(url) return response.json().get("access_token") def call_wenxin_api(content): access_token = get_access_token() url = f"https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro?access_token={access_token}" payload = json.dumps({ "messages": [ { "role": "user", "content": content }, ] }) headers = { 'Content-Type': 'application/json' } response = requests.post(url, headers=headers, data=payload) response_json = response.json() # 直接获取 result 的值 return response_json.get("result") call_wenxin_api('山东的简称是什么？') 输出：山东的简称是“鲁”
