symbolicai 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

symai/__init__.py

@@ -23,19 +23,20 @@ logging.getLogger("httpx").setLevel(logging.ERROR)
 logging.getLogger("httpcore").setLevel(logging.ERROR)
 logging.getLogger("huggingface_hub").setLevel(logging.ERROR)
 logging.getLogger("huggingface").setLevel(logging.ERROR)
-logging.getLogger('pydub').setLevel(logging.ERROR)
+logging.getLogger("pydub").setLevel(logging.ERROR)
 
 warnings.simplefilter("ignore")
 
 # set the environment variable for the transformers library
-os.environ['TOKENIZERS_PARALLELISM'] = "false"
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 # Create singleton instance
 config_manager = settings.SymAIConfig()
 
-SYMAI_VERSION = "1.0.0"
-__version__   = SYMAI_VERSION
-__root_dir__  = config_manager.config_dir
+SYMAI_VERSION = "1.1.1"
+__version__ = SYMAI_VERSION
+__root_dir__ = config_manager.config_dir
+
 
 def _start_symai():
     # Create config directories if they don't exist
@@ -44,97 +45,108 @@ def _start_symai():
 
     # CREATE THE SHELL CONFIGURATION FILE IF IT DOES NOT EXIST YET
     # *==============================================================================================================*
-    _symsh_config_path_ = config_manager.get_config_path('symsh.config.json')
+    _symsh_config_path_ = config_manager.get_config_path("symsh.config.json")
     if not _symsh_config_path_.exists():
