add context and config2

metagpt/configs/__init__.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 16:33
+@Author  : alexanderwu
+@File    : __init__.py
+"""

metagpt/configs/browser_config.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:06
+@Author  : alexanderwu
+@File    : browser_config.py
+"""
+from typing import Literal
+
+from metagpt.tools import WebBrowserEngineType
+from metagpt.utils.yaml_model import YamlModel
+
+
+class BrowserConfig(YamlModel):
+    """Config for Browser"""
+
+    engine: WebBrowserEngineType = WebBrowserEngineType.PLAYWRIGHT
+    browser: Literal["chrome", "firefox", "edge", "ie"] = "chrome"
+    driver: Literal["chromium", "firefox", "webkit"] = "chromium"
+    path: str = ""

metagpt/configs/llm_config.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 16:33
+@Author  : alexanderwu
+@File    : llm_config.py
+"""
+from enum import Enum
+from typing import Optional
+
+from pydantic import field_validator
+
+from metagpt.utils.yaml_model import YamlModel
+
+
+class LLMType(Enum):
+    OPENAI = "openai"
+    ANTHROPIC = "anthropic"
+    SPARK = "spark"
+    ZHIPUAI = "zhipuai"
+    FIREWORKS = "fireworks"
+    OPEN_LLM = "open_llm"
+    GEMINI = "gemini"
+    METAGPT = "metagpt"
+    AZURE_OPENAI = "azure"
+    OLLAMA = "ollama"
+
+
+class LLMConfig(YamlModel):
+    """Config for LLM
+
+    OpenAI: https://github.com/openai/openai-python/blob/main/src/openai/resources/chat/completions.py#L681
+    Optional Fields in pydantic: https://docs.pydantic.dev/latest/migration/#required-optional-and-nullable-fields
+    """
+
+    api_key: str
+    api_type: LLMType = LLMType.OPENAI
+    base_url: str = "https://api.openai.com/v1"
+    api_version: Optional[str] = None
+    model: Optional[str] = None  # also stands for DEPLOYMENT_NAME
+
+    # For Spark(Xunfei), maybe remove later
+    app_id: Optional[str] = None
+    api_secret: Optional[str] = None
+    domain: Optional[str] = None
+
+    # For Chat Completion
+    max_token: int = 4096
+    temperature: float = 0.0
+    top_p: float = 1.0
+    top_k: int = 0
+    repetition_penalty: float = 1.0
+    stop: Optional[str] = None
+    presence_penalty: float = 0.0
+    frequency_penalty: float = 0.0
+    best_of: Optional[int] = None
+    n: Optional[int] = None
+    stream: bool = False
+    logprobs: Optional[bool] = None  # https://cookbook.openai.com/examples/using_logprobs
+    top_logprobs: Optional[int] = None
+    timeout: int = 60
+
+    # For Network
+    proxy: Optional[str] = None
+
+    # Cost Control
+    calc_usage: bool = True
+
+    @field_validator("api_key")
+    @classmethod
+    def check_llm_key(cls, v):
+        if v in ["", None, "YOUR_API_KEY"]:
+            raise ValueError("Please set your API key in config.yaml")
+        return v

metagpt/configs/mermaid_config.py

@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:07
+@Author  : alexanderwu
+@File    : mermaid_config.py
+"""
+from typing import Literal
+
+from metagpt.utils.yaml_model import YamlModel
+
+
+class MermaidConfig(YamlModel):
+    """Config for Mermaid"""
+
+    engine: Literal["nodejs", "ink", "playwright", "pyppeteer"] = "nodejs"
+    path: str = ""
+    puppeteer_config: str = ""  # Only for nodejs engine

metagpt/configs/redis_config.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:06
+@Author  : alexanderwu
+@File    : redis_config.py
+"""
+from metagpt.utils.yaml_model import YamlModelWithoutDefault
+
+
+class RedisConfig(YamlModelWithoutDefault):
+    host: str
+    port: int
+    username: str = ""
+    password: str
+    db: str
+
+    def to_url(self):
+        return f"redis://{self.host}:{self.port}"
+
+    def to_kwargs(self):
+        return {
+            "username": self.username,
+            "password": self.password,
+            "db": self.db,
+        }

metagpt/configs/s3_config.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:07
+@Author  : alexanderwu
+@File    : s3_config.py
+"""
+from metagpt.utils.yaml_model import YamlModelWithoutDefault
+
+
+class S3Config(YamlModelWithoutDefault):
+    access_key: str
+    secret_key: str
+    endpoint: str
+    bucket: str

metagpt/configs/search_config.py

@@ -0,0 +1,17 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:06
+@Author  : alexanderwu
+@File    : search_config.py
+"""
+from metagpt.tools import SearchEngineType
+from metagpt.utils.yaml_model import YamlModel
+
+
+class SearchConfig(YamlModel):
+    """Config for Search"""
+
+    api_key: str
+    api_type: SearchEngineType = SearchEngineType.SERPAPI_GOOGLE
+    cse_id: str = ""  # for google

metagpt/configs/workspace_config.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2024/1/4 19:09
+@Author  : alexanderwu
+@File    : workspace_config.py
+"""
+from datetime import datetime
+from pathlib import Path
+from uuid import uuid4
+
+from pydantic import field_validator, model_validator
+
+from metagpt.const import DEFAULT_WORKSPACE_ROOT
+from metagpt.utils.yaml_model import YamlModel
+
+
+class WorkspaceConfig(YamlModel):
+    path: Path = DEFAULT_WORKSPACE_ROOT
+    use_uid: bool = False
+    uid: str = ""
+
+    @field_validator("path")
+    @classmethod
+    def check_workspace_path(cls, v):
+        if isinstance(v, str):
+            v = Path(v)
+        return v
+
+    @model_validator(mode="after")
+    def check_uid_and_update_path(self):
+        if self.use_uid and not self.uid:
+            self.uid = f"{datetime.now().strftime('%Y%m%d%H%M%S')}-{uuid4().hex[-8:]}"
+            self.path = self.path / self.uid
+
+        # Create workspace path if not exists
+        self.path.mkdir(parents=True, exist_ok=True)
+        return self