glchat-plugin 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

glchat_plugin/context/context_manager.py

@@ -1,6 +1,7 @@
 """Provides a class for context manager.
 
 Authors:
+    Ricky Setiawan (ricky.setiawan@gdplabs.id)
     Ryan Ignatius Hadiwijaya (ryan.i.hadiwijaya@gdplabs.id)
 
 References:
@@ -8,6 +9,7 @@ References:
 """
 
 from contextvars import ContextVar
+from typing import Any
 
 
 class ContextManager:
@@ -18,6 +20,7 @@ class ContextManager:
 
     _tenant: ContextVar[str | None] = ContextVar("tenant_id", default=None)
     _user: ContextVar[str | None] = ContextVar("user_id", default=None)
+    _user_session: ContextVar[Any | None] = ContextVar("user_session", default=None)
 
     @classmethod
     def set_tenant(cls, tenant_id: str | None) -> None:
@@ -55,3 +58,21 @@ class ContextManager:
             str | None: The user id.
         """
         return cls._user.get()
+
+    @classmethod
+    def set_user_session(cls, user_session: Any | None) -> None:
+        """Set the user session in the context.
+
+        Args:
+            user_session (Any | None): The user session.
+        """
+        cls._user_session.set(user_session)
+
+    @classmethod
+    def get_user_session(cls) -> Any | None:
+        """Get the user session from the context.
+
+        Returns:
+            Any | None: The user session.
+        """
+        return cls._user_session.get()

glchat_plugin/pipeline/pipeline_handler.py

@@ -166,7 +166,7 @@ class PipelineHandler(PluginHandler):
                 if pipeline_type != instance._chatbot_configs[chatbot_organization_id].pipeline_type:
                     continue
 
-                await cls._build_plugin(instance, chatbot_id, preset.supported_models, plugin)
+                await cls._build_plugin(instance, chatbot_id, preset.supported_models, plugin, organization_id)
             except Exception as e:
                 logger.warning(f"Failed when ainit pliugin {traceback.format_exc()}")
                 logger.warning(f"Error initializing plugin for chatbot `{chatbot_id}`: {e}")
@@ -186,7 +186,12 @@ class PipelineHandler(PluginHandler):
 
     @classmethod
     async def _build_plugin(
-        cls, instance: "PipelineHandler", chatbot_id: str, supported_models: list[dict[str, Any]], plugin: Plugin
+        cls,
+        instance: "PipelineHandler",
+        chatbot_id: str,
+        supported_models: list[dict[str, Any]],
+        plugin: Plugin,
+        organization_id: str = DEFAULT_ORGANIZATION_ID,
     ) -> None:
         """Build a plugin for the given chatbot.
 
@@ -195,8 +200,8 @@ class PipelineHandler(PluginHandler):
             chatbot_id (str): The chatbot ID.
             supported_models (list[dict[str, Any]]): List of models (including config).
             plugin (Plugin): The pipeline builder plugin instance.
+            organization_id (str): The organization ID.
         """
-        organization_id = plugin.organization_id
         chatbot_organization_id = (chatbot_id, organization_id)
         plugin.prompt_builder_catalogs = instance._chatbot_configs[chatbot_organization_id].prompt_builder_catalogs
         plugin.lmrp_catalogs = instance._chatbot_configs[chatbot_organization_id].lmrp_catalogs
@@ -376,7 +381,7 @@ class PipelineHandler(PluginHandler):
                 return
 
             # Use the existing _build_plugin method to rebuild the pipeline
-            await __class__._build_plugin(self, chatbot_id, [model_config], plugin)
+            await __class__._build_plugin(self, chatbot_id, [model_config], plugin, organization_id)
 
             logger.info(f"Successfully rebuilt pipeline for chatbot `{chatbot_organization_id}` model `{model_id}`")
 
@@ -519,7 +524,7 @@ class PipelineHandler(PluginHandler):
         )
 
         supported_models = list(pipeline_info["config"].get("supported_models", {}).values())
-        await __class__._build_plugin(self, chatbot_id, supported_models, plugin)
+        await __class__._build_plugin(self, chatbot_id, supported_models, plugin, organization_id)
 
     async def delete_chatbot(self, chatbot_id: str, organization_id: str = DEFAULT_ORGANIZATION_ID) -> None:
         """Delete a chatbot.
@@ -594,7 +599,7 @@ class PipelineHandler(PluginHandler):
                 self._builders.pop(chatbot_organization_id, None)
                 self._builders[chatbot_organization_id] = plugin
 
-                await __class__._build_plugin(self, chatbot_id, supported_models, plugin)
+                await __class__._build_plugin(self, chatbot_id, supported_models, plugin, organization_id)
             except Exception as e:
                 logger.warning(f"Error updating chatbot `{chatbot_id}`: {e}")
 
@@ -746,7 +751,7 @@ class PipelineHandler(PluginHandler):
                 return
 
             # Use the existing _build_plugin method to rebuild the plugin
-            await __class__._build_plugin(self, chatbot_id, supported_models, plugin)
+            await __class__._build_plugin(self, chatbot_id, supported_models, plugin, organization_id)
 
             logger.info(f"Successfully rebuilt pipeline builder for chatbot `{chatbot_id}`")
 

glchat_plugin/service/base_user_service.py

@@ -3,6 +3,7 @@
 Authors:
     Immanuel Rhesa (immanuel.rhesa@gdplabs.id)
     Hermes Vincentius Gani (hermes.v.gani@gdplabs.id)
+    Ricky Setiawan (ricky.setiawan@gdplabs.id)
 
 References:
     NONE