-        config_manager.save_config('symsh.config.json', {
-            "colors": {
-                "completion-menu.completion.current": "bg:#323232 #212121",
-                "completion-menu.completion":         "bg:#800080 #212121",
-                "scrollbar.background":               "bg:#222222",
-                "scrollbar.button":                   "bg:#776677",
-                "history-completion":                 "bg:#212121 #f5f5f5",
-                "path-completion":                    "bg:#800080 #f5f5f5",
-                "file-completion":                    "bg:#9040b2 #f5f5f5",
-                "history-completion-selected":        "bg:#efefef #b3d7ff",
-                "path-completion-selected":           "bg:#efefef #b3d7ff",
-                "file-completion-selected":           "bg:#efefef #b3d7ff"
+        config_manager.save_config(
+            "symsh.config.json",
+            {
+                "colors": {
+                    "completion-menu.completion.current": "bg:#323232 #212121",
+                    "completion-menu.completion": "bg:#800080 #212121",
+                    "scrollbar.background": "bg:#222222",
+                    "scrollbar.button": "bg:#776677",
+                    "history-completion": "bg:#212121 #f5f5f5",
+                    "path-completion": "bg:#800080 #f5f5f5",
+                    "file-completion": "bg:#9040b2 #f5f5f5",
+                    "history-completion-selected": "bg:#efefef #b3d7ff",
+                    "path-completion-selected": "bg:#efefef #b3d7ff",
+                    "file-completion-selected": "bg:#efefef #b3d7ff",
+                },
+                "map-nt-cmd": True,
+                "show-splash-screen": True,
+                "plugin_prefix": None,
             },
-            "map-nt-cmd":                             True,
-            "show-splash-screen":                     True,
-            "plugin_prefix":                          None
-        })
+        )
 
     # CREATE A SERVER CONFIGURATION FILE IF IT DOES NOT EXIST YET
     # *==============================================================================================================*
-    _symserver_config_path_ = config_manager.get_config_path('symserver.config.json')
+    _symserver_config_path_ = config_manager.get_config_path("symserver.config.json")
     if not _symserver_config_path_.exists():
-        config_manager.save_config('symserver.config.json', {})
+        config_manager.save_config("symserver.config.json", {})
 
     # Get appropriate config path (debug mode handling is now in config_manager)
-    _symai_config_path_ = config_manager.get_config_path('symai.config.json')
+    _symai_config_path_ = config_manager.get_config_path("symai.config.json")
 
     if not _symai_config_path_.exists():
         setup_wizard(_symai_config_path_)
-        UserMessage(f'No configuration file found for the environment. A new configuration file has been created at {_symai_config_path_}. Please configure your environment.')
+        UserMessage(
+            f"No configuration file found for the environment. A new configuration file has been created at {_symai_config_path_}. Please configure your environment."
+        )
         sys.exit(1)
 
     # Load and manage configurations
-    symai_config = config_manager.load_config('symai.config.json')
+    symai_config = config_manager.load_config("symai.config.json")
 
     # MIGRATE THE ENVIRONMENT VARIABLES
     # *==========================================================================================================*
-    if 'COLLECTION_URI' not in symai_config:
+    if "COLLECTION_URI" not in symai_config:
         updates = {
-            'COLLECTION_URI': "mongodb+srv://User:vt3epocXitd6WlQ6@extensityai.c1ajxxy.mongodb.net/?retryWrites=true&w=majority",
-            'COLLECTION_DB': "ExtensityAI",
-            'COLLECTION_STORAGE': "SymbolicAI",
-            'SUPPORT_COMMUNITY': False
+            "COLLECTION_URI": "mongodb+srv://User:vt3epocXitd6WlQ6@extensityai.c1ajxxy.mongodb.net/?retryWrites=true&w=majority",
+            "COLLECTION_DB": "ExtensityAI",
+            "COLLECTION_STORAGE": "SymbolicAI",
+            "SUPPORT_COMMUNITY": False,
         }
-        config_manager.migrate_config('symai.config.json', updates)
-        with ConsoleStyle('info') as console:
-            msg = 'Currently you are sharing your user experience with us by uploading the data to our research server, and thereby helping us improve future models and the overall SymbolicAI experience. We thank you very much for supporting the research community! If you wish to disable the data collection option go to your .symai config situated in your home directory or set the environment variable `SUPPORT_COMMUNITY` to `False`.'
+        config_manager.migrate_config("symai.config.json", updates)
+        with ConsoleStyle("info") as console:
+            msg = "Currently you are sharing your user experience with us by uploading the data to our research server, and thereby helping us improve future models and the overall SymbolicAI experience. We thank you very much for supporting the research community! If you wish to disable the data collection option go to your .symai config situated in your home directory or set the environment variable `SUPPORT_COMMUNITY` to `False`."
             console.print(msg)
 
     # POST-MIGRATION CHECKS
     # *==============================================================================================================*
-    if 'TEXT_TO_SPEECH_ENGINE_API_KEY' not in symai_config:
+    if "TEXT_TO_SPEECH_ENGINE_API_KEY" not in symai_config:
         updates = {
-            'TEXT_TO_SPEECH_ENGINE_API_KEY': symai_config.get('NEUROSYMBOLIC_ENGINE_API_KEY', '')
+            "TEXT_TO_SPEECH_ENGINE_API_KEY": symai_config.get("NEUROSYMBOLIC_ENGINE_API_KEY", "")
         }
-        config_manager.migrate_config('symai.config.json', updates)
+        config_manager.migrate_config("symai.config.json", updates)
 
     # Load all configurations
-    symai_config = config_manager.load_config('symai.config.json')
-    symsh_config = config_manager.load_config('symsh.config.json')
-    symserver_config = config_manager.load_config('symserver.config.json')
+    symai_config = config_manager.load_config("symai.config.json")
+    symsh_config = config_manager.load_config("symsh.config.json")
+    symserver_config = config_manager.load_config("symserver.config.json")
 
     # MIGRATE THE SHELL SPLASH SCREEN CONFIGURATION
     # *==============================================================================================================*
-    if 'show-splash-screen' not in symsh_config:
-        config_manager.migrate_config('symsh.config.json', {'show-splash-screen': True})
+    if "show-splash-screen" not in symsh_config:
+        config_manager.migrate_config("symsh.config.json", {"show-splash-screen": True})
 
     # CHECK IF THE USER HAS A NEUROSYMBOLIC API KEY
     # *==============================================================================================================*
     if not (
-        symai_config['NEUROSYMBOLIC_ENGINE_MODEL'].lower().startswith('llama') or \
-        symai_config['NEUROSYMBOLIC_ENGINE_MODEL'].lower().startswith('huggingface')) \
-        and \
-            (
-            symai_config['NEUROSYMBOLIC_ENGINE_API_KEY'] is None or \
-            len(symai_config['NEUROSYMBOLIC_ENGINE_API_KEY']) == 0):
-            # Try to fallback to the global (home) config if environment is not home
-            if config_manager.config_dir != config_manager._home_config_dir:
-                show_intro_menu()
-                UserMessage(f"You didn't configure your environment ({config_manager.config_dir})! Falling back to the global ({config_manager._home_config_dir}) configuration if it exists.")
-                # Force loading from home
-                symai_config = config_manager.load_config('symai.config.json', fallback_to_home=True)
-                symsh_config = config_manager.load_config('symsh.config.json', fallback_to_home=True)
-                symserver_config = config_manager.load_config('symserver.config.json', fallback_to_home=True)
-
-            # If still not valid, warn and exit
-            if not symai_config.get('NEUROSYMBOLIC_ENGINE_API_KEY'):
-                UserMessage('The mandatory neuro-symbolic engine is not initialized. Please set NEUROSYMBOLIC_ENGINE_MODEL and NEUROSYMBOLIC_ENGINE_API_KEY.')
-                sys.exit(1)
+        symai_config["NEUROSYMBOLIC_ENGINE_MODEL"].lower().startswith("llama")
+        or symai_config["NEUROSYMBOLIC_ENGINE_MODEL"].lower().startswith("huggingface")
+    ) and (
+        symai_config["NEUROSYMBOLIC_ENGINE_API_KEY"] is None
+        or len(symai_config["NEUROSYMBOLIC_ENGINE_API_KEY"]) == 0
+    ):
+        # Try to fallback to the global (home) config if environment is not home
+        if config_manager.config_dir != config_manager._home_config_dir:
+            show_intro_menu()
+            UserMessage(
+                f"You didn't configure your environment ({config_manager.config_dir})! Falling back to the global ({config_manager._home_config_dir}) configuration if it exists."
+            )
+            # Force loading from home
+            symai_config = config_manager.load_config("symai.config.json", fallback_to_home=True)
+            symsh_config = config_manager.load_config("symsh.config.json", fallback_to_home=True)
+            symserver_config = config_manager.load_config(
+                "symserver.config.json", fallback_to_home=True
+            )
+
+        # If still not valid, warn and exit
+        if not symai_config.get("NEUROSYMBOLIC_ENGINE_API_KEY"):
+            UserMessage(
+                "The mandatory neuro-symbolic engine is not initialized. Please set NEUROSYMBOLIC_ENGINE_MODEL and NEUROSYMBOLIC_ENGINE_API_KEY."
+            )
+            sys.exit(1)
 
     settings.SYMAI_CONFIG = symai_config
     settings.SYMSH_CONFIG = symsh_config
