import openai
import random
import io
import sys
import requests
import json
# 改變標(biāo)準(zhǔn)輸出的默認(rèn)編碼
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding=’utf8’)
# 示例用法
api_key = "****************************************************" # 替換為你的 API 密鑰
openai.api_key = api_key
class Leader:
def __init__(self):
self.task = None
def create_task(self, prompt):
headers = {
’Authorization’: f’Bearer {api_key}’,
’Content-Type’: ’application/json’
}
data = {
"model": "gpt-4",
"messages": [
{"role": "system", "content": "你是一個(gè)非常有用的智能助手."},
{"role": "user", "content": prompt}
]
}
response = requests.post(
’https://api.openai.com/v1/chat/completions’,
headers=headers,
data=json.dumps(data)
)
if response.status_code == 200:
return response.json()[’choices’][0][’message’][’content’]
else:
return f"Error: {response.text}"
class Agent:
def __init__(self, name):
self.name = name
def complete_task(self, task):
completion_prompt = f"{self.name} 需要一個(gè)有效的解決方案為了這個(gè)任務(wù): {task}"
headers = {
’Authorization’: f’Bearer {api_key}’,
’Content-Type’: ’application/json’
}
data = {
"model": "gpt-4",
"messages": [
{"role": "system", "content": "你是一個(gè)非常有用的智能助手."},
{"role": "user", "content": completion_prompt}
]
}
response = requests.post(
’https://api.openai.com/v1/chat/completions’,
headers=headers,
data=json.dumps(data)
)
if response.status_code == 200:
solution=response.json()[’choices’][0][’message’][’content’]
else:
solution=f"Error: {response.text}"
print(f"\n\n{self.name} 完成了任務(wù): ")
print(f"解決方案: {solution}")
return solution
class Evaluator:
def evaluate(self, task, solutions):
prompt = f"對(duì)以下任務(wù){task}的兩種解決方案進(jìn)行評(píng)分: 方案1: {solutions[0]}，方案2: {solutions[1]}"
headers = {
’Authorization’: f’Bearer {api_key}’,
’Content-Type’: ’application/json’
}
data = {
"model": "gpt-4",
"messages": [
{"role": "system", "content": "你是一個(gè)非常有用的智能助手."},
{"role": "user", "content": prompt}
]
}
response = requests.post(
’https://api.openai.com/v1/chat/completions’,
headers=headers,
data=json.dumps(data)
)
if response.status_code == 200:
return response.json()[’choices’][0][’message’][’content’]
else:
return f"Error: {response.text}"
# 示例執(zhí)行流程
leader = Leader()
agent1 = Agent("B1")
agent2 = Agent("B2")
evaluator = Evaluator()
task_prompt = "初中階段有效提高學(xué)生英語(yǔ)聽力的方法"
task = leader.create_task(task_prompt)
print(f"任務(wù): {task}")
solution1 = agent1.complete_task(task)
solution2 = agent2.complete_task(task)
scores = evaluator.evaluate(task, [solution1, solution2])
print(f"\n\n得分: {scores}")
業(yè)務(wù)實(shí)施流程