aistv 1.3.2__py3-none-any.whl → 1.3.4__py3-none-any.whl

aistv/core.py

@@ -17,19 +17,52 @@ import requests
 from groq import Groq
 
 DB_FILE = "usage.db"
-MAX_REQUESTS_PER_DAY = 20
 SPAM_DELAY_SECONDS = 5
+
+# Giới hạn theo loại token
+FREE_MAX_REQUESTS = 20
+NORMAL_MAX_REQUESTS = 30
+VIP_MAX_REQUESTS = None  # Không giới hạn
+
+# Ví dụ token thường và token VIP bạn quản lý
+TOKEN_VIP_SET = {
+    "aistv",
+    "phuc",
+}
+
+TOKEN_NORMAL_SET = {
+    "aistvgsk_wr9rnhdGCQYCaeAEFQusWGdyb3FYF4LVKrxM0I9JDSGkZIVIymwPgsk_wr9rnhdGCQYCaeAEFQusWGdyb3FYF4LVKrxM0I9JDSGkZIVIymwPgsk_wr9rnhdGCQYCaeAEFQusWGdyb3FYF4LVKrxM0I9JDSGkZIVIymwPgsk_wr9rnhdGCQYCaeAEFQusWGdyb3FYF4LVKrxM0I9JDSGkZIVIymwP",
+    "tokengsk_wr9rnhdGCQYCaeAEFQusWGdyb3FYF4LVKrxM0I9JDSGkZIVIymwP",
+}
+
 API_KEY = "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"
 
 class STVBot:
-    def __init__(self, user_id: str, system_prompt: str = None):
+    def __init__(self, token: str = None, system_prompt: str = None):
+        """
+        token: None (free user), or token string (normal or VIP)
+        """
         if not system_prompt:
             system_prompt = "Tôi là AI STV, được phát triển bởi Trọng Phúc."
         self.client = Groq(api_key=API_KEY)
         self.system_prompt = system_prompt
         self.model = "meta-llama/llama-4-maverick-17b-128e-instruct"
         self.history = [{"role": "system", "content": self.system_prompt}]
-        self.user_id = user_id
+
+        self.token = token
+        if token is None:
+            self.max_requests = FREE_MAX_REQUESTS
+            self.user_id = "free_user"
+        elif token in TOKEN_VIP_SET:
+            self.max_requests = VIP_MAX_REQUESTS
+            self.user_id = token
+        elif token in TOKEN_NORMAL_SET:
+            self.max_requests = NORMAL_MAX_REQUESTS
+            self.user_id = token
+        else:
+            # Token không hợp lệ tính free
+            self.max_requests = FREE_MAX_REQUESTS
+            self.user_id = "free_user"
 
         # Kết nối DB SQLite
         self.conn = sqlite3.connect(DB_FILE, check_same_thread=False)
@@ -83,10 +116,10 @@ class STVBot:
             usage["count"] = 0
             usage["last_date"] = today
 
-        if usage["count"] >= MAX_REQUESTS_PER_DAY:
+        if self.max_requests is not None and usage["count"] >= self.max_requests:
             return (
-                "⚠️ Bạn đã sử dụng hết giới hạn 20 câu hỏi miễn phí trong ngày.\n"
-                "Vui lòng thử lại vào ngày mai hoặc liên hệ để được cấp thêm quyền."
+                f"⚠️ Bạn đã sử dụng hết giới hạn {self.max_requests} câu hỏi trong ngày.\n"
+                "Vui lòng thử lại vào ngày mai hoặc liên hệ để được cấp thêm quyền.https://discord.gg/Ze7RTExgdv"
             )
 
         if now - usage["last_time"] < SPAM_DELAY_SECONDS:
@@ -104,11 +137,12 @@ class STVBot:
             reply = chat_completion.choices[0].message.content
             self.history.append({"role": "assistant", "content": reply})
 
-            usage["count"] += 1
-            usage["last_time"] = now
-            self._save_usage(usage["count"], usage["last_time"], usage["last_date"])
+            if self.max_requests is not None:
+                usage["count"] += 1
+                usage["last_time"] = now
+                self._save_usage(usage["count"], usage["last_time"], usage["last_date"])
 
             return reply.strip()
 
         except Exception as e:
-            return f"⚠️ Lỗi khi: {e}"
+            return f"⚠️ Lỗi khi gọi API: {e}"

{aistv-1.3.2.dist-info → aistv-1.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aistv
-Version: 1.3.2
+Version: 1.3.4
 Summary: STV AI Chatbot Library for Python
 Home-page: https://github.com/phuctrong1tuv
 Author: Trọng Phúc

aistv-1.3.4.dist-info/RECORD

@@ -0,0 +1,8 @@
+aistv/__init__.py,sha256=G1ZBskWNkKKEK9s51RrBuK-S2_d3Fw8UaLk0RKsbuKw,33
+aistv/aistv.py,sha256=8OClk7fg1g41NtW7ELj83-g0JF5fV1fQy0dJh2qlV8s,190
+aistv/core.py,sha256=U76vDxmN2310PCGV5YE8DXhkyutKN-6Dmk8ISHWeLsY,5437
+aistv/trainer.py,sha256=oqIHGAJAJgIdw3P6_BrnXt3JMXYwsGHfGVaJ_e635YU,1125
+aistv-1.3.4.dist-info/METADATA,sha256=xk-PfzOHD5PhC_GUDsa1EpM0IXHR_Gn2JgJ6MtXRCg8,4244
+aistv-1.3.4.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+aistv-1.3.4.dist-info/top_level.txt,sha256=P6gyYzXFvBPESQyFZQSi6Rs8v5ALhw-d4UN2Jo9Pkgw,6
+aistv-1.3.4.dist-info/RECORD,,

aistv-1.3.2.dist-info/RECORD

@@ -1,8 +0,0 @@
-aistv/__init__.py,sha256=G1ZBskWNkKKEK9s51RrBuK-S2_d3Fw8UaLk0RKsbuKw,33
-aistv/aistv.py,sha256=8OClk7fg1g41NtW7ELj83-g0JF5fV1fQy0dJh2qlV8s,190
-aistv/core.py,sha256=DxOgKiRzBBabccLnDqZXwWUEQIn74J-AO_5EwEvIO2Q,4129
-aistv/trainer.py,sha256=oqIHGAJAJgIdw3P6_BrnXt3JMXYwsGHfGVaJ_e635YU,1125
-aistv-1.3.2.dist-info/METADATA,sha256=iBLrsJMaCZZB26NSoezElgxGO1rN2UyOrCPJhE6Tu-Q,4244
-aistv-1.3.2.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-aistv-1.3.2.dist-info/top_level.txt,sha256=P6gyYzXFvBPESQyFZQSi6Rs8v5ALhw-d4UN2Jo9Pkgw,6
-aistv-1.3.2.dist-info/RECORD,,