@@ -30,6 +31,21 @@ class BaseUserService(ABC):
         """
         raise NotImplementedError
 
+    @abstractmethod
+    async def get_user_session(self, **kwargs: Any) -> Any | None:
+        """Abstract method to get the user session.
+
+        Args:
+            **kwargs (Any): Additional keyword arguments for user session retrieval.
+
+        Returns:
+            Any | None: User session, None if not found.
+
+        Raises:
+            NotImplementedError: If the method is not implemented.
+        """
+        raise NotImplementedError
+
     @abstractmethod
     async def register(self, **kwargs: Any):
         """Abstract method to register a new user.

{glchat_plugin-0.4.1.dist-info → glchat_plugin-0.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: glchat-plugin
-Version: 0.4.1
+Version: 0.4.3
 Author-email: GenAI SDK Team <gat-sdk@gdplabs.id>
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown

{glchat_plugin-0.4.1.dist-info → glchat_plugin-0.4.3.dist-info}/RECORD

@@ -3,23 +3,23 @@ glchat_plugin/config/__init__.py,sha256=DNnX8B_TvAN89oyMgq32zG1DeaezODrihiAXTwOP
 glchat_plugin/config/app_config.py,sha256=9_ShYtaQ7Rp14sSkrIFLoOAMlbwVlm13EuCxzOn4NCI,426
 glchat_plugin/config/constant.py,sha256=zZO4WCkozvm6i_rMquTp99Q3alom0SN_P9SNqAHO3ko,3621
 glchat_plugin/context/__init__.py,sha256=3Wx_apMIS6z-m6eRs6hoyOsJFLJfKmMFOkrPDkPQfJI,40
-glchat_plugin/context/context_manager.py,sha256=AVTs30Tg5EbPb5HDZfW9RI3lY24Pc0-NhOPFxw5XG5U,1330
+glchat_plugin/context/context_manager.py,sha256=dQUbejXUMrYwLo8xPGo1P-QWuMYPahmRddAx8Ig-fJE,1968
 glchat_plugin/handler/__init__.py,sha256=H5DJaAfwwtRsvMcOaEzHfGMQk25H7la0E7uPfksWtoQ,40
 glchat_plugin/handler/base_post_login_handler.py,sha256=48xSbe_LwTCjRY-lCuzWXqbnEr1ql8bAhQih1Xeh8f8,2835
 glchat_plugin/pipeline/__init__.py,sha256=Sk-NfIGyA9VKIg0Bt5OHatNUYyWVPh9i5xhE5DFAfbo,41
 glchat_plugin/pipeline/base_pipeline_preset_config.py,sha256=lbMH8y_HU3LrqtMYXLzQ2906ZkMXARKY5vBuIGvRVdA,2969
-glchat_plugin/pipeline/pipeline_handler.py,sha256=1XkTfhmoIViJKVhRbaKisCRSWR5lzQO6aCP6O8-qqOo,34022
+glchat_plugin/pipeline/pipeline_handler.py,sha256=Kp2LBqtVHzJ2twLaUvYXnL-1QGkqL_8h21IAR7FxBTI,34203
 glchat_plugin/pipeline/pipeline_plugin.py,sha256=eBrg4THS0P49Mz4glEjrwqe1pxnvOn_UitJ9gOBVcac,4562
 glchat_plugin/service/__init__.py,sha256=9T4qzyYL052qLqva5el1F575OTRNaaf9tb9UvW-leTc,47
 glchat_plugin/service/base_rate_limiter_service.py,sha256=tgKwdr4EqnGo5iDRVJPnlg8W9q0hiUzfeewAtdW4IjU,1232
 glchat_plugin/service/base_tenant_service.py,sha256=CB-jJWXi87nTcjO1XhPoSgNMf2YA_LfXoHr30ye0VtI,734
-glchat_plugin/service/base_user_service.py,sha256=qnRKl2u1xhGc4wMaOd_h_mAuxM7GduyDbqebj-k3GDE,2211
+glchat_plugin/service/base_user_service.py,sha256=DlVj0pimzTgvgz2dqpVkfhMK8PWEwQ9LyukzGoVZgYc,2703
 glchat_plugin/storage/__init__.py,sha256=VNO7ua2yAOLzXwe6H6zDZz9yXbhQfKqNJzxE3G8IyfA,50
 glchat_plugin/storage/base_anonymizer_storage.py,sha256=oFwovWrsjM7v1YjeN-4p-M3OGnAeo_Y7-9xqlToI180,1969
 glchat_plugin/storage/base_chat_history_storage.py,sha256=YIGM8zv7s5BQ8O7W1LfaLyQW4SF9Bt3aolowsoDaQw8,11257
 glchat_plugin/tools/__init__.py,sha256=OFotHbgQ8mZEbdlvlv5aVMdxfubPvkVWAcTwhIPdIqQ,542
 glchat_plugin/tools/decorators.py,sha256=AvQBV18wzXWdC483RSSmpfh92zsqTyp8SzDLIkreIGU,3925
-glchat_plugin-0.4.1.dist-info/METADATA,sha256=q3Po_AA7fAbEOEMH-EiCgX5VxEOH774fbfnQ_qD8tyc,2063
-glchat_plugin-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-glchat_plugin-0.4.1.dist-info/top_level.txt,sha256=fzKSXmct5dY4CAKku4-mkdHX-QPAyQVvo8vpQj8qizY,14
-glchat_plugin-0.4.1.dist-info/RECORD,,
+glchat_plugin-0.4.3.dist-info/METADATA,sha256=Annr1O57pSyLVG6niQiTXTZT7A2s9R1QWlo5nqzgur4,2063
+glchat_plugin-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+glchat_plugin-0.4.3.dist-info/top_level.txt,sha256=fzKSXmct5dY4CAKku4-mkdHX-QPAyQVvo8vpQj8qizY,14
+glchat_plugin-0.4.3.dist-info/RECORD,,