Implement user management REST API for multi-user LLM configuration (#33)

* Initial plan

* Implement user management API server and client integration

Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>

* Add testing documentation for user config API

Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>

* Address code review feedback: use lifespan and app.state, move imports to top

Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>

* Add comprehensive README for user_config_api

Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>

* Use task author username instead of config owner for API calls

Modified _fetch_config_from_api to accept username parameter and added
fetch_user_config function to extract username from issue/PR/MR author.
TaskHandler now fetches per-user config before processing each task.

Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>

* 実際に動かして修正

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: notfolder <20558197+notfolder@users.noreply.github.com>
Co-authored-by: notfolder <notfolder@gmail.com>

Author: Copilot

config.yaml

@@ -24,6 +24,10 @@ mcp_servers:
       - "-m"
       - "mcp_server_fetch"
 
+# APIサーバー設定
+api_server:
+  api_key: "your-secret-api-key-here"
+
 llm:
   provider: "openai"    # "ollama" | "openai"
   function_calling: true

docker-compose.yml

@@ -1,5 +1,15 @@
 version: '3.6'
 services:
+  user-config-api:
+    build: ./user_config_api
+    container_name: user-config-api
+    # Docker内部ネットワークのみで使用（外部公開不要）
+    # テスト用に外部アクセスが必要な場合のみ、以下のコメントを外す
+    ports:
+      - "8081:8080"
+    environment:
+      API_SERVER_KEY: ${API_SERVER_KEY:-your-secret-api-key-here}
+
   web:
     image: 'gitlab/gitlab-ce:latest'
     restart: always

handlers/task.py

@@ -79,3 +79,12 @@ def get_task_key(self) -> "TaskKey":
             タスクの一意なキーオブジェクト
 
         """
+
+    @abstractmethod
+    def get_user(self) -> str | None:
+        """タスクの作成者のユーザー名を取得する.
+
+        Returns:
+            ユーザー名、取得できない場合はNone
+
+        """

handlers/task_getter_github.py

@@ -99,6 +99,10 @@ def get_task_key(self) -> GitHubIssueTaskKey:
     def check(self) -> bool:
         return self.config["github"]["processing_label"] in self.labels
 
+    def get_user(self) -> str | None:
+        """Issueの作成者のユーザー名を取得する."""
+        return self.issue.get("user", {}).get("login")
+
 
 class TaskGitHubPullRequest(Task):
     def __init__(
@@ -177,6 +181,10 @@ def get_task_key(self) -> GitHubPullRequestTaskKey:
     def check(self) -> bool:
         return self.config["github"]["processing_label"] in self.labels
 
+    def get_user(self) -> str | None:
+        """Pull Requestの作成者のユーザー名を取得する."""
+        return self.pr.get("user", {}).get("login")
+
 
 class TaskGetterFromGitHub(TaskGetter):
     def __init__(self, config: dict[str, Any], mcp_clients: dict[str, MCPToolClient]) -> None:

handlers/task_getter_gitlab.py

@@ -87,6 +87,10 @@ def get_task_key(self) -> GitLabIssueTaskKey:
     def check(self) -> bool:
         return self.config["gitlab"]["processing_label"] in self.issue.get("labels", [])
 
+    def get_user(self) -> str | None:
+        """Issueの作成者のユーザー名を取得する."""
+        return self.issue.get("author", {}).get("username")
+
 
 class TaskGitLabMergeRequest(Task):
     def __init__(
@@ -163,6 +167,10 @@ def get_task_key(self) -> GitLabMergeRequestTaskKey:
     def check(self) -> bool:
         return self.config["gitlab"]["processing_label"] in self.labels
 
+    def get_user(self) -> str | None:
+        """Merge Requestの作成者のユーザー名を取得する."""
+        return self.mr.get("author", {}).get("username")
+
 
 class TaskGetterFromGitLab(TaskGetter):
     def __init__(self, config: dict[str, Any], mcp_clients: dict[str, MCPToolClient]) -> None:

handlers/task_handler.py

@@ -94,12 +94,15 @@ def handle(self, task: Task) -> None:
             task: 処理対象のタスクオブジェクト
 
         """
+        # タスク固有の設定を取得
+        task_config = self._get_task_config(task)
+        
         # 初期設定
-        self._setup_task_handling(task)
+        self._setup_task_handling(task, task_config)
 
         # 処理ループの初期化
         count = 0
-        max_count = self.config.get("max_llm_process_num", 1000)
+        max_count = task_config.get("max_llm_process_num", 1000)
 
         # 連続ツールエラー管理用の状態
         error_state = {"last_tool": None, "tool_error_count": 0}
@@ -110,12 +113,47 @@ def handle(self, task: Task) -> None:
                 break
             count += 1
 
-    def _setup_task_handling(self, task: Task) -> None:
-        """タスク処理の初期設定を行う."""
+    def _get_task_config(self, task: Task) -> dict[str, Any]:
+        """タスクに応じた設定を取得する.
+        
+        USE_USER_CONFIG_API環境変数がtrueの場合、タスクのユーザーに基づいて
+        API経由で設定を取得します。
+        
+        Args:
+            task: タスクオブジェクト
+        
+        Returns:
+            タスク用の設定辞書
+        """
+        import os
+        
+        # API使用フラグをチェック
+        use_api = os.environ.get("USE_USER_CONFIG_API", "false").lower() == "true"
+        if not use_api:
+            return self.config
+        
+        # main.pyのfetch_user_configを使用
+        try:
+            from main import fetch_user_config
+            return fetch_user_config(task, self.config)
+        except Exception as e:
+            self.logger.warning(f"ユーザー設定の取得に失敗しました: {e}。デフォルト設定を使用します。")
+            return self.config
+
+    def _setup_task_handling(self, task: Task, task_config: dict[str, Any] | None = None) -> None:
+        """タスク処理の初期設定を行う.
+        
+        Args:
+            task: タスクオブジェクト
+            task_config: タスク固有の設定（Noneの場合はself.configを使用）
+        """
+        if task_config is None:
+            task_config = self.config
+            
         prompt = task.get_prompt()
         self.logger.info("LLMに送信するプロンプト: %s", prompt)
 
-        self.llm_client.send_system_prompt(self._make_system_prompt())
+        self.llm_client.send_system_prompt(self._make_system_prompt(task_config))
         self.llm_client.send_user_message(prompt)
 
     def _process_llm_interaction(self, task: Task, count: int, error_state: dict) -> bool:
@@ -336,20 +374,26 @@ def _process_done_field(self, task: Task, data: dict) -> None:
 
     def get_system_prompt(self) -> str:
         """システムプロンプトを取得する(テスト用の公開メソッド)."""
-        return self._make_system_prompt()
+        return self._make_system_prompt(self.config)
 
-    def _make_system_prompt(self) -> str:
+    def _make_system_prompt(self, task_config: dict[str, Any] | None = None) -> str:
         """システムプロンプトを生成する.
 
         設定に基づいてfunction callingの有無を判定し、
         適切なシステムプロンプトファイルを読み込んで、
-        MCPプロンプトを埋め込んで返します。
+        MCPプロンプトを埋め込んで返します.
+        
+        Args:
+            task_config: タスク固有の設定（Noneの場合はself.configを使用）
 
         Returns:
             生成されたシステムプロンプト文字列
 
         """
-        if self.config.get("llm", {}).get("function_calling", True):
+        if task_config is None:
+            task_config = self.config
+            
+        if task_config.get("llm", {}).get("function_calling", True):
             # function callingが有効な場合
             with Path("system_prompt_function_call.txt").open() as f:
                 prompt = f.read()

main.py

@@ -14,6 +14,7 @@
 from pathlib import Path
 from typing import Any
 
+import requests
 import yaml
 
 from clients.lm_client import get_llm_client
@@ -24,6 +25,7 @@
 from queueing import InMemoryTaskQueue, RabbitMQTaskQueue
 
 
+
 def setup_logger() -> None:
     """ログ設定を初期化する.
 
@@ -50,6 +52,8 @@ def load_config(config_file: str = "config.yaml") -> dict[str, Any]:
 
     指定された設定ファイルを読み込み、環境変数で定義された値で
     設定を上書きします。LLM、MCP、RabbitMQ等の設定が対象です。
+    
+    USE_USER_CONFIG_API環境変数がtrueの場合、API経由で設定を取得します。
 
     Args:
         config_file: 読み込む設定ファイルのパス
@@ -62,19 +66,138 @@ def load_config(config_file: str = "config.yaml") -> dict[str, Any]:
         yaml.YAMLError: YAMLの解析に失敗した場合
 
     """
+    # ロガー取得
+    logger = logging.getLogger(__name__)
+    
     # 設定ファイルを読み込み
     with Path(config_file).open() as f:
         config = yaml.safe_load(f)
-
-    # 各種設定の上書き処理
-    _override_llm_config(config)
+    
+    # 環境変数による設定上書きを先に実行
     _override_mcp_config(config)
     _override_rabbitmq_config(config)
     _override_bot_config(config)
+    
+    # API経由でLLM設定を取得するかチェック
+    use_api = os.environ.get("USE_USER_CONFIG_API", "false").lower() == "true"
+    
+    if use_api:
+        try:
+            config = _fetch_config_from_api(config, logger)
+        except (requests.ConnectionError, requests.Timeout) as e:
+            logger.exception(f"API接続エラー、設定ファイルを使用{e}")
+        except (ValueError, requests.HTTPError) as e:
+            logger.exception(f"API設定取得エラー、設定ファイルを使用{e}")
+        except Exception as e:
+            logger.exception(f"予期しないエラー、設定ファイルを使用{e}")
+    
+    # LLM設定を環境変数で最終的に上書き
+    _override_llm_config(config)
 
     return config
 
 
+def _fetch_config_from_api(
+    config: dict[str, Any], logger: logging.Logger, username: str | None = None
+) -> dict[str, Any]:
+    """API経由で設定を取得する.
+    
+    Args:
+        config: ベースとなる設定辞書
+        logger: ロガー
+        username: ユーザー名（Noneの場合はconfig.yamlから取得）
+    
+    Returns:
+        API設定でマージされた設定辞書
+    
+    Raises:
+        ValueError: 設定エラーまたはAPI呼び出しエラー
+    """
+    # タスクソースを取得
+    task_source = os.environ.get("TASK_SOURCE", "github")
+    
+    # ユーザー名が指定されていない場合はconfig.yamlから取得
+    if username is None:
+        if task_source == "github":
+            username = config.get("github", {}).get("owner", "")
+        elif task_source == "gitlab":
+            username = config.get("gitlab", {}).get("owner", "")
+        else:
+            raise ValueError(f"Unknown task source: {task_source}")
+    
+    # APIエンドポイントとAPIキー
+    api_url = os.environ.get("USER_CONFIG_API_URL", "http://user-config-api:8080")
+    api_key = os.environ.get("USER_CONFIG_API_KEY", "")
+    
+    if not api_key:
+        raise ValueError("USER_CONFIG_API_KEY is not set")
+    
+    url = f"{api_url}/config/{task_source}/{username}"
+    
+    # Bearer トークンとしてAPIキーをヘッダーに含めて呼び出し
+    headers = {"Authorization": f"Bearer {api_key}"}
+    response = requests.get(url, headers=headers, timeout=5)
+    response.raise_for_status()
+
+    data = response.json()
+    if data.get("status") == "success":
+        # API設定を取得
+        api_data = data["data"]
+        
+        # LLM設定を上書き
+        config["llm"] = api_data["llm"]
+        
+        # システムプロンプトを上書き（環境変数で上書きされていない場合のみ）
+        if "system_prompt" in api_data:
+            config["system_prompt"] = api_data["system_prompt"]
+        
+        logger.info(f"API経由でLLM設定を取得: {task_source}:{username}")
+    else:
+        raise ValueError(f"API returned error: {data.get('message')}")
+    
+    return config
+
+
+def fetch_user_config(task: Any, base_config: dict[str, Any]) -> dict[str, Any]:
+    """タスクのユーザーに基づいて設定を取得する.
+
+    USE_USER_CONFIG_API環境変数がtrueの場合、タスクの作成者のユーザー名を使用して
+    API経由で設定を取得します。そうでない場合はbase_configをそのまま返します。
+
+    Args:
+        task: タスクオブジェクト（issue/PR/MRの情報を含む）
+        base_config: ベースとなる設定辞書
+
+    Returns:
+        ユーザー設定でマージされた設定辞書
+    """
+    # API使用フラグをチェック
+    use_api = os.environ.get("USE_USER_CONFIG_API", "false").lower() == "true"
+    if not use_api:
+        return base_config
+
+    # タスクからユーザー名を取得
+    try:
+        username = task.get_user()
+
+        if not username:
+            # ユーザー名が取得できない場合はベース設定を使用
+            logger = logging.getLogger(__name__)
+            logger.warning("タスクからユーザー名を取得できませんでした。デフォルト設定を使用します。")
+            return base_config
+
+        # API経由で設定を取得
+        logger = logging.getLogger(__name__)
+        return _fetch_config_from_api(base_config.copy(), logger, username)
+
+    except (ValueError, TypeError, AttributeError) as e:
+        # エラーが発生した場合はベース設定を使用
+        logger = logging.getLogger(__name__)
+        logger.warning("ユーザー設定の取得に失敗しました: %s。デフォルト設定を使用します。", e)
+        return base_config
+
+
+
 def _override_llm_config(config: dict[str, Any]) -> None:
     """LLM設定を環境変数で上書きする."""
     # function_calling設定の処理

sample_api.env

@@ -0,0 +1,19 @@
+# ユーザー設定API設定の例
+# このファイルをコピーして .env として使用してください
+
+# 既存の設定
+TASK_SOURCE=github
+GITHUB_PERSONAL_ACCESS_TOKEN=your_github_token_here
+
+# ユーザー設定API設定（オプション）
+# API経由で設定を取得する場合はtrueに設定
+USE_USER_CONFIG_API=false
+
+# ユーザー設定APIのURL（デフォルト: http://user-config-api:8080）
+USER_CONFIG_API_URL=http://user-config-api:8080
+
+# ユーザー設定APIのAPIキー（USE_USER_CONFIG_API=trueの場合は必須）
+USER_CONFIG_API_KEY=your-secret-api-key
+
+# APIサーバー用のAPIキー（環境変数で上書き可能）
+API_SERVER_KEY=your-secret-api-key-here

user_config_api/Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.13-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY server.py .
+COPY config.yaml .
+
+EXPOSE 8080
+
+CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "8080"]