43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# mcp/config.py
|
|
import os
|
|
from pathlib import Path
|
|
|
|
# ディレクトリ設定
|
|
BASE_DIR = Path.home() / ".config" / "aigpt"
|
|
MEMORY_DIR = BASE_DIR / "memory"
|
|
SUMMARY_DIR = MEMORY_DIR / "summary"
|
|
|
|
def init_directories():
|
|
"""必要なディレクトリを作成"""
|
|
BASE_DIR.mkdir(parents=True, exist_ok=True)
|
|
MEMORY_DIR.mkdir(parents=True, exist_ok=True)
|
|
SUMMARY_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
def load_config():
|
|
"""環境変数から設定を読み込み"""
|
|
provider = os.getenv("PROVIDER", "ollama")
|
|
model = os.getenv("MODEL", "syui/ai" if provider == "ollama" else "gpt-4o-mini")
|
|
api_key = os.getenv("OPENAI_API_KEY", "")
|
|
|
|
if provider == "ollama":
|
|
return {
|
|
"provider": "ollama",
|
|
"model": model,
|
|
"url": f"{os.getenv('OLLAMA_HOST', 'http://localhost:11434')}/api/generate"
|
|
}
|
|
elif provider == "openai":
|
|
return {
|
|
"provider": "openai",
|
|
"model": model,
|
|
"api_key": api_key,
|
|
"url": f"{os.getenv('OPENAI_API_BASE', 'https://api.openai.com/v1')}/chat/completions"
|
|
}
|
|
elif provider == "mcp":
|
|
return {
|
|
"provider": "mcp",
|
|
"model": model,
|
|
"url": os.getenv("MCP_URL", "http://localhost:5000/chat")
|
|
}
|
|
else:
|
|
raise ValueError(f"Unsupported provider: {provider}")
|