diff --git a/agent/agent.py b/agent/agent.py index ab0a2f2..8866db5 100644 --- a/agent/agent.py +++ b/agent/agent.py @@ -37,7 +37,8 @@ class Agent: response = openai.chat.completions.create( model="gpt-3.5-turbo", # GPT-3.5 model messages=[ - {"role": "user", "content": self.role +","+ input_text} + {"role": "system", "content": self.role}, + {"role": "user", "content": input_text} ], temperature=0.2 ) diff --git a/agent/manager_agent.py b/agent/manager_agent.py index b2abba0..7273f17 100644 --- a/agent/manager_agent.py +++ b/agent/manager_agent.py @@ -2,4 +2,4 @@ from agent.agent import Agent class ManagerAgent(Agent): def __init__(self): - super().__init__("你是一名总经理,负责与用户沟通需求,制定OKR中的O,并对下属的工作成果进行评估") + super().__init__("你是一名总经理,负责与用户沟通需求,制定OKR中的O,,并对下属的工作成果进行评估") diff --git a/main.py b/main.py index a066853..b2173a0 100644 --- a/main.py +++ b/main.py @@ -10,10 +10,11 @@ class AutoMate: def rule_define(self): o_kr = OKR() o_kr.set_objective(input("请输入任务: ")) + if __name__ == "__main__": automator = AutoMate() - automator.main() + automator.rule_define() # print(automator.call_chatgpt_api("Hello"))