@@ -144,17 +156,58 @@ def _start_symai():
 
 def run_server():
     _symserver_config_ = {}
-    if settings.SYMAI_CONFIG.get("NEUROSYMBOLIC_ENGINE_MODEL").startswith("llama") or settings.SYMAI_CONFIG.get("EMBEDDING_ENGINE_MODEL").startswith("llama"):
+
+    # Check for explicit Qdrant server request via command line
+    qdrant_requested = any("qdrant" in arg.lower() for arg in sys.argv[1:])
+
+    if (
+        qdrant_requested
+        or settings.SYMAI_CONFIG.get("INDEXING_ENGINE") == "qdrant"
+        or any(
+            "qdrant" in str(v).lower()
+            for v in [
+                settings.SYMAI_CONFIG.get("INDEXING_ENGINE_API_KEY", ""),
+                settings.SYMAI_CONFIG.get("INDEXING_ENGINE_URL", ""),
+            ]
+        )
+    ):
+        from .server.qdrant_server import qdrant_server  # noqa
+
+        command, args = qdrant_server()
+        _symserver_config_.update(zip(args[::2], args[1::2], strict=False))
+        _symserver_config_["online"] = True
+        _symserver_config_["url"] = (
+            f"http://{_symserver_config_.get('--host', 'localhost')}:{_symserver_config_.get('--port', 6333)}"
+        )
+
+        config_manager.save_config("symserver.config.json", _symserver_config_)
+        config_manager.save_config(
+            "symserver.config.json", _symserver_config_, fallback_to_home=True
+        )
+
+        try:
+            subprocess.run(command, check=True)
+        except KeyboardInterrupt:
+            UserMessage("Server stopped!")
+        except Exception as e:
+            UserMessage(f"Error running server: {e}")
+        finally:
+            config_manager.save_config("symserver.config.json", {"online": False})
+    elif settings.SYMAI_CONFIG.get("NEUROSYMBOLIC_ENGINE_MODEL").startswith(
+        "llama"
+    ) or settings.SYMAI_CONFIG.get("EMBEDDING_ENGINE_MODEL").startswith("llama"):
         # Keep optional llama_cpp dependencies lazy.
