repository_reader / config /llm_settings.py
DeL-TaiseiOzaki
st
1e951c6
raw
history blame
871 Bytes
import os
from dotenv import load_dotenv
class LLMSettings:
def __init__(self):
load_dotenv()
# デバッグ用に表示
print("ANTHROPIC_API_KEY:", bool(os.getenv("ANTHROPIC_API_KEY")))
self.anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
self.openai_api_key = os.getenv("OPENAI_API_KEY")
self.default_llm = "claude"
# 利用可能なモデルを確認
available = []
if self.anthropic_api_key:
available.append("claude")
if self.openai_api_key:
available.append("openai")
print("Available models:", available)
def get_available_models(self):
models = []
if self.anthropic_api_key:
models.append("claude")
if self.openai_api_key:
models.append("openai")
return models