user-simulator 0.1.11__py3-none-any.whl → 0.2.0__py3-none-any.whl

user_sim/cli/sensei_chat.py

@@ -4,8 +4,8 @@ import pandas as pd
 from collections import Counter
 from argparse import ArgumentParser
 from colorama import Fore, Style
-from technologies.chatbot_connectors import (Chatbot, ChatbotRasa, ChatbotTaskyto, ChatbotMillionBot,
-                                             ChatbotServiceform)
+# from technologies.chatbot_connectors import (Chatbot, ChatbotRasa, ChatbotTaskyto, ChatbotMillionBot,
+#                                              ChatbotServiceform)
 from user_sim.core.data_extraction import DataExtraction
 from user_sim.core.role_structure import *
 from user_sim.core.user_simulator import UserSimulator
@@ -14,6 +14,14 @@ from user_sim.utils.utilities import *
 from user_sim.utils.token_cost_calculator import create_cost_dataset
 from user_sim.utils.register_management import clean_temp_files
 
+from chatbot_connectors.implementations.custom import CustomChatbot
+from chatbot_connectors.implementations.taskyto import ChatbotTaskyto
+from chatbot_connectors.implementations.rasa import RasaChatbot
+from chatbot_connectors.implementations.millionbot import MillionBot
+
+from chatbot_connectors import ChatbotFactory, parse_connector_params
+
+
 # check_keys(["OPENAI_API_KEY"])
 current_script_dir = os.path.dirname(os.path.abspath(__file__))
 root_path = os.path.abspath(os.path.join(current_script_dir, ".."))
@@ -233,17 +241,18 @@ def parse_profiles(user_path):
         raise Exception(f'Invalid path for user profile operation: {user_path}')
 
 
-def build_chatbot(technology, connector) -> Chatbot:
-    chatbot_builder = {
-        'rasa': ChatbotRasa,
-        'taskyto': ChatbotTaskyto,
-        'serviceform': ChatbotServiceform,
-        'millionbot': ChatbotMillionBot
-    }
-    chatbot_class = chatbot_builder.get(technology, Chatbot)
-    return chatbot_class(connector)
-
-
+def build_chatbot(technology, connector):
+    # chatbot_builder = {
+    #     'rasa': RasaChatbot,
+    #     'taskyto': ChatbotTaskyto,
+    #     # 'serviceform': ChatbotServiceform(connector),
+    #     'millionbot': MillionBot,
+    #     'custom': CustomChatbot
+    # }
+    # chatbot_class = chatbot_builder.get(technology, CustomChatbot)
+    parsed_connector = parse_connector_params(connector)
+    chatbot = ChatbotFactory.create_chatbot(chatbot_type=technology, **parsed_connector)
+    return chatbot
 
 def generate_conversation(technology, connector, user,
                           personality, extract, project_folder):
@@ -412,13 +421,12 @@ def main():
     parser.add_argument('--technology', required=False,
                         choices=['rasa', 'taskyto', 'ada-uam', 'millionbot', 'genion', 'lola', 'serviceform', 'kuki', 'julie', 'rivas_catalina', 'saic_malaga'],
                         help='Technology the chatbot is implemented in')
-    # parser.add_argument('--chatbot', required=False, help='URL where the chatbot is deployed')
-    parser.add_argument('--connector', required=False, help='path to the connector configuration file')
-    parser.add_argument('--connector_parameters', required=False, help='dynamic parameters for the selected chatbot connector')
+    # parser.add_argument('--connector', required=False, help='path to the connector configuration file')
+    parser.add_argument('--connector-params', required=False, help='dynamic parameters for the selected chatbot connector')
     parser.add_argument('--project_path', required=False, help='Project folder PATH where all testing data is stored')
     parser.add_argument('--user_profile', required=False, help='User profile file or user profile folder to test the chatbot')
     parser.add_argument('--personality', required=False, help='Personality file')
-    parser.add_argument("--extract", default=False, help='Path to store conversation user-chatbot')
+    parser.add_argument('--extract', default=False, help='Path to store conversation user-chatbot')
     parser.add_argument('--verbose', action='store_true', help='Shows debug prints')
     parser.add_argument('--clean_cache', action='store_true', help='Deletes temporary files.')
     parser.add_argument('--ignore_cache', action='store_true', help='Ignores cache for temporary files')
