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])