Update chatgpt.py

chatgpt.py

@@ -2,13 +2,19 @@ import os
 import time
 import json
 import requests
+from datetime import datetime, timezone
 
 # ===== Channel.io 設定 =====
 GROUP_ID = "534868"
 CHANNEL_ID = "200605"
 
 GET_URL = f"https://desk-api.channel.io/desk/channels/{CHANNEL_ID}/groups/{GROUP_ID}/messages"
-POST_URL = f"https://desk-api.channel.io/desk/channels/{CHANNEL_ID}/groups/{GROUP_ID}/messages"
+POST_URL = GET_URL
+
+PARAMS = {
+    "sortOrder": "asc",  # createdAt 昇順で取得
+    "limit": 50,
+}
 
 X_ACCOUNT = os.getenv("channeliotokenbot2")
 if not X_ACCOUNT:
@@ -21,27 +27,32 @@ HEADERS = {
     "x-account": X_ACCOUNT,
 }
 
-PARAMS = {
-    "sortOrder": "asc",   # 昇順で取得
-    "limit": 50,
-    "logFolded": "false"
-}
-
 ASSISTANT_PERSON_ID = "595702"
 
-# ===== ChatGPT互換API =====
-CHAT_API_URL = "https://izuemon-gpt-free-api.hf.space/v1/chat/completions"
+# ===== GPT (HF Space) =====
+GPT_ENDPOINT = "https://izuemon-gpt-free-api.hf.space/v1/chat/completions"
+
+GPT_HEADERS = {
+    "Content-Type": "application/json"
+}
 
 # ===== Utils =====
-def fetch_channel_messages():
+def parse_created_at(value):
+    if isinstance(value, (int, float)):
+        return datetime.fromtimestamp(value / 1000, tz=timezone.utc)
+    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(messages, limit=10):
+
+def build_chat_messages(messages):
     chat_messages = []
 
-    for msg in messages[-limit:]:
+    for msg in messages:
         text = msg.get("plainText")
         person_id = msg.get("personId")
 
@@ -57,16 +68,16 @@ def build_chat_messages(messages, limit=10):
 
     return chat_messages
 
-def call_chat_api(chat_messages):
+
+def call_gpt(chat_messages):
     payload = {
         "model": "gpt-3.5-turbo",
-        "messages": chat_messages,
-        "temperature": 0.7,
+        "messages": chat_messages
     }
 
     res = requests.post(
-        CHAT_API_URL,
-        headers={"Content-Type": "application/json"},
+        GPT_ENDPOINT,
+        headers=GPT_HEADERS,
         data=json.dumps(payload),
         timeout=60
     )
@@ -75,12 +86,13 @@ def call_chat_api(chat_messages):
     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(
@@ -91,54 +103,55 @@ def send_to_channel(text):
     )
     res.raise_for_status()
 
-# ===== Main =====
+
+# ===== Main Loop =====
 def main():
-    processed_ids = set()
+    last_processed_created_at = 0
 
     while True:
         try:
-            messages = fetch_channel_messages()
+            messages = fetch_messages()
             if not messages:
-                time.sleep(2)
+                time.sleep(10)
                 continue
 
-            # asc なので最後が最新
-            latest = messages[-1]
-            latest_id = latest.get("id")
-            latest_person_id = latest.get("personId")
+            # createdAt が最大のメッセージを確認
+            latest_msg = max(
+                messages,
+                key=lambda m: m.get("createdAt", 0)
+            )
 
-            # すでに処理済み
-            if latest_id in processed_ids:
-                time.sleep(2)
+            latest_created_at = latest_msg.get("createdAt", 0)
+
+            # 既に処理済み
+            if latest_created_at <= last_processed_created_at:
+                time.sleep(10)
                 continue
 
-            # 自分の発言には反応しない
-            if latest_person_id == ASSISTANT_PERSON_ID:
-                processed_ids.add(latest_id)
-                time.sleep(2)
+            # 最後の送信者が assistant なら何もしない
+            if latest_msg.get("personId") == ASSISTANT_PERSON_ID:
+                time.sleep(10)
                 continue
 
+            # ChatGPT 形式に変換
             chat_messages = build_chat_messages(messages)
 
-            if not chat_messages:
-                processed_ids.add(latest_id)
-                continue
+            # GPT 呼び出し
+            reply = call_gpt(chat_messages)
 
-            reply = call_chat_api(chat_messages)
+            # Channel.io に送信
             send_to_channel(reply)
 
-            processed_ids.add(latest_id)
-
-            # メモリ肥大防止
-            if len(processed_ids) > 100:
-                processed_ids = set(list(processed_ids)[-50:])
-
-            print("返信送信完了")
+            last_processed_created_at = latest_created_at
+            print("送信完了")
 
         except Exception as e:
             print("エラー:", e)
+            import traceback
+            traceback.print_exc()
+
+        time.sleep(15)
 
-        time.sleep(2)
 
 if __name__ == "__main__":
     main()