libentry 1.25.1__py3-none-any.whl → 1.27__py3-none-any.whl

libentry/mcp/service.py

@@ -10,7 +10,7 @@ from dataclasses import dataclass
 from queue import Empty, Queue
 from threading import Lock
 from types import GeneratorType
-from typing import Any, Callable, Dict, Generator, Iterable, List, Literal, Optional, Tuple, Type, Union
+from typing import Any, Callable, Dict, Generator, Iterable, List, Literal, Optional, Tuple, Union
 
 from flask import Flask, request as flask_request
 from pydantic import BaseModel, Field, TypeAdapter
@@ -891,19 +891,41 @@ class GunicornApplication(BaseApplication):
 
     def load(self):
         logger.info("Initializing the service.")
-        if isinstance(self.service_type, type) or callable(self.service_type):
-            service = self.service_type(self.service_config) if self.service_config else self.service_type()
-        elif self.service_config is None:
+        if isinstance(self.service_type, (List, Tuple)):
+            if isinstance(self.service_config, (List, Tuple)) and len(self.service_config) == len(self.service_type):
+                service = [
+                    self._create_service(t, c)
+                    for t, c in zip(self.service_type, self.service_config)
+                ]
+            elif self.service_config is None:
+                service = [
+                    self._create_service(t, self.service_config)
+                    for t in self.service_type
+                ]
+            else:
+                raise RuntimeError(
+                    f"You are going to run {len(self.service_type)} services. "
+                    "In the multiple service mode, every `service_type` requires a `service_config` object."
+                )
+        else:
+            service = self._create_service(self.service_type, self.service_config)
+        logger.info("Service initialized.")
+
+        return FlaskServer(service)
+
+    @staticmethod
+    def _create_service(service_type, service_config):
+        if isinstance(service_type, type) or callable(service_type):
+            service = service_type(service_config) if service_config else service_type()
+        elif service_config is None:
             logger.warning(
                 "Be careful! It is not recommended to start the server from a service instance. "
                 "Use service_type and service_config instead."
             )
-            service = self.service_type
+            service = service_type
         else:
-            raise TypeError(f"Invalid service type \"{type(self.service_type)}\".")
-        logger.info("Service initialized.")
-
-        return FlaskServer(service)
+            raise TypeError(f"Invalid service type \"{type(service_type)}\".")
+        return service
 
 
 class RunServiceConfig(BaseModel):