@@ -449,7 +457,7 @@ def main():
         parser_args = ArgsNamespace(**yaml_args)
 
     else:
-        required_args = ['technology', 'user_profile', 'connector']
+        required_args = ['technology', 'user_profile', 'connector_params']
         missing_args = [arg for arg in required_args if getattr(parser_args, arg) is None]
 
         if missing_args:
@@ -468,14 +476,18 @@ def main():
     config.update_cache = parser_args.update_cache
     config.clean_cache = parser_args.clean_cache
 
-    if parser_args.connector_parameters:
-        connector = configure_connector(parser_args.connector, parser_args.connector_parameters)
-    else:
-        connector = configure_connector(parser_args.connector)
+    # if parser_args.connector_parameters:
+    #     connector = configure_connector(parser_args.connector, parser_args.connector_parameters)
+    # else:
+    #     connector = configure_connector(parser_args.connector)
 
-    generate_conversation(parser_args.technology, connector, profile_path,
-                          parser_args.personality, parser_args.extract, parser_args.project_path)
+    connector = parser_args.connector_params
 
+    try:
+        generate_conversation(parser_args.technology, connector, profile_path,
+                          parser_args.personality, parser_args.extract, parser_args.project_path)
+    except Exception as e:
+        logger.error(f"An error occurred while generating the conversation: {e}")
 
 if __name__ == '__main__':
     main()

user_sim/utils/utilities.py

@@ -9,6 +9,7 @@ import importlib.util
 import logging
 import platform
 
+from typing import Any
 from datetime import datetime, timedelta, date
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
@@ -63,6 +64,52 @@ def init_model():
     return model, llm
 
 
+# def parse_connector_params(connector_params_str: str | None) -> dict[str, Any]:
+#     """Parse connector parameters from string input.
+#
+#     Args:
+#         connector_params_str: JSON string or key=value pairs
+#
+#     Returns:
+#         Dictionary of connector parameters
+#
+#     Raises:
+#         ValueError: If parameter parsing fails
+#     """
+#     params = {}
+#
+#     if connector_params_str:
+#         try:
+#             # Try to parse as JSON first
+#             if connector_params_str.strip().startswith("{"):
+#                 params = json.loads(connector_params_str)
+#             else:
+#                 # Parse as key=value pairs
+#                 for pair in connector_params_str.split(","):
+#                     if "=" not in pair:
+#                         continue
+#                     key, value = pair.split("=", 1)
+#                     key = key.strip()
+#                     value = value.strip()
+#
+#                     # Try to convert to appropriate types
+#                     if value.lower() in ("true", "false"):
+#                         params[key] = value.lower() == "true"
+#                     elif value.isdigit():
+#                         params[key] = int(value)
+#                     else:
+#                         try:
+#                             params[key] = float(value)
+#                         except ValueError:
+#                             params[key] = value
+#
+#         except (json.JSONDecodeError, ValueError) as e:
+#             logger.exception("Failed to parse connector parameters: %s", connector_params_str)
+#             msg = f"Invalid connector parameters format: {e}"
+#             raise ValueError(msg) from e
+#
+#     return params
+
 
 def parse_content_to_text(messages):
     return " ".join([message["content"] for message in messages if "content" in message])

