glchat-plugin 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

glchat_plugin/config/constant.py

@@ -13,14 +13,13 @@ from enum import StrEnum
 class SearchType(StrEnum):
     """The type of search to perform.
 
-    Values:
+    Attributes:
         NORMAL: Get answer from chatbot knowledge.
-        SMART: Get more relevant information from your stored documents and knowledge base.
-            Knowledge Search is an AI with specialized knowledge. No agents are available in this mode.
         WEB: Get more relevant information from the web.
             Web Search uses real-time data. Agent selection isn't available in this mode.
+        DEEP_RESEARCH: Get answer from Deep Research Agent.
     """
 
     NORMAL = "normal"
-    SMART = "smart"
     WEB = "web"
+    DEEP_RESEARCH = "deep_research"

glchat_plugin/context/__init__.py

@@ -0,0 +1 @@
+"""Module for glchat_plugin context."""

glchat_plugin/context/context_manager.py

@@ -0,0 +1,57 @@
+"""Provides a class for context manager.
+
+Authors:
+    Ryan Ignatius Hadiwijaya (ryan.i.hadiwijaya@gdplabs.id)
+
+References:
+    NONE
+"""
+
+from contextvars import ContextVar
+
+
+class ContextManager:
+    """A Context Manager.
+
+    This class is used to manage the context of the application.
+    """
+
+    _tenant: ContextVar[str | None] = ContextVar("tenant_id", default=None)
+    _user: ContextVar[str | None] = ContextVar("user_id", default=None)
+
+    @classmethod
+    def set_tenant(cls, tenant_id: str) -> None:
+        """Set the tenant id in the context.
+
+        Args:
+            tenant_id (str): The tenant id.
+        """
+        cls._tenant.set(tenant_id)
+
+    @classmethod
+    def get_tenant(cls) -> str | None:
+        """Get the tenant id from the context.
+
+        Returns:
+            str | None: The tenant id.
+        """
+        tenant_id = cls._tenant.get()
+        return tenant_id
+
+    @classmethod
+    def set_user(cls, user_id: str) -> None:
+        """Set the user id in the context.
+
+        Args:
+            user_id (str): The user id.
+        """
+        cls._user.set(user_id)
+
+    @classmethod
+    def get_user(cls) -> str | None:
+        """Get the user id from the context.
+
+        Returns:
+            str | None: The user id.
+        """
+        return cls._user.get()

glchat_plugin/service/__init__.py

@@ -1 +1 @@
-"""Module for glchat_plugin user service."""
+"""Module for glchat_plugin custom service."""

glchat_plugin/service/base_tenant_service.py

@@ -0,0 +1,31 @@
+"""Provides a base class for tenant service.
+
+Authors:
+    Ryan Ignatius Hadiwijaya (ryan.i.hadiwijaya@gdplabs.id)
+
+References:
+    NONE
+"""
+
+from abc import ABC, abstractmethod
+from typing import Any
+
+
+class BaseTenantService(ABC):
+    """A base class for tenant service."""
+
+    @abstractmethod
+    async def get_tenant_id(self, **kwargs: Any) -> str:
+        """Abstract method to get the tenant id.
+
+        Args:
+            **kwargs (Any): Additional keyword arguments for tenant identification.
+                The kwargs are Fast API Request object.
+
+        Returns:
+            str: tenant id.
+
+        Raises:
+            NotImplementedError: If the method is not implemented.
+        """
+        raise NotImplementedError

glchat_plugin/service/base_user_service.py

@@ -16,14 +16,14 @@ class BaseUserService(ABC):
     """A base class for user service."""
 
     @abstractmethod
