chore: update imports in debugging assistant

feat: add config_path param to model initialization functionality

Author: maciejmajek

examples/debugging_assistant.py

@@ -14,8 +14,8 @@
 
 import streamlit as st
 from rai import get_llm_model
-from rai.agents.conversational_agent import create_conversational_agent
-from rai.frontend.streamlit import run_streamlit_app
+from rai.agents import create_conversational_agent
+from rai.frontend import run_streamlit_app
 from rai.tools.ros2 import ROS2CLIToolkit
 
 

src/rai_core/rai/initialization/model_initialization.py

@@ -85,9 +85,13 @@ class RAIConfig:
     tracing: TracingConfig
 
 
-def load_config() -> RAIConfig:
-    with open("config.toml", "rb") as f:
-        config_dict = tomli.load(f)
+def load_config(config_path: Optional[str] = None) -> RAIConfig:
+    if config_path is None:
+        with open("config.toml", "rb") as f:
+            config_dict = tomli.load(f)
+    else:
+        with open(config_path, "rb") as f:
+            config_dict = tomli.load(f)
     return RAIConfig(
         vendor=VendorConfig(**config_dict["vendor"]),
         aws=AWSConfig(**config_dict["aws"]),
@@ -104,8 +108,9 @@ def load_config() -> RAIConfig:
 def get_llm_model_config_and_vendor(
     model_type: Literal["simple_model", "complex_model"],
     vendor: Optional[str] = None,
+    config_path: Optional[str] = None,
 ) -> Tuple[str, str]:
-    config = load_config()
+    config = load_config(config_path)
     if vendor is None:
         if model_type == "simple_model":
             vendor = config.vendor.simple_model
@@ -119,9 +124,12 @@ def get_llm_model_config_and_vendor(
 def get_llm_model(
     model_type: Literal["simple_model", "complex_model"],
     vendor: Optional[str] = None,
+    config_path: Optional[str] = None,
     **kwargs,
 ):
-    model_config, vendor = get_llm_model_config_and_vendor(model_type, vendor)
+    model_config, vendor = get_llm_model_config_and_vendor(
+        model_type, vendor, config_path
+    )
     model = getattr(model_config, model_type)
     logger.info(f"Initializing {model_type}: Vendor: {vendor}, Model: {model}")
     if vendor == "openai":
@@ -149,8 +157,8 @@ def get_llm_model(
         raise ValueError(f"Unknown LLM vendor: {vendor}")
 
 
-def get_embeddings_model(vendor: str = None):
-    config = load_config()
+def get_embeddings_model(vendor: str = None, config_path: Optional[str] = None):
+    config = load_config(config_path)
     if vendor is None:
         vendor = config.vendor.embeddings_model