prompt change

agent/context_manager/manager.py

@@ -27,7 +27,7 @@ class ContextManager:
         """Load the system prompt"""
 
         # TODO: get system prompt from jinja template
-        return "You are a helpful assistant."
+        return "You are a helpful assistant, primarly for ML. Do the task you are asked to as efficiently as possible. Don't invent tasks."
 
     def add_message(self, message: Message, token_count: int = None) -> None:
         """Add a message to the history"""

agent/core/session.py

@@ -34,7 +34,7 @@ class Session:
         event_queue: asyncio.Queue,
         config: Config | None = None,
     ):
-        self.context_manager = ContextManager(max_context=4_000, compact_size=0.1, untouched_messages=5)
+        self.context_manager = ContextManager(max_context=180_000, compact_size=0.1, untouched_messages=5)
         self.event_queue = event_queue
         self.session_id = str(uuid.uuid4())
         self.config = config or Config(