Trying to rebase with main and commit the files again

Signed-off-by: Vineeth Kalluru <[email protected]>

# Conflicts:
#	industries/asset_lifecycle_management_agent/README.md
#	industries/asset_lifecycle_management_agent/configs/config-reasoning.yml

Author: vikalluru

industries/asset_lifecycle_management_agent/.gitignore

@@ -1,46 +1,122 @@
-# macOS system files
+# Misc
+config_examples.yml
+env.sh
+frontend/
+prompts.md
+
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+*.egg
+*.egg-info/
+dist/
+build/
+*.whl
+pip-wheel-metadata/
 .DS_Store
-.DS_Store?
-._*
-.Spotlight-V100
-.Trashes
-ehthumbs.db
-Thumbs.db
 
-# Database and vector store files
-database/
-*.db
-*.sqlite3
+# Virtual environments
+.venv/
+venv/
+ENV/
+env/
+
+# IDEs and Editors
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+.DS_Store
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+.tox/
+.hypothesis/
+
+# Jupyter Notebook
+.ipynb_checkpoints/
+*.ipynb_checkpoints/
 
-# Output and generated files
+# Output and Data Directories
 output_data/
-moment/
-readmes/
-*.html
-*.csv
-*.npy
+eval_output/
+example_eval_output/
+output/
+results/
+logs/
 
