colcaroli
234ebba103
Introduces websearch.py (Tavily) + knowledge.py to ground chat responses in search results, tagging.py to tag user and assistant messages via LLM and pick related topics for the Discord useless-tips job, and word_memory.py to track recent conversation nouns (surfaced in the new WordPool.vue sidebar) used to pick tip topics close to a related tag.
55 lines
3.2 KiB
Python
55 lines
3.2 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"))
|
|
|
|
# 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")
|