35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# scripts/config.py
|
|
import os
|
|
from pathlib import Path
|
|
|
|
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"
|
|
}
|
|
else:
|
|
raise ValueError(f"Unsupported provider: {provider}")
|
|
|
|
# ディレクトリ設定
|
|
BASE_DIR = Path.home() / ".config" / "aigpt"
|
|
MEMORY_DIR = BASE_DIR / "memory"
|
|
SUMMARY_DIR = MEMORY_DIR / "summary"
|
|
|
|
# 初期化（必要に応じて作成）
|
|
BASE_DIR.mkdir(parents=True, exist_ok=True)
|
|
MEMORY_DIR.mkdir(parents=True, exist_ok=True)
|
|
SUMMARY_DIR.mkdir(parents=True, exist_ok=True)
|