colcaroli
347230e3c8
- Split backend/app/core into weather/, discord/, memory_kb/, search/ subpackages by responsibility, keeping config.py and llm.py as the shared foundation; updated all import sites accordingly. - Add local Markdown knowledge base support (local_knowledge.py, knowledge_base/fgo.md, LOCAL_KB_* settings) that embeds and injects relevant docs into the system prompt by topic similarity. - Add ruff and pyright as backend dev dependencies with basic config in pyproject.toml for linting and type checking. - Document local machine quirks in CLAUDE.md (Postgres on port 8080, SSLKEYLOGFILE breaking litellm/pyright via OPENSSL_Uplink crashes). - Add backend/run_dev.ps1 dev launcher and docs/discord-bot.md.
61 lines
3.6 KiB
Python
61 lines
3.6 KiB
Python
from dotenv import load_dotenv
|
|
import os
|
|
|
|
load_dotenv()
|
|
|
|
DATABASE_URL: str = os.getenv("DATABASE_URL", "sqlite:///./playchat.db")
|
|
DEFAULT_MODEL: str = os.getenv("DEFAULT_MODEL", "gpt-4o")
|
|
MEMORY_SUMMARIZE_THRESHOLD: int = int(os.getenv("MEMORY_SUMMARIZE_THRESHOLD", "20"))
|
|
CORS_ORIGINS: list[str] = os.getenv("CORS_ORIGINS", "http://localhost:5173").split(",")
|
|
DEBUG: bool = os.getenv("DEBUG", "false").lower() in ("1", "true", "yes")
|
|
|
|
# メッセージへの自動タグ付けに使うモデル（未設定ならDEFAULT_MODELを使用）
|
|
TAGGING_MODEL: str | None = os.getenv("TAGGING_MODEL")
|
|
|
|
# Tavily Web検索
|
|
TAVILY_API_KEY: str | None = os.getenv("TAVILY_API_KEY")
|
|
# 蓄積した知識をシステムプロンプトに注入する際の最大件数
|
|
KNOWLEDGE_CONTEXT_LIMIT: int = int(os.getenv("KNOWLEDGE_CONTEXT_LIMIT", "15"))
|
|
|
|
# ローカルMarkdownナレッジベース（話題に応じて関連ファイルをシステムプロンプトに注入）
|
|
LOCAL_KB_DIR: str = os.getenv("LOCAL_KB_DIR", "knowledge_base")
|
|
LOCAL_KB_EMBEDDING_MODEL: str = os.getenv("LOCAL_KB_EMBEDDING_MODEL", "text-embedding-3-small")
|
|
LOCAL_KB_TOP_K: int = int(os.getenv("LOCAL_KB_TOP_K", "3"))
|
|
LOCAL_KB_MIN_SCORE: float = float(os.getenv("LOCAL_KB_MIN_SCORE", "0.75"))
|
|
|
|
# Amazon Bedrock
|
|
# 以下はすべて LiteLLM(boto3) が os.environ から直接読み取るため、.env に書くだけで自動的に有効になる。
|
|
# IAM経由（推奨）: AWS_ACCESS_KEY_ID / AWS_SECRET_ACCESS_KEY を設定する。
|
|
# API key方式（旧）: AWS_BEARER_TOKEN_BEDROCK のみを設定する。
|
|
# 両方設定されている場合はIAM認証（Access Key）が優先される。
|
|
AWS_ACCESS_KEY_ID: str | None = os.getenv("AWS_ACCESS_KEY_ID")
|
|
AWS_SECRET_ACCESS_KEY: str | None = os.getenv("AWS_SECRET_ACCESS_KEY")
|
|
AWS_SESSION_TOKEN: str | None = os.getenv("AWS_SESSION_TOKEN") # 一時的なIAMクレデンシャルを使う場合のみ
|
|
AWS_BEARER_TOKEN_BEDROCK: str | None = os.getenv("AWS_BEARER_TOKEN_BEDROCK")
|
|
AWS_REGION_NAME: str = os.getenv("AWS_REGION_NAME", "us-east-1")
|
|
|
|
# 天気予報 → Discord 通知
|
|
DISCORD_WEBHOOK_URL: str | None = os.getenv("DISCORD_WEBHOOK_URL")
|
|
WEATHER_AREA_CODE: str = os.getenv("WEATHER_AREA_CODE", "120000") # 千葉県
|
|
WEATHER_AREA_KEYWORD: str = os.getenv("WEATHER_AREA_KEYWORD", "北西部") # 市川市は千葉県北西部
|
|
WEATHER_NOTIFY_HOUR: int = int(os.getenv("WEATHER_NOTIFY_HOUR", "22"))
|
|
WEATHER_NOTIFY_MINUTE: int = int(os.getenv("WEATHER_NOTIFY_MINUTE", "0"))
|
|
# 通知メッセージをLLMで生成する場合に使うモデル（未設定ならDEFAULT_MODELを使用）
|
|
WEATHER_NOTIFY_MODEL: str | None = os.getenv("WEATHER_NOTIFY_MODEL")
|
|
|
|
# Discord Bot（メンションへの自動返信用）
|
|
DISCORD_BOT_TOKEN: str | None = os.getenv("DISCORD_BOT_TOKEN")
|
|
DISCORD_BOT_SYSTEM_PROMPT: str = os.getenv(
|
|
"DISCORD_BOT_SYSTEM_PROMPT",
|
|
"あなたは「お天気アプリ」という名前のDiscord Botです。"
|
|
"天気や洗濯指数の質問には親しみやすく簡潔に答え、"
|
|
"雑談にも気さくに応じてください。絵文字を適度に使って親しみやすい口調で話します。",
|
|
)
|
|
|
|
# どうでもいいTipsを定期的にDiscordへ投稿する機能
|
|
USELESS_TIPS_ENABLED: bool = os.getenv("USELESS_TIPS_ENABLED", "false").lower() in ("1", "true", "yes")
|
|
USELESS_TIPS_INTERVAL_HOURS: int = int(os.getenv("USELESS_TIPS_INTERVAL_HOURS", "3"))
|
|
USELESS_TIPS_MODEL: str | None = os.getenv("USELESS_TIPS_MODEL")
|
|
# 未設定なら天気通知と同じ DISCORD_WEBHOOK_URL を使う
|
|
DISCORD_WEBHOOK_URL_TIPS: str | None = os.getenv("DISCORD_WEBHOOK_URL_TIPS") or os.getenv("DISCORD_WEBHOOK_URL")
|