-    async def get_user_id(self, **kwargs: Any) -> str:
+    async def get_user_id(self, **kwargs: Any) -> str | None:
         """Abstract method to get the user id.
 
         Args:
             **kwargs (Any): Additional keyword arguments for user identification.
 
         Returns:
-            str: user id.
+            str | None: user id, None if not found.
 
         Raises:
             NotImplementedError: If the method is not implemented.

{glchat_plugin-0.2.0.dist-info → glchat_plugin-0.2.2.dist-info}/METADATA

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

{glchat_plugin-0.2.0.dist-info → glchat_plugin-0.2.2.dist-info}/RECORD

@@ -1,19 +1,22 @@
 glchat_plugin/__init__.py,sha256=SHSBMz7JDU6MecyIrhHu5-3NVs89JkXhyvD3ZGOkWOE,37
 glchat_plugin/config/__init__.py,sha256=DNnX8B_TvAN89oyMgq32zG1DeaezODrihiAXTwOPT5o,39
 glchat_plugin/config/app_config.py,sha256=9_ShYtaQ7Rp14sSkrIFLoOAMlbwVlm13EuCxzOn4NCI,426
-glchat_plugin/config/constant.py,sha256=hdTXG5mIULV86DepClE97VGhn9E-HtsjZ9PyLKKEdmc,689
+glchat_plugin/config/constant.py,sha256=U3Qg-MmrHUw51D8NcnCRbKfdhphCp_xEGbAjaGbaWHM,573
+glchat_plugin/context/__init__.py,sha256=3Wx_apMIS6z-m6eRs6hoyOsJFLJfKmMFOkrPDkPQfJI,40
+glchat_plugin/context/context_manager.py,sha256=0lhO0w_hd5dUdIEJQ2LOJFZsgpzitQU_aPZfTfQK3vw,1302
 glchat_plugin/pipeline/__init__.py,sha256=Sk-NfIGyA9VKIg0Bt5OHatNUYyWVPh9i5xhE5DFAfbo,41
 glchat_plugin/pipeline/base_pipeline_preset_config.py,sha256=2DxJlroZ_tvVO6DYo1R07cjhWgKQJpysw1GsJ3BNRnw,1100
 glchat_plugin/pipeline/pipeline_handler.py,sha256=vDvIPI7Y39K_DI3jvQZsXC9YzQLUI4TuC-boyasdey8,24405
 glchat_plugin/pipeline/pipeline_plugin.py,sha256=1i7w7WepM5gON4Kx10SjL7NV7-SYVAmRE1EW2m_dOYg,4028
-glchat_plugin/service/__init__.py,sha256=eVyDgTcbOG0slfGT2GSUnUM9MHC2Wh3aU6fcLV_JHs4,45
-glchat_plugin/service/base_user_service.py,sha256=agpziGm7ua2wIv-rrrnTOiYlyUjp6L6dArWY2q0m63g,2178
+glchat_plugin/service/__init__.py,sha256=9T4qzyYL052qLqva5el1F575OTRNaaf9tb9UvW-leTc,47
+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/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=JvUUFMu_9jRBQ9yug_x7S4rQjZEA1vM5ombDvz-7zCE,11095
 glchat_plugin/tools/__init__.py,sha256=OFotHbgQ8mZEbdlvlv5aVMdxfubPvkVWAcTwhIPdIqQ,542
 glchat_plugin/tools/decorators.py,sha256=AvQBV18wzXWdC483RSSmpfh92zsqTyp8SzDLIkreIGU,3925
-glchat_plugin-0.2.0.dist-info/METADATA,sha256=Vhjz_PklkL2HzLFrJBi27NPAox_4ytlLDX137ltMKLg,2063
-glchat_plugin-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-glchat_plugin-0.2.0.dist-info/top_level.txt,sha256=fzKSXmct5dY4CAKku4-mkdHX-QPAyQVvo8vpQj8qizY,14
-glchat_plugin-0.2.0.dist-info/RECORD,,
+glchat_plugin-0.2.2.dist-info/METADATA,sha256=9ayNjHwIrARFRZPvCUEoqDkqJdpf_hYlMFDUXE4bylg,2063
+glchat_plugin-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+glchat_plugin-0.2.2.dist-info/top_level.txt,sha256=fzKSXmct5dY4CAKku4-mkdHX-QPAyQVvo8vpQj8qizY,14
+glchat_plugin-0.2.2.dist-info/RECORD,,