dao-ai 0.1.16__py3-none-any.whl → 0.1.18__py3-none-any.whl

dao_ai/cli.py

@@ -64,6 +64,7 @@ def detect_cloud_provider(profile: Optional[str] = None) -> Optional[str]:
     """
     try:
         import os
+
         from databricks.sdk import WorkspaceClient
 
         # Check for environment variables that might override profile
@@ -520,14 +521,19 @@ def handle_chat_command(options: Namespace) -> None:
                             )
                             continue
 
+                # Normalize user_id for memory namespace compatibility (replace . with _)
+                # This matches the normalization in models.py _convert_to_context
+                if configurable.get("user_id"):
+                    configurable["user_id"] = configurable["user_id"].replace(".", "_")
+
                 # Create Context object from configurable dict
                 from dao_ai.state import Context
 
                 context = Context(**configurable)
 
-                # Prepare config with thread_id for checkpointer
-                # Note: thread_id is needed in config for checkpointer/memory
-                config = {"configurable": {"thread_id": options.thread_id}}
+                # Prepare config with all context fields for checkpointer/memory
+                # Note: langmem tools require user_id in config.configurable for namespace resolution
+                config = {"configurable": context.model_dump()}
 
                 # Invoke the graph and handle interrupts (HITL)
                 # Wrap in async function to maintain connection pool throughout
@@ -1208,7 +1214,9 @@ def run_databricks_command(
             f"Using CLI-specified deployment target: {resolved_deployment_target}"
         )
     elif app_config and app_config.app and app_config.app.deployment_target:
-        resolved_deployment_target = app_config.app.deployment_target.value
+        # deployment_target is DeploymentTarget enum (str subclass) or string
+        # str() works for both since DeploymentTarget inherits from str
+        resolved_deployment_target = str(app_config.app.deployment_target)
         logger.debug(
             f"Using config file deployment target: {resolved_deployment_target}"
         )