-        from .server.llama_cpp_server import llama_cpp_server # noqa
+        from .server.llama_cpp_server import llama_cpp_server  # noqa
 
         command, args = llama_cpp_server()
         _symserver_config_.update(zip(args[::2], args[1::2], strict=False))
-        _symserver_config_['online'] = True
+        _symserver_config_["online"] = True
 
         config_manager.save_config("symserver.config.json", _symserver_config_)
-        #@NOTE: Save in both places since you can start the server from anywhere and still not have a nesy engine configured
-        config_manager.save_config("symserver.config.json", _symserver_config_, fallback_to_home=True)
+        # @NOTE: Save in both places since you can start the server from anywhere and still not have a nesy engine configured
+        config_manager.save_config(
+            "symserver.config.json", _symserver_config_, fallback_to_home=True
+        )
 
         try:
             subprocess.run(command, check=True)
@@ -163,15 +216,15 @@ def run_server():
         except Exception as e:
             UserMessage(f"Error running server: {e}")
         finally:
-            config_manager.save_config("symserver.config.json", {'online': False})
+            config_manager.save_config("symserver.config.json", {"online": False})
 
     elif settings.SYMAI_CONFIG.get("NEUROSYMBOLIC_ENGINE_MODEL").startswith("huggingface"):
         # HuggingFace server stack is optional; import only when requested.
-        from .server.huggingface_server import huggingface_server # noqa
+        from .server.huggingface_server import huggingface_server  # noqa
 
         command, args = huggingface_server()
         _symserver_config_.update(vars(args))
-        _symserver_config_['online'] = True
+        _symserver_config_["online"] = True
 
         config_manager.save_config("symserver.config.json", _symserver_config_)
 
@@ -182,11 +235,11 @@ def run_server():
         except Exception as e:
             UserMessage(f"Error running server: {e}")
         finally:
-            config_manager.save_config("symserver.config.json", {'online': False})
+            config_manager.save_config("symserver.config.json", {"online": False})
     else:
         msg = (
             "You're trying to run a local server without a valid neuro-symbolic engine model. "
-            "Please set a valid model in your configuration file. Current available options are 'llamacpp' and 'huggingface'."
+            "Please set a valid model in your configuration file. Current available options are 'llamacpp', 'huggingface' and 'qdrant'."
         )
         UserMessage(msg, raise_with=ValueError)
 
@@ -196,30 +249,33 @@ def format_config_content(config: dict) -> str:
     """Format config content for display, truncating API keys."""
     formatted = {}
     for k, v in config.items():
-        if isinstance(v, str) and ('KEY' in k or 'URI' in k) and v:
+        if isinstance(v, str) and ("KEY" in k or "URI" in k) and v:
             # Show first/last 4 chars of keys/URIs
             formatted[k] = f"{v[:4]}...{v[-4:]}" if len(v) > 8 else v
         else:
             formatted[k] = v
     return json.dumps(formatted, indent=2)
 
+
 def display_config():
     """Display all configuration paths and their content."""
 
     console = Console()
 
     # Create header
-    console.print(Panel.fit(
-        f"[bold cyan]SymbolicAI Configuration Inspector v{__version__}[/bold cyan]",
-        border_style="cyan"
-    ))
+    console.print(
+        Panel.fit(
+            f"[bold cyan]SymbolicAI Configuration Inspector v{__version__}[/bold cyan]",
+            border_style="cyan",
+        )
+    )
 
     # Create main tree
     tree = Tree("[bold]Configuration Locations[/bold]")
 
     # Debug config
     debug_branch = tree.add("[yellow]Debug Mode Config (CWD)[/yellow]")
-    debug_config = config_manager._debug_dir / 'symai.config.json'
+    debug_config = config_manager._debug_dir / "symai.config.json"
     if debug_config.exists():
         with debug_config.open() as f:
             content = json.load(f)
