first commit

2026-07-22 13:48:46 +08:00
commit c87751c3dc
2820 changed files with 726976 additions and 0 deletions
@@ -0,0 +1,5 @@
+from config.config import get_config
+
+__all__ = [
+    "get_config",
+]
@@ -0,0 +1,53 @@
+from bootstrap_env import load_project_env
+
+load_project_env()
+
+from SimpleLLMFunc import OpenAICompatible
+from functools import lru_cache
+import os
+from typing import Optional
+from dotenv import load_dotenv
+
+# Load the .env file under the working directory.
+load_dotenv()
+
+
+current_file_dir = (
+    __file__.rsplit("/", 1)[0] if "/" in __file__ else __file__.rsplit("\\", 1)[0]
+)
+
+
+class Config:
+    JSON_FILE = current_file_dir + "/provider.json"
+    INTERFACE_COLLECTION = OpenAICompatible.load_from_json_file(JSON_FILE)
+
+    BASIC_INTERFACE = INTERFACE_COLLECTION["openrouter"]["anthropic/claude-sonnet-4.6"]
+    CODE_INTERFACE = INTERFACE_COLLECTION["openrouter"]["google/gemini-3.1-pro-preview"]
+    REASONING_INTERFACE = INTERFACE_COLLECTION["openrouter"]["anthropic/claude-sonnet-4.6"]
+    QUICK_INTERFACE = INTERFACE_COLLECTION["openrouter"]["google/gemini-3-flash-preview"]
+    MULTIMODALITY_INTERFACE = INTERFACE_COLLECTION["openrouter"]["google/gemini-3-flash-preview"]
+    CONTEXT_SUMMARY_INTERFACE = INTERFACE_COLLECTION["openrouter"]["google/gemini-3-flash-preview"]
+
+    # ==================== Context & Sketch configuration parameters ====================
+
+    # redis url
+    REDIS_DB: int = int(os.getenv("REDIS_DB", 0))
+    REDIS_PASSWORD: Optional[str] = os.getenv("REDIS_PASSWORD", None)
+    REDIS_HOST: str = os.getenv("REDIS_HOST", "localhost")
+    REDIS_PORT: int = int(os.getenv("REDIS_PORT", 9736))
+
+    # Context storage directory configuration.
+    CONTEXT_DIR: str = os.getenv("CONTEXT_DIR", "workspace/data/contexts")
+    CONTEXT_MAX_HISTORY_LENGTH: int = int(os.getenv("CONTEXT_MAX_HISTORY_LENGTH", 10))
+    CONTEXT_AUTO_SUMMARIZE_TRIGGER: int = int(
+        os.getenv("CONTEXT_AUTO_SUMMARIZE_TRIGGER", 1000000)
+    )
+
+    # SketchPad storage directory configuration.
+    SKETCH_DIR: str = os.getenv("SKETCH_DIR", "workspace/data/sketches")
+
+
+@lru_cache()
+def get_config() -> Config:
+    """Get the configuration instance."""
+    return Config()
@@ -0,0 +1,52 @@
+{
+  "volc_engine": [
+    {
+      "model_name": "deepseek-v3-250324",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://ark.cn-beijing.volces.com/api/v3/",
+      "max_retries": 3,
+      "retry_delay": 1
+    },
+    {
+      "model_name": "doubao-1-5-pro-32k-250115",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://ark.cn-beijing.volces.com/api/v3/",
+      "max_retries": 3,
+      "retry_delay": 1
+    }
+  ],
+  "openrouter": [
+    {
+      "model_name": "z-ai/glm-5",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://openrouter.ai/api/v1"
+    },
+    {
+      "model_name": "google/gemini-3.1-pro-preview",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://openrouter.ai/api/v1"
+    },
+    {
+      "model_name": "anthropic/claude-sonnet-4.6",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://openrouter.ai/api/v1"
+    },
+    {
+      "model_name": "google/gemini-3-flash-preview",
+      "api_keys": [
+        "your_keys"
+      ],
+      "base_url": "https://openrouter.ai/api/v1"
+    }
+  ]
+}