@@ -969,7 +991,7 @@ class RunServiceConfig(BaseModel):
 
 
 def run_service(
-        service_type: Union[Type, Callable],
+        service_type: Union[Union[type, Callable], List, Tuple],
         service_config=None,
         run_config: Optional[RunServiceConfig] = None,
         *,

libentry/test_api.py

@@ -11,7 +11,8 @@ from typing import Literal, Optional
 
 from pydantic import BaseModel
 
-from libentry.api import APIClient
+from libentry.mcp.client import APIClient
+from libentry.mcp.types import HTTPOptions
 
 
 class TestRequest(BaseModel):
@@ -49,14 +50,13 @@ def test(request: TestRequest):
                 if request.method == "GET":
                     response = APIClient().get(
                         request.url,
-                        **kwargs
+                        HTTPOptions(**kwargs)
                     )
                 else:
                     response = APIClient().post(
                         request.url,
                         request.data,
-                        stream=request.stream,
-                        **kwargs
+                        HTTPOptions(stream=request.stream, **kwargs)
                     )
                 if not request.stream:
                     t = time() - t

libentry/utils.py

@@ -2,10 +2,14 @@
 
 __author__ = "xi"
 __all__ = [
-    "Timer"
+    "Timer",
+    "TimedLRUCache",
 ]
 
+import threading
 import time
+from collections import OrderedDict
+from functools import wraps
 
 
 class Timer:
@@ -32,3 +36,42 @@ class Timer:
             raise RuntimeError("The timer has not started yet.")
         self.end_time = time.perf_counter()
         self.cost = self.end_time - self.start_time
+
+
+class TimedLRUCache:
+
+    def __init__(self, max_size: int = 256, timeout: float = 12):
+        self.maxsize = max_size
+        self.timeout = timeout  # seconds
+
+        self.lock = threading.Lock()
+        self.cache = OrderedDict()
+
+    def __call__(self, func):
+        @wraps(func)
+        def wrapped(*args, **kwargs):
+            key = (args, frozenset(kwargs.items()))
+            now = time.time()
+
+            with self.lock:
+                # clean expired items
+                if key in self.cache:
+                    result, timestamp = self.cache[key]
+                    if now - timestamp < self.timeout:
+                        self.cache.move_to_end(key)
+                        return result
+                    else:
+                        del self.cache[key]  # expired
+
+                # compute new result
+                result = func(*args, **kwargs)
+                self.cache[key] = (result, now)
+                self.cache.move_to_end(key)
+
+                # exceed the max size
+                if len(self.cache) > self.maxsize:
+                    self.cache.popitem(last=False)  # remove the oldest item
+
+                return result
+
+        return wrapped

{libentry-1.25.1.dist-info → libentry-1.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: libentry
-Version: 1.25.1
+Version: 1.27
 Summary: Entries for experimental utilities.
 Home-page: https://github.com/XoriieInpottn/libentry
 Author: xi

{libentry-1.25.1.dist-info → libentry-1.27.dist-info}/RECORD

@@ -8,12 +8,12 @@ libentry/json.py,sha256=gTjogVG6eGUMkVrjilKkpFC8ZkElGAyZIutO7mbmSzg,1956
 libentry/logging.py,sha256=IiYoCUzm8XTK1fduA-NA0FI2Qz_m81NEPV3d3tEfgdI,1349
 libentry/resource.py,sha256=qAPXF7RzRYeCM4xWft6XR3SYk7lEiQ6ymOHbpNoi0mQ,1821
 libentry/schema.py,sha256=40SOhCF_eytWOF47MWKCRHKHl_lCaQVetx1Af62PkiI,10439
-libentry/test_api.py,sha256=Xw7B7sH6g1iCTV5sFzyBF3JAJzeOr9xg0AyezTNsnIk,4452
-libentry/utils.py,sha256=O7P6GadtUIjq0N2IZH7PhHZDUM3NebzcqyDqytet7CM,683
+libentry/test_api.py,sha256=bs7P8gveLtuUvHxrpujSrfuYqLVJ3fOXVg_MZnZxoJ4,4504
+libentry/utils.py,sha256=vCm6UyAlibnPOlPJHZO57u3TXhw5PZmGM5_vBAPUnB4,1981
 libentry/mcp/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 libentry/mcp/api.py,sha256=GDErVCz_hh_ZeMxLS8bTPyBUhCTHw3Mm-nGFMV2W2yo,3669
 libentry/mcp/client.py,sha256=qFbigNIOJvEf8ZICN2K8nnmVcITIzDhJgaw3c0P-WXM,23078
-libentry/mcp/service.py,sha256=PpdQBxqKPiZ2tpnsqDYiC29_yjMjK9J3VWynn4uvpVg,38109
+libentry/mcp/service.py,sha256=YMzP9sTR6RaeguwbwapFZFcc7MjKuDDfpeXSCD09Xzg,39081
 libentry/mcp/types.py,sha256=aAoVO4jjqEvDzNneuZapmRYonLLnGsbcLoypVyRNNYg,12389
 libentry/service/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 libentry/service/common.py,sha256=OVaW2afgKA6YqstJmtnprBCqQEUZEWotZ6tHavmJJeU,42
@@ -22,10 +22,10 @@ libentry/service/list.py,sha256=ElHWhTgShGOhaxMUEwVbMXos0NQKjHsODboiQ-3AMwE,1397
 libentry/service/running.py,sha256=FrPJoJX6wYxcHIysoatAxhW3LajCCm0Gx6l7__6sULQ,5105
 libentry/service/start.py,sha256=mZT7b9rVULvzy9GTZwxWnciCHgv9dbGN2JbxM60OMn4,1270
 libentry/service/stop.py,sha256=wOpwZgrEJ7QirntfvibGq-XsTC6b3ELhzRW2zezh-0s,1187
-libentry-1.25.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-libentry-1.25.1.dist-info/METADATA,sha256=7HVCqasAkkqvvXhodRtTmaoZTG4FC70Q6CxsgZWnC8U,1135
-libentry-1.25.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-libentry-1.25.1.dist-info/entry_points.txt,sha256=1v_nLVDsjvVJp9SWhl4ef2zZrsLTBtFWgrYFgqvQBgc,61
-libentry-1.25.1.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
-libentry-1.25.1.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-libentry-1.25.1.dist-info/RECORD,,
+libentry-1.27.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+libentry-1.27.dist-info/METADATA,sha256=4Kmb2RNxxAYb26wbovxkx8haDAaW7tBg_JEuxPwkwuI,1133
+libentry-1.27.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+libentry-1.27.dist-info/entry_points.txt,sha256=1v_nLVDsjvVJp9SWhl4ef2zZrsLTBtFWgrYFgqvQBgc,61
+libentry-1.27.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
+libentry-1.27.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+libentry-1.27.dist-info/RECORD,,