@@ -230,9 +286,9 @@ def display_config():
     # Environment config
     env_branch = tree.add("[yellow]Environment Config[/yellow]")
     env_configs = {
-        'symai.config.json': '⚙️',
-        'symsh.config.json': '🖥️',
-        'symserver.config.json': '🌐'
+        "symai.config.json": "⚙️",
+        "symsh.config.json": "🖥️",
+        "symserver.config.json": "🌐",
     }
 
     for config_file, icon in env_configs.items():
@@ -251,7 +307,9 @@ def display_config():
         if config_path.exists():
             with config_path.open() as f:
                 content = json.load(f)
-            home_branch.add(f"{icon} [green]{config_path}[/green]\n{format_config_content(content)}")
+            home_branch.add(
+                f"{icon} [green]{config_path}[/green]\n{format_config_content(content)}"
+            )
         else:
             home_branch.add(f"[dim]{icon} {config_file} (not found)[/dim]")
 
@@ -262,9 +320,9 @@ def display_config():
 
     active_paths = {
         "Primary Config Dir": config_manager.get_active_config_dir(),
-        "symai.config.json": config_manager.get_active_path('symai.config.json'),
-        "symsh.config.json": config_manager.get_active_path('symsh.config.json'),
-        "symserver.config.json": config_manager.get_active_path('symserver.config.json')
+        "symai.config.json": config_manager.get_active_path("symai.config.json"),
+        "symsh.config.json": config_manager.get_active_path("symsh.config.json"),
+        "symserver.config.json": config_manager.get_active_path("symserver.config.json"),
     }
 
     for config_type, path in active_paths.items():
@@ -281,64 +339,70 @@ def display_config():
     console.print("🖥️  symsh.config.json (Shell configuration)")
     console.print("🌐  symserver.config.json (Server configuration)")
     console.print("\n[dim]Note: API keys and URIs are truncated for security[/dim]")
+
+
 # *==============================================================================================================*
 
 
 def setup_wizard(_symai_config_path_):
     show_intro_menu()
 
