bypass limit

Author: Adam Balogh

agent/agent_executors.py

@@ -58,7 +58,7 @@ def create_routing_model() -> ChatOpenAI:
         openai_api_base=BASE_URL,
         openai_api_key=API_KEY,
         request_timeout=60,
-        default_headers={"X-Title": "two-ligma"},
+        default_headers={"X-Title": "opengradient-bitquant"},
     )
 
 
@@ -71,7 +71,7 @@ def create_suggestions_model() -> ChatOpenAI:
         request_timeout=60,
         max_tokens=500,
         streaming=False,
-        default_headers={"X-Title": "two-ligma"},
+        default_headers={"X-Title": "opengradient-bitquant"},
     )
 
 
@@ -84,7 +84,7 @@ def create_investor_executor() -> CompiledGraph:
         request_timeout=60,
         max_tokens=4096,
         streaming=False,
-        default_headers={"X-Title": "two-ligma"},
+        default_headers={"X-Title": "opengradient-bitquant"},
     )
     agent_executor = create_react_agent(
         model=openai_model, tools=create_investor_agent_toolkit()
@@ -102,7 +102,7 @@ def create_analytics_executor(token_metadata_repo: TokenMetadataRepo) -> Compile
         request_timeout=60,
         max_tokens=4096,
         streaming=False,
-        default_headers={"X-Title": "two-ligma"},
+        default_headers={"X-Title": "opengradient-bitquant"},
     )
     analytics_executor = create_react_agent(
         model=openai_model,

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
-name = "two-ligma"
+name = "bitquant"
 version = "0.1.0"
-description = "OpenGradient TwoLigma"
+description = "OpenGradient BitQuant"
 requires-python = ">=3.8"
 
 [tool.black]

server/activity_tracker.py

@@ -3,6 +3,8 @@
 from datetime import datetime, timezone
 from typing import Optional
 
+from config import MINER_WALLET_ADDRESS
+
 
 @dataclass
 class ActivityStats:
@@ -51,8 +53,8 @@ def increment_message_count(self, user_address: str) -> bool:
             if last_message_date != today:
                 daily_message_count = 0
 
-            # Check if daily limit reached
-            if daily_message_count >= self.DAILY_MESSAGE_LIMIT:
+            # Check if daily limit reached, except for Subnet miner wallet
+            if daily_message_count >= self.DAILY_MESSAGE_LIMIT and user_address != MINER_WALLET_ADDRESS:
                 return False
 
             # Update both total and daily message counts

server/config.py

@@ -4,3 +4,6 @@
 # See if we are running in subnet mode
 SUBNET_MODE = os.getenv("subnet_mode", "false").lower() == "true"
 logging.info(f"Running in subnet mode: {SUBNET_MODE}")
+
+# Bypass daily limit for miner wallet
+MINER_WALLET_ADDRESS = "7FVPurQDkbj6g9dm5B52oCUr7JxqRpoYQcNitKaWVSgS"