glchat-plugin 0.0.0__py3-none-any.whl

glchat_plugin/__init__.py

@@ -0,0 +1 @@
+"""Main module for glchat_plugin."""

glchat_plugin/config/__init__.py

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

glchat_plugin/config/app_config.py

@@ -0,0 +1,22 @@
+"""Configuration schema for the GLLM backend.
+
+Authors:
+    Anggara Setiawan (anggara.t.setiawan@gdplabs.id)
+    Hermes Vincentius Gani (hermes.v.gani@gdplabs.id)
+
+References:
+    None
+"""
+
+from typing import TypeVar
+
+from pydantic import BaseModel
+
+Chatbot = TypeVar("Chatbot")
+
+
+class AppConfig(BaseModel):
+    """Application configuration model."""
+
+    chatbots: dict[str, Chatbot]
+    user_chatbots: dict[str, list[str]]

glchat_plugin/config/constant.py

@@ -0,0 +1,26 @@
+"""Constants for the API.
+
+Authors:
+    Ryan Ignatius Hadiwijaya (ryan.i.hadiwijaya@gdplabs.id)
+
+References:
+    None
+"""
+
+from enum import StrEnum
+
+
+class SearchType(StrEnum):
+    """The type of search to perform.
+
+    Values:
+        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.
+    """
+
+    NORMAL = "normal"
+    SMART = "smart"
+    WEB = "web"

glchat_plugin/pipeline/__init__.py

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

glchat_plugin/pipeline/base_pipeline_preset_config.py

@@ -0,0 +1,36 @@
+"""Base Pipeline Preset Config.
+
+Authors:
+    Ryan Ignatius Hadiwijaya (ryan.i.hadiwijaya@gdplabs.id)
+
+References:
+    NONE
+"""
+
+from typing import Any
+
+from pydantic import BaseModel
+
+from glchat_plugin.config.constant import SearchType
+
+
+class BasePipelinePresetConfig(BaseModel):
+    """A Pydantic model representing the base preset configuration of all pipelines.
+
+    Attributes:
+        pipeline_preset_id (str): The pipeline preset id.
+        supported_models (dict[str, Any]): The supported models.
+        supported_agents (list[str]): The supported agents.
+        support_pii_anonymization (bool): Whether the pipeline supports pii anonymization.
+        support_multimodal (bool): Whether the pipeline supports multimodal.
+        use_docproc (bool): Whether to use the document processor.
+        search_types (list[SearchType]): The supported search types.
+    """
+
+    pipeline_preset_id: str
+    supported_models: dict[str, Any]
+    supported_agents: list[str]
+    support_pii_anonymization: bool
+    support_multimodal: bool
+    use_docproc: bool
+    search_types: list[SearchType]