[Google Jules] I've identified some potential performance improvements and a bug fix.

app/bot/dialogue_manager/models.py

@@ -1,5 +1,5 @@
 from typing import Optional, Dict, List, Any, Text
-from datetime import datetime, UTC
+from datetime import datetime, timezone
 from copy import deepcopy
 from dataclasses import dataclass
 from app.admin.intents.schemas import Intent
@@ -104,7 +104,7 @@ def __init__(
         self.current_node = current_node
         self.parameters = parameters or []
         self.owner = owner
-        self.date = date or datetime.now(UTC).isoformat()
+        self.date = date or datetime.now(timezone.utc).isoformat()
 
     @classmethod
     def from_json(cls, request_json: Dict):

app/bot/memory/models.py

@@ -1,5 +1,5 @@
 from typing import Optional, Dict, List, Any, Text
-from datetime import datetime, UTC
+from datetime import datetime, timezone
 from app.bot.dialogue_manager.models import UserMessage
 
 
@@ -29,7 +29,7 @@ def __init__(
         self.missing_parameters = missing_parameters or []
         self.complete = complete
         self.current_node = current_node
-        self.date = date or datetime.now(UTC)
+        self.date = date or datetime.now(timezone.utc)
 
     def to_dict(self) -> Dict:
         return {
@@ -63,7 +63,7 @@ def from_dict(cls, state_dict: Dict) -> "State":
 
     def update(self, user_message: UserMessage):
         self.user_message = user_message
-        self.date = datetime.now(UTC)
+        self.date = datetime.now(timezone.utc)
         self.context.update(user_message.context)
 
         if self.complete:

docker-compose.yml

@@ -1,5 +1,5 @@
 version: '2'
-name: ai-chatbot-framework
+# name: ai-chatbot-framework
 services:
   frontend:
     image: alfredfrancis/ai-chatbot-framework_frontend:latest

manage.py

@@ -4,7 +4,7 @@
 import logging
 import spacy
 from pymongo.errors import DuplicateKeyError
 from app.main import app
 
 cli = typer.Typer()
 
@@ -51,7 +51,7 @@
         from app.bot.nlu.pipeline_utils import train_pipeline
 
         logger.info("Training models...")
-        await train_pipeline(app)
+        await train_pipeline()
         logger.info("Training models finished.")
 
     asyncio.run(async_train())

requirements.txt

@@ -45,7 +45,7 @@ mdurl==0.1.2
 motor==3.6.1
 multidict==6.1.0
 murmurhash==1.0.12
-numpy==2.2.2
+numpy~=1.26.0
 openai==1.60.2
 orjson==3.10.15
 packaging==24.2