add mcp

mcp/scripts/ask.py

@@ -1,55 +1,28 @@
-import os
-import json
-import httpx
-import openai
-
-from context_loader import load_context_from_repo
-from prompt_template import PROMPT_TEMPLATE
-
-PROVIDER = os.getenv("PROVIDER", "ollama")  # "ollama" or "openai"
-
-# Ollama用
-OLLAMA_HOST = os.getenv("OLLAMA_HOST", "http://localhost:11434")
-OLLAMA_URL = f"{OLLAMA_HOST}/api/generate"
-OLLAMA_MODEL = os.getenv("MODEL", "syui/ai")
-
-# OpenAI用
-OPENAI_BASE = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
-OPENAI_KEY = os.getenv("OPENAI_API_KEY", "")
-OPENAI_MODEL = os.getenv("MODEL", "gpt-4o-mini")
-
-def ask_question(question, repo_path="."):
-    context = load_context_from_repo(repo_path)
-    prompt = PROMPT_TEMPLATE.format(context=context[:10000], question=question)
-
-    if PROVIDER == "ollama":
-        payload = {
-            "model": OLLAMA_MODEL,
-            "prompt": prompt,
-            "stream": False
-        }
-        response = httpx.post(OLLAMA_URL, json=payload, timeout=60.0)
-        result = response.json()
-        return result.get("response", "返答がありませんでした。")
-
-    elif PROVIDER == "openai":
-        import openai
-        openai.api_key = OPENAI_KEY
-        openai.api_base = OPENAI_BASE
-
-        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-        response = client.chat.completions.create(
-            model=OPENAI_MODEL,
-            messages=[{"role": "user", "content": prompt}]
-        )
-        return response.choices[0].message.content
+## scripts/ask.py
+import sys
+import requests
+from config import load_config
 
+def ask(prompt):
+    cfg = load_config()
+    if cfg["provider"] == "ollama":
+        payload = {"model": cfg["model"], "prompt": prompt, "stream": False}
+        response = requests.post(cfg["url"], json=payload)
+        print(response.json().get("response", "❌ No response"))
     else:
-        return f"❌ 未知のプロバイダです: {PROVIDER}"
-
+        headers = {
+            "Authorization": f"Bearer {cfg['api_key']}",
+            "Content-Type": "application/json"
+        }
+        payload = {
+            "model": cfg["model"],
+            "messages": [{"role": "user", "content": prompt}]
+        }
+        response = requests.post(cfg["url"], headers=headers, json=payload)
+        print(response.json().get("choices", [{}])[0].get("message", {}).get("content", "❌ No content"))
 
 if __name__ == "__main__":
-    import sys
-    question = " ".join(sys.argv[1:])
-    answer = ask_question(question)
-    print("\n🧠 回答:\n", answer)
+    if len(sys.argv) < 2:
+        print("Usage: python ask.py 'your message'")
+        sys.exit(1)
+    ask(sys.argv[1])

mcp/scripts/config.py

@@ -0,0 +1,34 @@
+# 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)

mcp/scripts/summarize.py

@@ -0,0 +1,76 @@
+# scripts/summarize.py
+import json
+from datetime import datetime
+from config import MEMORY_DIR, SUMMARY_DIR, load_config
+import requests
+
+def load_memory(date_str):
+    path = MEMORY_DIR / f"{date_str}.json"
+    if not path.exists():
+        print(f"⚠️ メモリファイルが見つかりません: {path}")
+        return None
+    with open(path, "r") as f:
+        return json.load(f)
+
+def save_summary(date_str, content):
+    SUMMARY_DIR.mkdir(parents=True, exist_ok=True)
+    path = SUMMARY_DIR / f"{date_str}_summary.json"
+    with open(path, "w") as f:
+        json.dump(content, f, indent=2, ensure_ascii=False)
+    print(f"✅ 要約を保存しました: {path}")
+
+def build_prompt(logs):
+    messages = [
+        {"role": "system", "content": "あなたは要約AIです。以下の会話ログを要約してください。"},
+        {"role": "user", "content": "\n".join(f"{entry['sender']}: {entry['message']}" for entry in logs)}
+    ]
+    return messages
+
+def summarize_with_llm(messages):
+    cfg = load_config()
+    if cfg["provider"] == "openai":
+        headers = {
+            "Authorization": f"Bearer {cfg['api_key']}",
+            "Content-Type": "application/json",
+        }
+        payload = {
+            "model": cfg["model"],
+            "messages": messages,
+            "temperature": 0.7
+        }
+        response = requests.post(cfg["url"], headers=headers, json=payload)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+
+    elif cfg["provider"] == "ollama":
+        payload = {
+            "model": cfg["model"],
+            "prompt": "\n".join(m["content"] for m in messages),
+            "stream": False,
+        }
+        response = requests.post(cfg["url"], json=payload)
+        response.raise_for_status()
+        return response.json()["response"]
+
+    else:
+        raise ValueError(f"Unsupported provider: {cfg['provider']}")
+
+def main():
+    date_str = datetime.now().strftime("%Y-%m-%d")
+    logs = load_memory(date_str)
+    if not logs:
+        return
+
+    prompt_messages = build_prompt(logs)
+    summary_text = summarize_with_llm(prompt_messages)
+
+    summary = {
+        "date": date_str,
+        "summary": summary_text,
+        "total_messages": len(logs)
+    }
+
+    save_summary(date_str, summary)
+
+if __name__ == "__main__":
+    main()