Create chatgpt.py

chatgpt.py

@@ -0,0 +1,151 @@
+import os
+import time
+import json
+import requests
+from datetime import datetime, timezone
+
+# ===== Channel.io =====
+CHANNEL_ID = "200605"
+GROUP_ID = "534868"
+
+GET_URL = f"https://desk-api.channel.io/desk/channels/{CHANNEL_ID}/groups/{GROUP_ID}/messages"
+POST_URL = GET_URL
+
+X_ACCOUNT = os.getenv("channeliotokenbot2")
+if not X_ACCOUNT:
+    raise RuntimeError("環境変数 channeliotokenbot2 が設定されていません")
+
+HEADERS = {
+    "accept": "application/json",
+    "accept-language": "ja",
+    "content-type": "application/json",
+    "x-account": X_ACCOUNT,
+}
+
+PARAMS = {
+    "sortOrder": "asc",
+    "limit": 50,
+}
+
+ASSISTANT_PERSON_ID = "595702"
+
+# ===== LLM API =====
+LLM_ENDPOINT = "https://izuemon-gpt-free-api.hf.space/v1"
+
+
+# ===== Utils =====
+def parse_updated_at(value):
+    if isinstance(value, (int, float)):
+        return datetime.fromtimestamp(value / 1000, tz=timezone.utc)
+    elif isinstance(value, str):
+        return datetime.fromisoformat(value.replace("Z", "+00:00"))
+    return None
+
+
+def fetch_messages():
+    res = requests.get(GET_URL, headers=HEADERS, params=PARAMS, timeout=30)
+    res.raise_for_status()
+    return res.json().get("messages", [])
+
+
+def build_chat_messages(channel_messages):
+    """
+    Channel.io → ChatGPT messages 変換
+    """
+    chat_messages = []
+
+    for msg in channel_messages:
+        text = msg.get("plainText")
+        if not text:
+            continue
+
+        role = "assistant" if msg.get("personId") == ASSISTANT_PERSON_ID else "user"
+        chat_messages.append({
+            "role": role,
+            "content": text
+        })
+
+    return chat_messages
+
+
+def call_llm(messages):
+    payload = {
+        "model": "gpt-3.5-turbo",
+        "messages": messages
+    }
+
+    res = requests.post(
+        LLM_ENDPOINT,
+        headers={"Content-Type": "application/json"},
+        data=json.dumps(payload),
+        timeout=60
+    )
+    res.raise_for_status()
+
+    data = res.json()
+    return data["choices"][0]["message"]["content"]
+
+
+def send_to_channel(text):
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {"type": "text", "value": text}
+        ]
+    }
+
+    res = requests.post(
+        POST_URL,
+        headers=HEADERS,
+        data=json.dumps(payload),
+        timeout=30
+    )
+    res.raise_for_status()
+
+
+# ===== Main =====
+def main():
+    processed_until = None
+
+    while True:
+        try:
+            messages = fetch_messages()
+
+            # 新しいメッセージのみ抽出
+            new_messages = []
+            for msg in messages:
+                t = parse_updated_at(msg.get("updatedAt"))
+                if not t:
+                    continue
+
+                if processed_until is None or t > processed_until:
+                    new_messages.append(msg)
+
+            if not new_messages:
+                time.sleep(10)
+                continue
+
+            # ChatGPT 用メッセージ構築
+            chat_messages = build_chat_messages(messages)
+
+            # LLM 呼び出し
+            reply = call_llm(chat_messages)
+
+            # Channel.io に送信
+            send_to_channel(reply)
+
+            # 処理済み時刻更新
+            processed_until = max(
+                parse_updated_at(m["updatedAt"]) for m in new_messages
+            )
+
+            print("返信送信完了")
+
+        except Exception as e:
+            print("エラー:", e)
+
+        time.sleep(15)
+
+
+if __name__ == "__main__":
+    main()