aistv 1.3.6__tar.gz → 1.3.7__tar.gz

{aistv-1.3.6 → aistv-1.3.7}/PKG-INFO

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

{aistv-1.3.6 → aistv-1.3.7}/aistv/core.py

@@ -1,10 +1,11 @@
-# aistv.py
 import subprocess
 import sys
 import sqlite3
 import time
+import requests
+from groq import Groq
 
-# Tự cài thư viện nếu thiếu
+# Tự động cài thư viện nếu thiếu
 def install_package(pkg):
     try:
         __import__(pkg)
@@ -14,69 +15,46 @@ def install_package(pkg):
 install_package("groq")
 install_package("requests")
 
-from groq import Groq
+# Cấu hình
+DB_FILE = "usage.db"
+SPAM_DELAY_SECONDS = 5
+FREE_MAX_REQUESTS = 1
+NORMAL_MAX_REQUESTS = 2
+VIP_MAX_REQUESTS = None  # Không giới hạn
 
 API_KEYS = [
     "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",
-    # Thêm key phụ tại đây nếu muốn:
-    # "your_second_key",
 ]
 
-DB_FILE = "usage.db"
-SPAM_DELAY_SECONDS = 5
-
-FREE_MAX_REQUESTS = 20
-NORMAL_MAX_REQUESTS = 30
-VIP_MAX_REQUESTS = None
-
-TOKEN_VIP_SET = {
-    "aistv",
-    "phuc",
-}
-
-TOKEN_NORMAL_SET = {
-    "tokengsk_...",
-}
-
-MODEL_MAP = {
-    "Chat-ai-stv-3.5": "meta-llama/llama-3-8b-instruct",
-    "Chat-ai-stv-4.0": "meta-llama/llama-3-70b-instruct",
-    "Chat-ai-stv-2.3": "meta-llama/llama-4-maverick-17b-128e-instruct",
-    "Chat-ai-stv-4.5": "mistralai/mixtral-8x7b-instruct",
-    "Chat-ai-stv-5.0": "google/gemma-7b-it",
-}
+# Danh sách token
+TOKEN_VIP_SET = {"aistv", "phuc"}
+TOKEN_NORMAL_SET = {"tokengsk_...", "another_token..."}
 
 class STVBot:
-    def __init__(self, token, model, system_prompt=None):
-        if token is None:
-            raise ValueError("Bạn phải truyền token khi khởi tạo aistv")
-        if model is None:
-            raise ValueError("Bạn phải truyền model khi khởi tạo aistv")
-
-        # Bắt buộc model phải nằm trong MODEL_MAP, dùng key nguyên gốc
-        if model not in MODEL_MAP:
-            raise ValueError(f"Mô hình không hợp lệ: {model}. Dùng: {list(MODEL_MAP.keys())}")
-
-        self.model_key = model
-        self.model = MODEL_MAP[self.model_key]
-        self.system_prompt = system_prompt or "Tôi là AI STV, được phát triển bởi Trọng Phúc."
+    def __init__(self, token: str = None, system_prompt: str = None):
         self.token = token
-
-        self.api_key_index = 0
-        self.client = Groq(api_key=API_KEYS[self.api_key_index])
-
+        self.system_prompt = system_prompt or "Tôi là AI STV, được phát triển bởi Trọng Phúc."
+        self.api_keys = API_KEYS
+        self.api_index = 0
+        self.client = self._create_client()
+        self.model = "meta-llama/llama-4-maverick-17b-128e-instruct"
         self.history = [{"role": "system", "content": self.system_prompt}]
 
-        if token in TOKEN_VIP_SET:
-            self.max_requests = VIP_MAX_REQUESTS
+        # Xác định user & giới hạn
+        if not token:
+            self.user_id = "free_user"
+            self.max_requests = FREE_MAX_REQUESTS
+        elif token in TOKEN_VIP_SET:
             self.user_id = token
+            self.max_requests = VIP_MAX_REQUESTS
         elif token in TOKEN_NORMAL_SET:
-            self.max_requests = NORMAL_MAX_REQUESTS
             self.user_id = token
+            self.max_requests = NORMAL_MAX_REQUESTS
         else:
-            self.max_requests = FREE_MAX_REQUESTS
             self.user_id = "free_user"
+            self.max_requests = FREE_MAX_REQUESTS
 
