Spaces:
Running
Running
File size: 1,708 Bytes
ae239a7 fb8c051 ae239a7 a0d1776 fb8c051 ae239a7 fb8c051 c9efba3 fb8c051 ae239a7 d18c569 ae239a7 fb8c051 ae239a7 d18c569 ae239a7 d18c569 ae239a7 9d9ac18 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import os
import openai
import logging
import requests
log = logging.getLogger(__name__)
def get_gpt_responses(systems, prompts, model="gpt-4", temperature=0.4):
conversation_history = [
{"role": "system", "content": systems},
{"role": "user", "content": prompts}
]
response = openai.ChatCompletion.create(
model=model,
messages=conversation_history,
n=1, # Number of responses you want to generate
temperature=temperature, # Controls the creativity of the generated response
)
assistant_message = response['choices'][0]["message"]["content"]
usage = response['usage']
log.info(assistant_message)
return assistant_message, usage
def get_gpt_responses_test(systems, prompts, model="gpt-4", temperature=0.4, base_url=None, key=None):
end_point = r"/v1/completions"
if base_url is None:
base_url = r"https://api.openai.com" + end_point
if key is None:
key = os.getenv("OPENAI_API_KEY")
url = base_url + end_point
headers = {
'Content-Type': 'application/json',
'Authorization': f'Bearer {key}' # <-- 把 fkxxxxx 替换成你自己的 Forward Key，注意前面的 Bearer 要保留，并且和 Key 中间有一个空格。
}
message = [{"role": "system", "content": systems},
{"role": "user", "content": prompts}]
data = {
"model": model,
"message": message,
"temperature": temperature
}
print(data)
response = requests.post(url, headers=headers, json=data)
print(response)
response = response.json()
return response['choices'][0]["message"]["content"]
if __name__ == "__main__":
pass
|