{user_simulator-0.1.11.dist-info → user_simulator-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: user-simulator
-Version: 0.1.11
+Version: 0.2.0
 Summary: LLM-based user simulator for chatbot testing.
 Author: Alejandro Del Pozzo Escalera, Juan de Lara Jaramillo, Esther Guerra Sánchez
 License: MIT License
@@ -30,6 +30,8 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: allpairspy>=2.5.1
 Requires-Dist: beautifulsoup4>=4.13.4
+Requires-Dist: build>=1.3.0
+Requires-Dist: chatbot-connectors>=0.2.2
 Requires-Dist: colorama>=0.4.6
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: inflect>=7.0.0
@@ -44,6 +46,7 @@ Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: requests>=2.32.4
 Requires-Dist: scikit-learn>=1.7.0
 Requires-Dist: selenium>=4.33.0
+Requires-Dist: twine>=6.1.0
 Requires-Dist: webdriver-manager>=4.0.2
 Dynamic: license-file
 

{user_simulator-0.1.11.dist-info → user_simulator-0.2.0.dist-info}/RECORD

@@ -12,7 +12,7 @@ user_sim/__init__.py,sha256=hLzeYUaZDKIQet60JxHFB7Bhe3CL23yUmbJQPcfQFGE,479
 user_sim/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 user_sim/cli/gen_user_profile.py,sha256=PSm9wi2OhflLWHM1TNxL5W2xYFR82z2-24NhOQRye2M,1462
 user_sim/cli/init_project.py,sha256=t4z4Omfo2_uh-mNjVOpFIxnnlWdtH9VUa3mD3q14v6A,1947
-user_sim/cli/sensei_chat.py,sha256=YsuZJPc9TJRlAmOP3PV56mlBUPfaZOeH5o_SBV_8fwA,20512
+user_sim/cli/sensei_chat.py,sha256=bmRS7Zrp0qutL50gsvavgnx0nHRuQMpX8ELXVL7dz0c,21132
 user_sim/cli/sensei_check.py,sha256=l_HIAPTw621MxXJ9fwmdE8BzDABt1XdalHVaKYH_QsQ,3900
 user_sim/cli/validation_check.py,sha256=VUJlePbTfBJOgOx2sDamKD4cplvUlS7VKqr-v1rvFPE,5363
 user_sim/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -38,10 +38,10 @@ user_sim/utils/register_management.py,sha256=ZUmnPZLzNZQs17GezD5iM9hxMHNW1rZyeBz
 user_sim/utils/show_logs.py,sha256=pqwksfYYFZEu3pb-4KJrV4OEaiAUrfc5-G1qz8akJmo,1754
 user_sim/utils/token_cost_calculator.py,sha256=4vFUh6OHgoZJcfxaRGARFUbyLkXPYUug6YJCM7luw6s,12540
 user_sim/utils/url_management.py,sha256=MeUoT-O7XCPWJJ-1r3T_n1zFjj3NyeJXRt_b9Nkpzdw,1998
-user_sim/utils/utilities.py,sha256=MBviUg1xCyJeDbIDPIQWWno-3jhTwwpn-z6TB5zM63s,19792
-user_simulator-0.1.11.dist-info/licenses/LICENSE.txt,sha256=Qv2ilebwoUtMJnRsZwRy729xS5JZQzLauJ0tQzkAkTA,1088
-user_simulator-0.1.11.dist-info/METADATA,sha256=uubTAy5xcmHKm4yT3gttZp77Ucw6cDBYI6OrHmII0zE,31799
-user_simulator-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-user_simulator-0.1.11.dist-info/entry_points.txt,sha256=kfr0rD6mglTGGzYRfUwMTGfTtTOBnLjv3R-Uy61vvDQ,283
-user_simulator-0.1.11.dist-info/top_level.txt,sha256=IGlrGt_QHrM2G3Mlo76gMp8ArUR_F9k4RrzJd8YHpE4,34
-user_simulator-0.1.11.dist-info/RECORD,,
+user_sim/utils/utilities.py,sha256=ixv48MMPaIaj12aBO0pIxIgwWie0oEMX2lckPm1_SeU,21568
+user_simulator-0.2.0.dist-info/licenses/LICENSE.txt,sha256=Qv2ilebwoUtMJnRsZwRy729xS5JZQzLauJ0tQzkAkTA,1088
+user_simulator-0.2.0.dist-info/METADATA,sha256=nrOOEB9GsUBCUIr4S0ywV_lzHrzRBq3hI0VIewzNErg,31898
+user_simulator-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+user_simulator-0.2.0.dist-info/entry_points.txt,sha256=kfr0rD6mglTGGzYRfUwMTGfTtTOBnLjv3R-Uy61vvDQ,283
+user_simulator-0.2.0.dist-info/top_level.txt,sha256=IGlrGt_QHrM2G3Mlo76gMp8ArUR_F9k4RrzJd8YHpE4,34
+user_simulator-0.2.0.dist-info/RECORD,,