+        # CSDL lưu lượt dùng
         self.conn = sqlite3.connect(DB_FILE, check_same_thread=False)
         self.cursor = self.conn.cursor()
         self.cursor.execute('''
@@ -90,6 +68,9 @@ class STVBot:
         self.conn.commit()
         self._init_user()
 
+    def _create_client(self):
+        return Groq(api_key=self.api_keys[self.api_index])
+
     def _init_user(self):
         today = time.strftime("%Y-%m-%d")
         self.cursor.execute("SELECT * FROM usage WHERE user_id = ?", (self.user_id,))
@@ -103,7 +84,11 @@ class STVBot:
     def _get_usage(self):
         self.cursor.execute("SELECT count, last_time, last_date FROM usage WHERE user_id = ?", (self.user_id,))
         row = self.cursor.fetchone()
-        return {"count": row[0], "last_time": row[1], "last_date": row[2]} if row else {"count": 0, "last_time": 0, "last_date": time.strftime("%Y-%m-%d")}
+        return {
+            "count": row[0],
+            "last_time": row[1],
+            "last_date": row[2]
+        } if row else {"count": 0, "last_time": 0, "last_date": time.strftime("%Y-%m-%d")}
 
     def _save_usage(self, count, last_time, last_date):
         self.cursor.execute(
@@ -112,51 +97,53 @@ class STVBot:
         )
         self.conn.commit()
 
-    def _switch_api_key(self):
-        if self.api_key_index + 1 < len(API_KEYS):
-            self.api_key_index += 1
-            self.client = Groq(api_key=API_KEYS[self.api_key_index])
-            return True
-        return False
-
     def chat(self, prompt: str) -> str:
         usage = self._get_usage()
         now = time.time()
         today = time.strftime("%Y-%m-%d")
 
+        # Reset lượt mỗi ngày
         if usage["last_date"] != today:
             usage["count"] = 0
             usage["last_date"] = today
 
+        # Kiểm tra giới hạn lượt dùng
         if self.max_requests is not None and usage["count"] >= self.max_requests:
-            return f"⚠️ Đã hết {self.max_requests} lượt/ngày. Thử lại mai hoặc xin thêm quyền. https://discord.gg/Ze7RTExgdv"
+            return (
+                f"⚠️ Bạn đã dùng hết {self.max_requests} lượt trong ngày.\n"
+                "Hãy thử lại vào ngày mai hoặc liên hệ để nâng cấp quyền. https://discord.gg/Ze7RTExgdv"
+            )
 
+        # Kiểm tra spam
         if now - usage["last_time"] < SPAM_DELAY_SECONDS:
-            wait_time = SPAM_DELAY_SECONDS - int(now - usage["last_time"])
-            return f"⚠️ Vui lòng chờ thêm {wait_time} giây giữa các câu hỏi."
+            wait = SPAM_DELAY_SECONDS - int(now - usage["last_time"])
+            return f"⚠️ Vui lòng đợi {wait} giây nữa trước khi gửi câu hỏi tiếp theo."
 
         self.history.append({"role": "user", "content": prompt})
+        tries = len(self.api_keys)
 
-        while True:
+        for _ in range(tries):
             try:
-                chat_completion = self.client.chat.completions.create(
+                self.client = self._create_client()
+                response = self.client.chat.completions.create(
                     messages=self.history,
                     model=self.model,
-                    stream=False,
+                    stream=False
                 )
-                reply = chat_completion.choices[0].message.content
+                reply = response.choices[0].message.content.strip()
                 self.history.append({"role": "assistant", "content": reply})
 
+                # Lưu lượt dùng (trừ VIP)
                 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()
+                return reply
 
             except Exception as e:
-                if "quota" in str(e).lower() or "token" in str(e).lower():
-                    if not self._switch_api_key():
-                        return f"⚠️ API hết hạn hoặc lỗi: {e}"
-                else:
-                    return f"⚠️ Lỗi khi gọi API: {e}"
+                self.api_index += 1
+                if self.api_index >= len(self.api_keys):
+                    return f"⚠️ Tất cả API key đều lỗi hoặc hết lượt.\nLỗi cuối cùng: {e}"
+
+        return "⚠️ Đã xảy ra lỗi. Vui lòng thử lại sau."

{aistv-1.3.6 → aistv-1.3.7}/aistv.egg-info/PKG-INFO

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

{aistv-1.3.6 → aistv-1.3.7}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aistv",
-    version="1.3.6",
+    version="1.3.7",
     description="STV AI Chatbot Library for Python",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",