File size: 3,131 Bytes
7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 dc62d4a 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 7c8b666 6c88771 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import traceback
import datetime
import random
import logging
logger = logging.getLogger(__name__)
def get_fallback_response(scene=None):
"""
シーンに応じた固定のフォールバック応答を生成する
"""
scene_responses = {
"aquarium_night": "（水槽の青い光を見つめて）…夜の水族館か。悪くないかもな。",
"beach_sunset": "（夕日に照らされた海を見つめて）…こんな景色、久しぶりに見た。",
"festival_night": "（提灯の明かりを見上げて）…意外と、悪くない雰囲気だな。",
"shrine_day": "（静かな境内を見回して）…静かで、落ち着くな。",
"cafe_afternoon": "（窓の外を見ながら）…こういう時間の過ごし方も、悪くない。",
"room_night": "（窓の外の夜景を見て）…夜の景色って、なんか落ち着くな。",
}
if scene and scene in scene_responses:
return scene_responses[scene]
if scene:
return f"（{scene}の様子を静かに見回している）…ここか。"
return "（……何か言おうとしたが、言葉に詰まった）"
def generate_dialogue_with_swallow(history, message, affection, stage_name, scene_params, instruction=None, swallow_model=None, SYSTEM_PROMPT_MARI=None):
"""
Swallowモデル（GGUF版）を使用して対話応答を生成する
"""
if swallow_model is None:
logger.warning("Swallowモデルが利用できないため、フォールバック応答を返します。")
return get_fallback_response(scene_params.get("theme"))
history_text = "\n".join([f"ユーザー: {u}\n麻理: {m}" for u, m in history[-5:]]) # 直近5件に制限
prompt = f"""{SYSTEM_PROMPT_MARI}
# 現在の状況
- シーン: {scene_params.get("theme", "default")}
- 好感度: {affection} ({stage_name})
- 性格: {scene_params.get("personality_mod", "特になし")}
- 口調: {scene_params.get("tone", "特になし")}
# 会話履歴
{history_text}
---
# 指示
{f"特別な指示: {instruction}" if instruction else f"ユーザーの発言「{message}」に、麻理として応答してください。"}
麻理: """
try:
output = swallow_model(
prompt,
max_tokens=150,
temperature=0.8,
top_p=0.95,
stop=["ユーザー:", "user:", "\n\n"],
echo=False # プロンプトは返さない
)
response_text = output["choices"][0]["text"].strip()
if not response_text:
logger.warning("LLMが空の応答を返しました。フォールバックを使用します。")
return get_fallback_response(scene_params.get("theme"))
logger.info(f"Swallowモデル生成応答: {response_text}")
return response_text
except Exception as e:
logger.error(f"応答生成エラー(Swallow): {e}", exc_info=True)
return get_fallback_response(scene_params.get("theme")) |