From 68ca7869fa3c331078286b07ed1f512ca69b6d1a Mon Sep 17 00:00:00 2001
From: yuruo <yuruotong1@163.com>
Date: Sat, 20 Jan 2024 10:26:04 +0800
Subject: [PATCH] new

---
 README.md                             |  18 +++++-
 __pycache__/okr_define.cpython-39.pyc | Bin 0 -> 1166 bytes
 agent/agent.py                        |  47 ++++++++++++++++
 agent/manager_agent.py                |   5 ++
 agent/plan_agent.py                   |   5 ++
 agent/woker_agent.py                  |   5 ++
 main.py                               |  77 ++------------------------
 okr_define.py                         |  35 ++++++++++++
 tools/tools_base.py                   |   3 +
 tools/web_browser_tools.py            |  52 +++++++++++++++++
 10 files changed, 173 insertions(+), 74 deletions(-)
 create mode 100644 __pycache__/okr_define.cpython-39.pyc
 create mode 100644 agent/agent.py
 create mode 100644 agent/manager_agent.py
 create mode 100644 agent/plan_agent.py
 create mode 100644 agent/woker_agent.py
 create mode 100644 okr_define.py
 create mode 100644 tools/tools_base.py
 create mode 100644 tools/web_browser_tools.py

diff --git a/README.md b/README.md
index 9a60617..01bac54 100644
--- a/README.md
+++ b/README.md
@@ -1,3 +1,17 @@
-让大模型为你打工！
+# autoMate
 
