small things

Author: drf7

nuclia/lib/nuclia_nua_chat.py

@@ -14,7 +14,7 @@
 except ImportError:
     raise ImportError(
         "The 'litellm' library is required to use this functionality. "
-        "Install it with: pip install nuclia[ai]"
+        "Install it with: pip install nuclia[litellm]"
     )
 
 # Nuclia (sync & async)
@@ -79,16 +79,8 @@ def _process_messages(self, messages: list[dict[str, str]]) -> tuple[str, str]:
             elif role == "user":
                 user_messages.append(content)
 
-        formatted_system = (
-            "\n--- System Messages ---\n" + "\n".join(system_messages)
-            if system_messages
-            else ""
-        )
-        formatted_user = (
-            "\n--- User Messages ---\n" + "\n".join(user_messages)
-            if user_messages
-            else ""
-        )
+        formatted_system = "\n".join(system_messages) if system_messages else ""
+        formatted_user = "\n".join(user_messages) if user_messages else ""
 
         return formatted_system, formatted_user
 

setup.py

@@ -63,6 +63,6 @@ def load_reqs(filename):
         "Slack": "https://join.slack.com/t/nuclia-community/shared_invite/zt-2ldfznncw-LW6GJjOicdEA18kgI~95Xw",
     },
     extras_require={
-        "ai": ["litellm"],
+        "litellm": ["litellm"],
     },
 )