-    _nesy_engine_api_key            = ''
-    _nesy_engine_model              = ''
-    _symbolic_engine_api_key        = ''
-    _symbolic_engine_model          = ''
-    _embedding_engine_api_key       = ''
-    _embedding_model                = ''
-    _drawing_engine_api_key         = ''
-    _drawing_engine_model           = ''
-    _vision_engine_model            = ''
-    _search_engine_api_key          = ''
-    _search_engine_model            = ''
-    _ocr_engine_api_key             = ''
-    _speech_to_text_engine_model    = ''
-    _speech_to_text_api_key         = ''
-    _text_to_speech_engine_api_key  = ''
-    _text_to_speech_engine_model    = ''
-    _text_to_speech_engine_voice    = ''
-    _indexing_engine_api_key        = ''
-    _indexing_engine_environment    = ''
-    _caption_engine_environment     = ''
-    _support_comminity              = False
-
-    config_manager.save_config(_symai_config_path_, {
-        "NEUROSYMBOLIC_ENGINE_API_KEY":   _nesy_engine_api_key,
-        "NEUROSYMBOLIC_ENGINE_MODEL":     _nesy_engine_model,
-        "SYMBOLIC_ENGINE_API_KEY":        _symbolic_engine_api_key,
-        "SYMBOLIC_ENGINE":                _symbolic_engine_model,
-        "EMBEDDING_ENGINE_API_KEY":       _embedding_engine_api_key,
-        "EMBEDDING_ENGINE_MODEL":         _embedding_model,
-        "DRAWING_ENGINE_API_KEY":         _drawing_engine_api_key,
-        "DRAWING_ENGINE_MODEL":           _drawing_engine_model,
-        "VISION_ENGINE_MODEL":            _vision_engine_model,
-        "SEARCH_ENGINE_API_KEY":          _search_engine_api_key,
-        "SEARCH_ENGINE_MODEL":            _search_engine_model,
-        "OCR_ENGINE_API_KEY":             _ocr_engine_api_key,
-        "SPEECH_TO_TEXT_ENGINE_MODEL":    _speech_to_text_engine_model,
-        "SPEECH_TO_TEXT_API_KEY":         _speech_to_text_api_key,
-        "TEXT_TO_SPEECH_ENGINE_API_KEY":  _text_to_speech_engine_api_key,
-        "TEXT_TO_SPEECH_ENGINE_MODEL":    _text_to_speech_engine_model,
-        "TEXT_TO_SPEECH_ENGINE_VOICE":    _text_to_speech_engine_voice,
-        "INDEXING_ENGINE_API_KEY":        _indexing_engine_api_key,
-        "INDEXING_ENGINE_ENVIRONMENT":    _indexing_engine_environment,
-        "CAPTION_ENGINE_MODEL":           _caption_engine_environment,
-        "COLLECTION_URI":                 "mongodb+srv://User:vt3epocXitd6WlQ6@extensityai.c1ajxxy.mongodb.net/?retryWrites=true&w=majority",
-        "COLLECTION_DB":                  "ExtensityAI",
-        "COLLECTION_STORAGE":             "SymbolicAI",
-        "SUPPORT_COMMUNITY":              _support_comminity
-    })
+    _nesy_engine_api_key = ""
+    _nesy_engine_model = ""
+    _symbolic_engine_api_key = ""
+    _symbolic_engine_model = ""
+    _embedding_engine_api_key = ""
+    _embedding_model = ""
+    _drawing_engine_api_key = ""
+    _drawing_engine_model = ""
+    _vision_engine_model = ""
+    _search_engine_api_key = ""
+    _search_engine_model = ""
+    _ocr_engine_api_key = ""
+    _speech_to_text_engine_model = ""
+    _speech_to_text_api_key = ""
+    _text_to_speech_engine_api_key = ""
+    _text_to_speech_engine_model = ""
+    _text_to_speech_engine_voice = ""
+    _indexing_engine_api_key = ""
+    _indexing_engine_environment = ""
+    _caption_engine_environment = ""
+    _support_comminity = False
+
+    config_manager.save_config(
+        _symai_config_path_,
+        {
+            "NEUROSYMBOLIC_ENGINE_API_KEY": _nesy_engine_api_key,
+            "NEUROSYMBOLIC_ENGINE_MODEL": _nesy_engine_model,
+            "SYMBOLIC_ENGINE_API_KEY": _symbolic_engine_api_key,
+            "SYMBOLIC_ENGINE": _symbolic_engine_model,
+            "EMBEDDING_ENGINE_API_KEY": _embedding_engine_api_key,
+            "EMBEDDING_ENGINE_MODEL": _embedding_model,
+            "DRAWING_ENGINE_API_KEY": _drawing_engine_api_key,
+            "DRAWING_ENGINE_MODEL": _drawing_engine_model,
+            "VISION_ENGINE_MODEL": _vision_engine_model,
+            "SEARCH_ENGINE_API_KEY": _search_engine_api_key,
+            "SEARCH_ENGINE_MODEL": _search_engine_model,
+            "OCR_ENGINE_API_KEY": _ocr_engine_api_key,
+            "SPEECH_TO_TEXT_ENGINE_MODEL": _speech_to_text_engine_model,
+            "SPEECH_TO_TEXT_API_KEY": _speech_to_text_api_key,
+            "TEXT_TO_SPEECH_ENGINE_API_KEY": _text_to_speech_engine_api_key,
+            "TEXT_TO_SPEECH_ENGINE_MODEL": _text_to_speech_engine_model,
+            "TEXT_TO_SPEECH_ENGINE_VOICE": _text_to_speech_engine_voice,
+            "INDEXING_ENGINE_API_KEY": _indexing_engine_api_key,
+            "INDEXING_ENGINE_ENVIRONMENT": _indexing_engine_environment,
+            "CAPTION_ENGINE_MODEL": _caption_engine_environment,
+            "COLLECTION_URI": "mongodb+srv://User:vt3epocXitd6WlQ6@extensityai.c1ajxxy.mongodb.net/?retryWrites=true&w=majority",
+            "COLLECTION_DB": "ExtensityAI",
+            "COLLECTION_STORAGE": "SymbolicAI",
+            "SUPPORT_COMMUNITY": _support_comminity,
+        },
+    )
+
 
 _symai_config_, _symsh_config_, _symserver_config_ = _start_symai()
 
-from .backend.base import Engine # noqa
+from .backend.base import Engine  # noqa
 from .components import Function, PrimitiveDisabler  # noqa
 from .core import few_shot, zero_shot  # noqa
 from .extended import Conversation  # noqa