-# Python package metadata
-src/**/*.egg-info/
-*.egg-info/
+# Database files
+*.db
+*.sqlite
+*.sqlite3
+database/*.db
+database/*.sqlite
 
-# Environment files (if they contain secrets)
-env.sh
+# Vector store data (ChromaDB)
+database/
+chroma_db/
+vector_store/
+vanna_vector_store/
 
-# Model files (if large/binary)
+# Model files (large binary files)
 models/*.pkl
-models/*.joblib
-models/*.model
+models/*.h5
+models/*.pt
+models/*.pth
+models/*.ckpt
+*.pkl
+*.h5
+*.pt
+*.pth
+moment/
 
-# Logs
-*.log
-logs/
+# Data files (CSV, JSON, etc. - be selective)
+*.csv
+*.json
+!training_data.json
+!vanna_training_data.yaml
+!config*.json
+!config*.yaml
+!config*.yml
+!pyproject.toml
+!package.json
+
+# Frontend build artifacts
+frontend/node_modules/
+frontend/dist/
+frontend/build/
+frontend/.next/
+frontend/out/
+
+# Environment and secrets
+.env
+.env.local
+.env.*.local
+*.secret
+secrets/
+credentials/
 
 # Temporary files
 *.tmp
 *.temp
-.pytest_cache/
-__pycache__/ 
+*.log
+*.cache
+
+# OS specific
+Thumbs.db
+Desktop.ini
+
+# Experiment tracking
+mlruns/
+wandb/
 
-# dot env
-mydot.env
+# Documentation builds
+docs/_build/
+docs/.doctrees/
+site/

industries/asset_lifecycle_management_agent/README.md

@@ -175,6 +175,21 @@ Now install the ALM workflow:
 uv pip install -e .
 ```
 
+#### Installation Options
+
+**Base Installation** (default - includes ChromaDB + SQLite):
+```bash
+uv pip install -e .
+```
+
+**Optional Database Support:**
+- PostgreSQL: `uv pip install -e ".[postgres]"`
+- MySQL: `uv pip install -e ".[mysql]"`
+- All databases: `uv pip install -e ".[all-databases]"`
+
+**Optional Vector Store:**
+- Elasticsearch: `uv pip install -e ".[elasticsearch]"`
+
 ### [Optional] Verify if all prerequisite packages are installed
 ```bash
 uv pip list | grep -E "nvidia-nat|nvidia-nat-ragaai|nvidia-nat-phoenix|vanna|chromadb|xgboost|pytest|torch|matplotlib"
@@ -463,7 +478,9 @@ def your_custom_utility(file_path: str, param: int = 100) -> str:
 4. **Consistent Interface**: All utilities return descriptive success messages
 5. **Documentation**: Use `utils.show_utilities()` to discover available functions
 
-### Setup Web Interface
+### Alternative: Generic NeMo-Agent-Toolkit UI
+
+If you prefer the generic NeMo Agent Toolkit UI instead of our custom interface:
 
 ```bash
 git clone https://github.com/NVIDIA/NeMo-Agent-Toolkit-UI.git
@@ -509,7 +526,7 @@ Retrieve and detect anomalies in sensor 4 measurements for engine number 78 in t
 
 **Workspace Utilities Demo**
 ```
-Retrieve ground truth RUL values and time in cycles from FD001 train dataset. Apply piecewise RUL transformation with MAXLIFE=100. Finally, Plot a line chart of the transformed values across time.
+Retrieve RUL values and time in cycles for engine unit 24 from FD001 train dataset. Use the piece wise RUL transformation code utility to perform piecewise RUL transformation on the ground truth RUL values with MAXLIFE=100.Finally, Plot a comparison line chart with RUL values and its transformed values across time.
 ```
 
 *This example demonstrates how to discover and use workspace utilities directly. The system will show available utilities and then apply the RUL transformation using the pre-built, reliable utility functions.*
@@ -518,9 +535,9 @@ Retrieve ground truth RUL values and time in cycles from FD001 train dataset. Ap
 ```
 Perform the following steps:
 
-1.Retrieve the time in cycles, all sensor measurements, and ground truth RUL values for engine unit 24 from FD001 train dataset.
+1.Retrieve the time in cycles, all sensor measurements, and ground truth RUL values, partition by unit number for engine unit 24 from FD001 train dataset.
 2.Use the retrieved data to predict the Remaining Useful Life (RUL). 
-3.Use the piece wise RUL transformation code utility to apply piecewise RUL transformation only to the observed RUL column. 
+3.Use the piece wise RUL transformation code utility to apply piecewise RUL transformation only to the observed RUL column with MAXLIFE of 100.
 4.Generate a plot that compares the transformed RUL values and the predicted RUL values across time.
 ```
 ![Prediction Example](imgs/test_prompt_3.png)

industries/asset_lifecycle_management_agent/configs/config-reasoning.yml

@@ -19,21 +19,37 @@ general:
     logging:
       console:
         _type: console
+<<<<<<< HEAD:industries/asset_lifecycle_management_agent/configs/config-reasoning.yml
         level: DEBUG
     #     level: INFO
     #   file:
     #     _type: file
     #     path: "alm.log"
     #     level: DEBUG
+=======
+        level: INFO
+      file:
+        _type: file
+        path: "pdm.log"
+        level: DEBUG
+>>>>>>> fbde275 (Trying to rebase with main and commit the files again):industries/predictive_maintenance_agent/configs/config-reasoning.yml
     tracing:
       phoenix:
         _type: phoenix
         endpoint: http://localhost:6006/v1/traces
+<<<<<<< HEAD:industries/asset_lifecycle_management_agent/configs/config-reasoning.yml
         project: alm-agent
     #   catalyst:
     #     _type: catalyst
     #     project: "alm-agent"
     #     dataset: "alm-agent"
+=======
+        project: pdm-demo-day
+      catalyst:
+        _type: catalyst
+        project: "pdm-demo-day"
+        dataset: "pdm-demo-day"
+>>>>>>> fbde275 (Trying to rebase with main and commit the files again):industries/predictive_maintenance_agent/configs/config-reasoning.yml
 
 llms:
   # SQL query generation model
@@ -43,13 +59,11 @@ llms:
 
   # Data analysis and tool calling model
   analyst_llm:
-    _type: nim
-    model_name: "qwen/qwen2.5-coder-32b-instruct"
-    # _type: openai
-    # model_name: "gpt-4.1-mini"
+    _type: openai
+    model_name: "gpt-4.1-mini"
 
   # Python code generation model
-  coding_llm:
+  coding_llm: 
     _type: nim
     model_name: "qwen/qwen2.5-coder-32b-instruct"
 
@@ -67,15 +81,20 @@ embedders:
   # Text embedding model for vector database operations
   vanna_embedder:
     _type: nim
-    model_name: "nvidia/nv-embed-v1"
+    model_name: "nvidia/llama-3.2-nv-embedqa-1b-v2"
 
 functions:
   sql_retriever:
     _type: generate_sql_query_and_retrieve_tool
     llm_name: sql_llm
     embedding_name: vanna_embedder
+    # Vector store configuration
+    vector_store_type: chromadb  # Optional, chromadb is default
     vector_store_path: "database"
-    db_path: "database/nasa_turbo.db"
+    # Database configuration
+    db_type: sqlite  # Optional, sqlite is default
+    db_connection_string_or_path: "database/nasa_turbo.db"
+    # Output configuration
     output_folder: "output_data"
     vanna_training_data_path: "vanna_training_data.yaml"
 
@@ -129,8 +148,8 @@ functions:
       plot_line_chart, 
       plot_comparison, 
       anomaly_detection, 
-      plot_anomaly,
-      code_generation_assistant
+      plot_anomaly
+      # code_generation_assistant
     ]
     parse_agent_response_max_retries: 2
     system_prompt: |

industries/asset_lifecycle_management_agent/pyproject.toml

@@ -11,6 +11,7 @@ dependencies = [
   "pydantic ~= 2.10.0, <2.11.0",
   "vanna==0.7.9",
   "chromadb",
+  "sqlalchemy>=2.0.0",
   "xgboost",
   "matplotlib",
   "torch",
@@ -23,6 +24,36 @@ classifiers = ["Programming Language :: Python"]
 authors = [{ name = "Vineeth Kalluru" }]
 maintainers = [{ name = "NVIDIA Corporation" }]
 
+[project.optional-dependencies]
+elasticsearch = [
+  "elasticsearch>=8.0.0"
+]
+postgres = [
+  "psycopg2-binary>=2.9.0"
+]
+mysql = [
+  "pymysql>=1.0.0"
+]
+sqlserver = [
+  "pyodbc>=4.0.0"
+]
+oracle = [
+  "cx_Oracle>=8.0.0"
+]
+all-databases = [
+  "psycopg2-binary>=2.9.0",
+  "pymysql>=1.0.0",
+  "pyodbc>=4.0.0",
+  "cx_Oracle>=8.0.0"
+]
+all = [
+  "elasticsearch>=8.0.0",
+  "psycopg2-binary>=2.9.0",
+  "pymysql>=1.0.0",
+  "pyodbc>=4.0.0",
+  "cx_Oracle>=8.0.0"
+]
+
 [project.entry-points.'nat.components']
 asset_lifecycle_management_agent = "asset_lifecycle_management_agent.register"