Merge pull request #280 from crestalnetwork/feat/owner-mode

Feat: clean memory after agent update
crestalnetwork · Feb 24, 2025 · c9178b7 · c9178b7
2 parents ad1b52d + 1b719b7
commit c9178b7
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/app/admin/api.py b/app/admin/api.py
@@ -66,6 +66,8 @@ async def _process_agent(
         if agent_data and agent_data.cdp_wallet_data:
             has_wallet = True
             wallet_data = json.loads(agent_data.cdp_wallet_data)
+        # Clean agent memory
+        await clean_agent_memory(latest_agent.id, clean_agent_memory=True)
 
     if not has_wallet:
         # create the wallet
@@ -131,9 +133,9 @@ async def _process_agent(
                     {"title": "Name", "short": True, "value": latest_agent.name},
                     {"title": "Model", "short": True, "value": latest_agent.model},
                     {
-                        "title": "Enso Enabled",
+                        "title": "GOAT Enabled",
                         "short": True,
-                        "value": str(latest_agent.enso_enabled),
+                        "value": str(latest_agent.goat_enabled),
                     },
                     {
                         "title": "CDP Enabled",

diff --git a/app/config/config.py b/app/config/config.py
@@ -88,6 +88,7 @@ def __init__(self):
         self.deepseek_api_key = self.load("DEEPSEEK_API_KEY")
         self.xai_api_key = self.load("XAI_API_KEY")
         self.system_prompt = self.load("SYSTEM_PROMPT")
+        self.input_token_limit = int(self.load("INPUT_TOKEN_LIMIT", "60000"))
         # Telegram server settings
         self.tg_base_url = self.load("TG_BASE_URL")
         self.tg_server_host = self.load("TG_SERVER_HOST", "127.0.0.1")

diff --git a/app/core/engine.py b/app/core/engine.py
@@ -124,7 +124,7 @@ async def initialize_agent(aid):
         raise HTTPException(status_code=500, detail=str(e))
 
     # ==== Initialize LLM.
-    input_token_limit = 120000
+    input_token_limit = config.input_token_limit
     # TODO: model name whitelist
     if agent.model.startswith("deepseek"):
         llm = ChatOpenAI(
@@ -136,7 +136,8 @@ async def initialize_agent(aid):
             temperature=agent.temperature,
             timeout=300,
         )
-        input_token_limit = 60000
+        if input_token_limit > 60000:
+            input_token_limit = 60000
     elif agent.model.startswith("grok"):
         llm = ChatXAI(
             model_name=agent.model,
@@ -146,6 +147,8 @@ async def initialize_agent(aid):
             temperature=agent.temperature,
             timeout=180,
         )
+        if input_token_limit > 120000:
+            input_token_limit = 120000
     else:
         llm = ChatOpenAI(
             model_name=agent.model,
@@ -155,6 +158,8 @@ async def initialize_agent(aid):
             temperature=agent.temperature,
             timeout=180,
         )
+        if input_token_limit > 120000:
+            input_token_limit = 120000
 
     # ==== Store buffered conversation history in memory.
     memory = AsyncPostgresSaver(get_pool())