Cognition UI Facelift (#174)

* New fields

* Update porject icon

* helper func

Author: JWittmeyer

cognition_objects/project.py

@@ -1,7 +1,7 @@
 from typing import List, Optional, Dict, Any, Iterable
 from ..business_objects import general, team_resource, user
 from ..cognition_objects import consumption_log, consumption_summary
-from ..session import session, request_id_ctx_var
+from ..session import session
 from ..models import CognitionProject, TeamMember, TeamResource
 from .. import enums
 from datetime import datetime
@@ -214,6 +214,7 @@ def update(
     macro_config: Optional[Dict[str, Any]] = None,
     llm_config: Optional[Dict[str, Any]] = None,
     tokenizer: Optional[str] = None,
+    icon: Optional[str] = None,
     with_commit: bool = True,
 ) -> CognitionProject:
     project: CognitionProject = get(project_id)
@@ -285,6 +286,8 @@ def update(
         flag_modified(project, "operator_routing_config")
     if tokenizer is not None:
         project.tokenizer = tokenizer
+    if icon is not None:
+        project.icon = icon
     general.flush_or_commit(with_commit)
     return project
 

models.py

@@ -226,6 +226,7 @@ class User(Base):
     created_at = Column(DateTime, default=sql.func.now())
     metadata_public = Column(JSON)
     sso_provider = Column(String)
+    use_new_cognition_ui = Column(Boolean, default=True)
 
 
 class Team(Base):
@@ -1110,6 +1111,8 @@ class CognitionProject(Base):
     # holds e.g. show, admin macro setting etc.
     macro_config = Column(JSON)
     tokenizer = Column(String)
+    # options from <SVGIcon/> component - only visible with new UI selected (user setting)
+    icon = Column(String, default="IconBolt")
 
 
 class CognitionStrategy(Base):

util.py

@@ -16,6 +16,12 @@
 
 CAMEL_CASE_PATTERN = compile(r"^([a-z]+[A-Z]?)*$")
 
+STRING_TRUE_VALUES = {"true", "x", "1", "y"}
+
+
+def is_string_true_value(value: str) -> bool:
+    return value.lower() in STRING_TRUE_VALUES
+
 
 def collect_engine_variables() -> Tuple[int, int, bool, bool]:
     # amount of simultaneous connections to the database
@@ -50,7 +56,7 @@ def collect_engine_variables() -> Tuple[int, int, bool, bool]:
     os_pool_use_lifo = os.getenv("POSTGRES_POOL_USE_LIFO")
     if os_pool_use_lifo:
         try:
-            pool_use_lifo = os_pool_use_lifo.lower() in ["true", "x", "1", "y"]
+            pool_use_lifo = is_string_true_value(os_pool_use_lifo)
         except ValueError:
             print(
                 f"POSTGRES_POOL_USE_LIFO is not an boolean, using default {pool_use_lifo}",
@@ -64,7 +70,7 @@ def collect_engine_variables() -> Tuple[int, int, bool, bool]:
     os_pool_pre_ping = os.getenv("POSTGRES_POOL_PRE_PING")
     if os_pool_pre_ping:
         try:
-            pool_pre_ping = os_pool_pre_ping.lower() in ["true", "x", "1", "y"]
+            pool_pre_ping = is_string_true_value(os_pool_pre_ping)
         except ValueError:
             print(
                 f"POSTGRES_POOL_PRE_PING is not an boolean, using default {pool_pre_ping}",