admin-api-lib 3.2.0__py3-none-any.whl

admin_api_lib/main.py

@@ -0,0 +1,54 @@
+"""Module for the main FastAPI application."""
+
+# coding: utf-8
+
+import logging.config
+
+import yaml
+from dependency_injector.containers import Container
+from fastapi import FastAPI
+
+from admin_api_lib.apis.admin_api import router
+from admin_api_lib.dependency_container import DependencyContainer
+from admin_api_lib.impl import admin_api
+
+with open("/config/logging.yaml", "r") as stream:
+    config = yaml.safe_load(stream)
+logging.config.dictConfig(config)
+
+app = FastAPI(
+    title="admin-api-lib",
+    description="The API is used for the communication between the \
+        admin frontend and the admin backend in the rag project.",
+    version="1.0.0",
+)
+
+app.include_router(router)
+
+container = DependencyContainer()
+container.wire(modules=[admin_api])
+app.container = container
+
+
+def register_dependency_container(new_container: Container):
+    """
+    Register a new dependency container and rewire the application container.
+
+    Parameters
+    ----------
+    new_container : Container
+        The new dependency container to be registered.
+
+    Notes
+    -----
+    This function preserves the old wiring configuration and then overrides the
+    application container with the new container. It rewires the application
+    container to include modules from both the old and new wiring configurations.
+    """
+    # preserve old wiring
+    wiring_target = container.wiring_config.modules
+    app.container.override(new_container)
+
+    # rewire
+    wiring_target = list(set(wiring_target + new_container.wiring_config.modules))
+    app.container.wire(modules=wiring_target)

admin_api_lib/models/document_status.py

@@ -0,0 +1,86 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from pydantic import BaseModel, ConfigDict, StrictStr
+from typing import Any, ClassVar, Dict, List
+from admin_api_lib.models.status import Status
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class DocumentStatus(BaseModel):
+    """
+    DocumentStatus
+    """  # noqa: E501
+
+    name: StrictStr
+    status: Status
+    __properties: ClassVar[List[str]] = ["name", "status"]
+
+    model_config = {
+        "populate_by_name": True,
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return self.model_dump_json(by_alias=True, exclude_unset=True)
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of DocumentStatus from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude={},
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Dict) -> Self:
+        """Create an instance of DocumentStatus from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({"name": obj.get("name"), "status": obj.get("status")})
+        return _obj

admin_api_lib/models/extra_models.py

@@ -0,0 +1,9 @@
+# coding: utf-8
+
+from pydantic import BaseModel
+
+
+class TokenModel(BaseModel):
+    """Defines a token model."""
+
+    sub: str

admin_api_lib/models/http_validation_error.py

@@ -0,0 +1,105 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the         admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from pydantic import BaseModel, ConfigDict
+from typing import Any, ClassVar, Dict, List, Optional
+from admin_api_lib.models.validation_error import ValidationError
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class HTTPValidationError(BaseModel):
+    """
+    HTTPValidationError
+    """  # noqa: E501
+
+    detail: Optional[List[ValidationError]] = None
+    __properties: ClassVar[List[str]] = ["detail"]
+
+    model_config = {
+        "populate_by_name": True,
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return self.model_dump_json(by_alias=True, exclude_unset=True)
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of HTTPValidationError from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude={},
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in detail (list)
+        _items = []
+        if self.detail:
+            for _item in self.detail:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["detail"] = _items
+        # set to None if detail (nullable) is None
+        # and model_fields_set contains the field
+        if self.detail is None and "detail" in self.model_fields_set:
+            _dict["detail"] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Dict) -> Self:
+        """Create an instance of HTTPValidationError from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "detail": (
+                    [ValidationError.from_dict(_item) for _item in obj.get("detail")]
+                    if obj.get("detail") is not None
+                    else None
+                )
+            }
+        )
+        return _obj

admin_api_lib/models/key_value_pair.py

@@ -0,0 +1,85 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the         admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from pydantic import BaseModel, ConfigDict, StrictStr
+from typing import Any, ClassVar, Dict, List
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class KeyValuePair(BaseModel):
+    """
+    KeyValuePair
+    """  # noqa: E501
+
+    key: StrictStr
+    value: StrictStr
+    __properties: ClassVar[List[str]] = ["key", "value"]
+
+    model_config = {
+        "populate_by_name": True,
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return self.model_dump_json(by_alias=True, exclude_unset=True)
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of KeyValuePair from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude={},
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Dict) -> Self:
+        """Create an instance of KeyValuePair from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({"key": obj.get("key"), "value": obj.get("value")})
+        return _obj

admin_api_lib/models/status.py

@@ -0,0 +1,44 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+import pprint
+import re  # noqa: F401
+from enum import Enum
+
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class Status(str, Enum):
+    """
+    allowed enum values
+    """
+
+    """
+    allowed enum values
+    """
+    UPLOADING = "UPLOADING"
+    PROCESSING = "PROCESSING"
+    READY = "READY"
+    ERROR = "ERROR"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of Status from a JSON string"""
+        return cls(json.loads(json_str))

admin_api_lib/models/validation_error.py