-输入任务后就能上网找资料，然后参考优秀案例执行，最后给一个结果！你只需要像一个老板一样反馈结果好还是坏，autoMate会自动优化方案。
\ No newline at end of file
+autoMate是一个让大模型为你打工的工具。它能够根据你输入的任务，在网上找资料并参考优秀案例来执行任务，并最终给出一个结果。你只需要像一个老板一样反馈结果好还是坏，autoMate会自动优化方案。
+
+
+# 角色
+
+autoMate定义了三种角色：
+
+- 总经理：负责整体的任务管理和决策。
+- 计划者：负责制定任务的具体计划和策略。
+- 工人：负责执行具体的任务。
+
+
+# OKR工作体系
+
+市面上的大模型是对任务理解不透彻，如何解决这个问题呢？我想到了OKR工作体系，把上层的O向下拆解成KR，并且中间不断对焦！我认为这是一个非常高效的工作体系，AutoMate 引入OKR就像是给整个团队配上了高级导航系统，各agent都能清清楚楚知道自己要完成的任务，同时不断与上级对焦能够避免任务失真。
\ No newline at end of file
diff --git a/__pycache__/okr_define.cpython-39.pyc b/__pycache__/okr_define.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..25356e6ebaeb62f874b024c36f18fa2fadf17130
GIT binary patch
literal 1166
zcmZ`&&5P4O6n`^G(==^YcNN_#2woN<(t`+sEV8f%4`KxuJe0x^GqX#(*{05<g)O<c
ze~2E7|52`<JbUxxdy`H2fxhrQX5O2b-~8q!>UIM_`SkpUdgTE8q`{`(9Gsw<12k_z
zf@C+SU-=LuKZDSTaehlUj!sa`OEe0GOu~>$Cix8v9qC9Hy(=5iL*I~1>7(~#O9tqh
zvW<Io3g^SSk1@mbl~7C@pQ4((XvUC(C@mgv1}ioQc@d<aCo0ay3uTJ-rMebcnR$|#
zkQJ^`$@vPjhl~F_jbD$XTBszQjiPy$zK=3BN-wpL>U=y^{n>Tl3o)LKGa=|$<DvoP
zv1JF5^{Z4&xk@M|wk*Uw3?$280y#TI*6>z*@QX@UOjA|Jbpu_|su5dcrD;;8Yg8uc
zEWtMRIJVp;3|rZlwycC=(o<CP0!?m#EI{{d#1>3Hxj_CVjKGz?2j6(kzcXgRYC0)&
zEK*H(FT7|rQ&Wj^1Y5UhE?6_X3x1E_t<x<mH$EC*f&G>T)g9JF|G&f`F03zcKj)jg
zTBw?ucnh09)9D4izS%|gV2e0WmEFM&ZeKZI%SRQun*X{mwyl8GBox<FKf^(d@~A@D
z10?<^cS_7X$a&chCB@9L8@g*&HMNeW7T-nwaGTQ_(LJoag?`AEk1ObFV(ZF5=u=-$
zqbnsu5eRXW%6UR@M~E-;C@C|VUP^b!5G{J24E_Hbh^QYm^{J*`lhn`~_{?{G$M<~K
W&JRLn{l69#{&9Ky8(RGv75)Oz&gy;u

literal 0
HcmV?d00001

diff --git a/agent/agent.py b/agent/agent.py
new file mode 100644
index 0000000..ab0a2f2
--- /dev/null
+++ b/agent/agent.py
@@ -0,0 +1,47 @@
+import openai
+import yaml
+
+
+class Agent:
+    def __init__(self, role):
+        """
+        Initialize an Agent object.
+
+        Args:
+            role (str): The role of the agent.
+
+        Returns:
+            None
+        """
+        self.role = role
+
+    
+
+    def call(self, input_text):
+        """
+        Make a call to the OpenAI chat API.
+
+        Args:
+            input_text (str): The input text for the chat API.
+
+        Returns:
+            str: The generated text from the chat API.
+        """
+        # Load API configuration from YAML file
+        with open("config.yaml", "r") as file:
+            config = yaml.safe_load(file).get("openai")
+
+        openai.api_key = config.get("api_key")
+        openai.base_url = config.get("api_url")
+
+        response = openai.chat.completions.create(
+            model="gpt-3.5-turbo",  # GPT-3.5 model
+            messages=[
+                {"role": "user", "content": self.role +","+ input_text}
+            ],
+            temperature=0.2
+        )
+
+        generated_text = response.choices[0].message.content
+
+        return generated_text
\ No newline at end of file
diff --git a/agent/manager_agent.py b/agent/manager_agent.py
new file mode 100644
index 0000000..b2abba0
--- /dev/null
+++ b/agent/manager_agent.py
@@ -0,0 +1,5 @@
+from agent.agent import Agent
+
+class ManagerAgent(Agent):
+    def __init__(self):
+        super().__init__("你是一名总经理，负责与用户沟通需求，制定OKR中的O，并对下属的工作成果进行评估")
diff --git a/agent/plan_agent.py b/agent/plan_agent.py
new file mode 100644
index 0000000..e89d280
--- /dev/null
+++ b/agent/plan_agent.py
@@ -0,0 +1,5 @@
+from agent.agent import Agent
+
+class WokerAgent(Agent):
+    def __init__(self):
+        super().__init__("你是一名计划拆解者，负责对OKR中的O进行拆解并制定KR，向总经理汇报")
diff --git a/agent/woker_agent.py b/agent/woker_agent.py
new file mode 100644
index 0000000..d18255d
--- /dev/null
+++ b/agent/woker_agent.py
@@ -0,0 +1,5 @@
+from agent.agent import Agent
+
+class WokerAgent(Agent):
+    def __init__(self):
+        super().__init__("你是一名工作者，负责执行OKR中的KR，向总计划制定者汇报")
diff --git a/main.py b/main.py
index c24759b..69d8f86 100644
--- a/main.py
+++ b/main.py
@@ -1,79 +1,12 @@
-import os
-import requests
-import yaml
-from selenium import webdriver
-from selenium.webdriver.chrome.service import Service
-from selenium.webdriver.common.by import By
-from selenium.webdriver.chrome.options import Options
-from webdriver_manager.chrome import ChromeDriverManager
-from webdriver_manager.firefox import GeckoDriverManager
-from webdriver_manager.microsoft import EdgeChromiumDriverManager
-from webdriver_manager.opera import OperaDriverManager
-from selenium.webdriver.edge.service import Service as EdgeService
-import openai
-from selenium.webdriver.common.by import By
-
 
 class AutoMate:
     def __init__(self):
         pass
-    def main(self):
-        # Load browser configuration from YAML file
-        with open("config.yaml", "r") as file:
-            config = yaml.safe_load(file)
-
-        # Check if webdriver is available
-        if not os.path.exists("webdriver.exe"):
-            # Download webdriver based on browser type
-            browser_type = config.get("browser_type")
-            driver = None
-            if browser_type == "chrome":
-                options = webdriver.ChromeOptions()
-                options.add_argument("--headless")  # Enable headless mode
-                webdriver_manager = ChromeDriverManager()
-                driver = webdriver.Chrome(webdriver_manager.install(), options=options)
-            elif browser_type == "firefox":
-                options = webdriver.FirefoxOptions()
-                options.add_argument("--headless")  # Enable headless mode
-                webdriver_manager = GeckoDriverManager()
-                driver = webdriver.Firefox(webdriver_manager.install(), options=options)
-            elif browser_type == "edge":
-                options = webdriver.EdgeOptions()
-                options.add_argument("--headless")  # Enable headless mode
-                webdriver_manager = EdgeChromiumDriverManager()
-                driver = webdriver.Edge(service=EdgeService(webdriver_manager.install()), options=options)
-            elif browser_type == "opera":
-                options = webdriver.ChromeOptions()
-                options.add_argument("--headless")  # Enable headless mode
-                webdriver_manager = OperaDriverManager()
-                driver = webdriver.Opera(webdriver_manager.install(), options=options)
-        driver.implicitly_wait(10)
-        browser_url = config.get("browser_url")
-        driver.get(browser_url)
-        driver.find_element(by=By.XPATH, value='//*[@id="kw"]').send_keys("陈")
-        driver.find_element(by=By.XPATH, value='//*[@id="su"]').click()
-        print(driver.page_source)
-        driver.quit()
-
-    def call_chatgpt_api(self, input_text):
-        # Load API configuration from YAML file
-        with open("config.yaml", "r") as file:
-            config = yaml.safe_load(file).get("openai")
-
-        openai.api_key = config.get("api_key")
-        openai.base_url = config.get("api_url")
-
-        response = openai.chat.completions.create(
-            model="gpt-3.5-turbo",  # GPT-3.5 model
-            messages=[
-                {"role": "user", "content": input_text}
-            ],
-            temperature=0.7
-        )
-
-        generated_text = response.choices[0].message.content
-
-        return generated_text
+    
+    
+    def rule_define(self):
+        pass
+    
 
 
 
diff --git a/okr_define.py b/okr_define.py
new file mode 100644
index 0000000..77984fc
--- /dev/null
+++ b/okr_define.py
@@ -0,0 +1,35 @@
+class OKR:
+    """
+    Represents an Objectives and Key Results (OKR) tracker.
+
+    Attributes:
+        objectives (dict): A dictionary to store the objectives and their progress.
+        key_results (dict): A dictionary to store the key results for each objective.
+
+    Methods:
+        set_objective(objective): Sets a new objective with initial progress of 0.
+        set_key_result(objective, key_result): Sets a new key result for the given objective.
+        set_objective_progress(objective, progress): Updates the progress of the given objective.
+        set_key_result_progress(objective, key_result, progress): Updates the progress of the given key result.
+    """
+
+    def __init__(self):
+        self.objectives = {}
+        self.key_results = {}
+
+    def set_objective(self, objective):
+        self.objectives[objective] = 0
+
+    def set_key_result(self, objective, key_result):
+        if objective in self.objectives:
+            if objective not in self.key_results:
+                self.key_results[objective] = []
+            self.key_results[objective].append(key_result)
+
+    def set_objective_progress(self, objective, progress):
+        if objective in self.objectives:
+            self.objectives[objective] = progress
+
+    def set_key_result_progress(self, objective, key_result, progress):
+        if objective in self.key_results and key_result in self.key_results[objective]:
+            self.key_results[objective][key_result] = progress
diff --git a/tools/tools_base.py b/tools/tools_base.py
new file mode 100644
index 0000000..ba594a8
--- /dev/null
+++ b/tools/tools_base.py
@@ -0,0 +1,3 @@
+class ToolsBase:
+    def run(self):
+        pass
diff --git a/tools/web_browser_tools.py b/tools/web_browser_tools.py
new file mode 100644
index 0000000..cf7da4c
--- /dev/null
+++ b/tools/web_browser_tools.py
@@ -0,0 +1,52 @@
+import os
+import yaml
+from selenium import webdriver
+from selenium.webdriver.common.by import By
+from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.firefox import GeckoDriverManager
+from webdriver_manager.microsoft import EdgeChromiumDriverManager
+from webdriver_manager.opera import OperaDriverManager
+from selenium.webdriver.edge.service import Service as EdgeService
+from selenium.webdriver.common.by import By
+
+from tools.tools_base import ToolsBase
+
+class WebBrowser(ToolsBase):
+
+    def run(self):
+        # Load browser configuration from YAML file
+        with open("config.yaml", "r") as file:
+            config = yaml.safe_load(file)
+
+        # Check if webdriver is available
+        if not os.path.exists("webdriver.exe"):
+            # Download webdriver based on browser type
+            browser_type = config.get("browser_type")
+            driver = None
+            if browser_type == "chrome":
+                options = webdriver.ChromeOptions()
+                options.add_argument("--headless")  # Enable headless mode
+                webdriver_manager = ChromeDriverManager()
+                driver = webdriver.Chrome(webdriver_manager.install(), options=options)
+            elif browser_type == "firefox":
+                options = webdriver.FirefoxOptions()
+                options.add_argument("--headless")  # Enable headless mode
+                webdriver_manager = GeckoDriverManager()
+                driver = webdriver.Firefox(webdriver_manager.install(), options=options)
+            elif browser_type == "edge":
+                options = webdriver.EdgeOptions()
+                options.add_argument("--headless")  # Enable headless mode
+                webdriver_manager = EdgeChromiumDriverManager()
+                driver = webdriver.Edge(service=EdgeService(webdriver_manager.install()), options=options)
+            elif browser_type == "opera":
+                options = webdriver.ChromeOptions()
+                options.add_argument("--headless")  # Enable headless mode
+                webdriver_manager = OperaDriverManager()
+                driver = webdriver.Opera(webdriver_manager.install(), options=options)
+        driver.implicitly_wait(10)
+        browser_url = config.get("browser_url")
+        driver.get(browser_url)
+        driver.find_element(by=By.XPATH, value='//*[@id="kw"]').send_keys("陈")
+        driver.find_element(by=By.XPATH, value='//*[@id="su"]').click()
+        print(driver.page_source)
+        driver.quit()