@@ -0,0 +1,104 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the         admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from pydantic import BaseModel, ConfigDict, StrictStr
+from typing import Any, ClassVar, Dict, List
+from admin_api_lib.models.validation_error_loc_inner import ValidationErrorLocInner
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class ValidationError(BaseModel):
+    """
+    ValidationError
+    """  # noqa: E501
+
+    loc: List[ValidationErrorLocInner]
+    msg: StrictStr
+    type: StrictStr
+    __properties: ClassVar[List[str]] = ["loc", "msg", "type"]
+
+    model_config = {
+        "populate_by_name": True,
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return self.model_dump_json(by_alias=True, exclude_unset=True)
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ValidationError from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude={},
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in loc (list)
+        _items = []
+        if self.loc:
+            for _item in self.loc:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["loc"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Dict) -> Self:
+        """Create an instance of ValidationError from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "loc": (
+                    [ValidationErrorLocInner.from_dict(_item) for _item in obj.get("loc")]
+                    if obj.get("loc") is not None
+                    else None
+                ),
+                "msg": obj.get("msg"),
+                "type": obj.get("type"),
+            }
+        )
+        return _obj

admin_api_lib/models/validation_error_loc_inner.py

@@ -0,0 +1,114 @@
+# coding: utf-8
+
+"""
+admin-api-lib
+
+The API is used for the communication between the         admin frontend and the admin backend in the rag project.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+
+try:
+    from typing import Self
+except ImportError:
+    from typing_extensions import Self
+
+
+class ValidationErrorLocInner(BaseModel):
+    """
+    ValidationErrorLocInner
+    """  # noqa: E501
+
+    anyof_schema_1_validator: Optional[StrictStr] = None
+    anyof_schema_2_validator: Optional[StrictInt] = None
+    actual_instance: Optional[Any] = None
+    any_of_schemas: Optional[List[StrictStr]] = None
+    __properties: ClassVar[List[str]] = [
+        "anyof_schema_1_validator",
+        "anyof_schema_2_validator",
+        "actual_instance",
+        "any_of_schemas",
+    ]
+
+    model_config = {
+        "populate_by_name": True,
+        "validate_assignment": True,
+        "protected_namespaces": (),
+    }
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return self.model_dump_json(by_alias=True, exclude_unset=True)
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ValidationErrorLocInner from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude={},
+            exclude_none=True,
+        )
+        # set to None if anyof_schema_1_validator (nullable) is None
+        # and model_fields_set contains the field
+        if self.anyof_schema_1_validator is None and "anyof_schema_1_validator" in self.model_fields_set:
+            _dict["anyof_schema_1_validator"] = None
+
+        # set to None if anyof_schema_2_validator (nullable) is None
+        # and model_fields_set contains the field
+        if self.anyof_schema_2_validator is None and "anyof_schema_2_validator" in self.model_fields_set:
+            _dict["anyof_schema_2_validator"] = None
+
+        # set to None if actual_instance (nullable) is None
+        # and model_fields_set contains the field
+        if self.actual_instance is None and "actual_instance" in self.model_fields_set:
+            _dict["actual_instance"] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Dict) -> Self:
+        """Create an instance of ValidationErrorLocInner from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate(
+            {
+                "anyof_schema_1_validator": obj.get("anyof_schema_1_validator"),
+                "anyof_schema_2_validator": obj.get("anyof_schema_2_validator"),
+                "actual_instance": obj.get("actual_instance"),
+                "any_of_schemas": obj.get("any_of_schemas"),
+            }
+        )
+        return _obj

admin_api_lib/prompt_templates/summarize_prompt.py

@@ -0,0 +1,14 @@
+"""Module for the summarize prompt template."""
+
+from langchain_core.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
+
+SUMMARIZE_PROMPT = ChatPromptTemplate.from_messages(
+    [
+        SystemMessagePromptTemplate.from_template(
+            "Fasse den folgenden Text zusammen. Die Ausgabe soll auf Deutsch sein. "
+            "Gebe nur die deutsche Zusammenfassung aus, keinen zusätzlichen Text. "
+            "Halte dich möglichst kurz mit deiner Antwort."
+        ),
+        HumanMessagePromptTemplate.from_template("{text}"),
+    ]
+)

admin_api_lib/rag_backend_client/openapi_client/__init__.py

@@ -0,0 +1,60 @@
+# coding: utf-8
+
+# flake8: noqa
+
+"""
+STACKIT RAG
+
+The perfect rag solution.
+
+The version of the OpenAPI document: 1.0.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+
+__version__ = "1.0.0"
+
+# import apis into sdk package
+from admin_api_lib.rag_backend_client.openapi_client.api.rag_api import RagApi
+from admin_api_lib.rag_backend_client.openapi_client.api_client import ApiClient
+
+# import ApiClient
+from admin_api_lib.rag_backend_client.openapi_client.api_response import ApiResponse
+from admin_api_lib.rag_backend_client.openapi_client.configuration import Configuration
+from admin_api_lib.rag_backend_client.openapi_client.exceptions import (
+    ApiAttributeError,
+    ApiException,
+    ApiKeyError,
+    ApiTypeError,
+    ApiValueError,
+    OpenApiException,
+)
+
+# import models into sdk package
+from admin_api_lib.rag_backend_client.openapi_client.models.chat_history import (
+    ChatHistory,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.chat_history_message import (
+    ChatHistoryMessage,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.chat_request import (
+    ChatRequest,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.chat_response import (
+    ChatResponse,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.chat_role import ChatRole
+from admin_api_lib.rag_backend_client.openapi_client.models.content_type import (
+    ContentType,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.delete_request import (
+    DeleteRequest,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.information_piece import (
+    InformationPiece,
+)
+from admin_api_lib.rag_backend_client.openapi_client.models.key_value_pair import (
+    KeyValuePair,
+)

admin_api_lib/rag_backend_client/openapi_client/api/__init__.py

@@ -0,0 +1,4 @@
+# flake8: noqa
+
+# import apis into api package
+from admin_api_lib.rag_backend_client.openapi_client.api.rag_api import RagApi