libentry 1.24.8__py3-none-any.whl → 1.25__py3-none-any.whl

libentry/argparse.py

@@ -14,9 +14,10 @@ __all__ = [
 
 import argparse
 import ast
+import json
 import re
 from dataclasses import fields, is_dataclass
-from typing import Optional, Sequence, Type, Union, get_args, get_origin
+from typing import Dict, List, Optional, Sequence, Type, Union, get_args, get_origin
 
 import yaml
 from pydantic import BaseModel
@@ -158,12 +159,29 @@ class ArgumentParser(argparse.ArgumentParser):
                     instance=default_value
                 )
             else:
+                desc = info.description + " " if info.description else ""
+                if info.is_required():
+                    desc += "(required)"
+                else:
+                    if default_value is None:
+                        desc += "(default=None)"
+                    elif isinstance(default_value, (str, int, float, bool)):
+                        desc += f"(default={default_value})"
+                    elif isinstance(default_value, (Dict, List)):
+                        try:
+                            desc += f"(default={json.dumps(default_value)})"
+                        except TypeError:
+                            default_type = type(default_value).__name__
+                            desc += f"(default={default_type} object)"
+                    else:
+                        default_type = type(default_value).__name__
+                        desc += f"(default={default_type} object)"
+
                 self.add_argument(
                     f"--{prefix}.{name}",
                     type=literal_eval,
                     default=DefaultValue(default_value),
-                    # required=info.is_required(),
-                    help=info.description
+                    help=desc
                 )
 
     def parse_args(self, args=None, namespace=None):

libentry/mcp/service.py

@@ -10,10 +10,10 @@ 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, Optional, Type, Union
+from typing import Any, Callable, Dict, Generator, Iterable, List, Literal, Optional, Type, Union
 
 from flask import Flask, request as flask_request
-from pydantic import BaseModel, TypeAdapter
+from pydantic import BaseModel, Field, TypeAdapter
 
 from libentry import json, logger
 from libentry.mcp import api
@@ -21,8 +21,8 @@ from libentry.mcp.api import APIInfo, list_api_info
 from libentry.mcp.types import BlobResourceContents, CallToolRequestParams, CallToolResult, Implementation, \
     InitializeRequestParams, InitializeResult, JSONRPCError, JSONRPCNotification, JSONRPCRequest, JSONRPCResponse, \
     ListResourcesResult, ListToolsResult, MIME, ReadResourceRequestParams, ReadResourceResult, Resource, SSE, \
-    ServerCapabilities, SubroutineError, SubroutineResponse, TextContent, TextResourceContents, Tool, ToolProperty, \
-    ToolSchema, ToolsCapability
+    ServerCapabilities, SubroutineError, SubroutineResponse, TextContent, TextResourceContents, Tool, ToolSchema, \
+    ToolsCapability
 from libentry.schema import APISignature, get_api_signature, query_api
 
 try:
@@ -904,26 +904,116 @@ class GunicornApplication(BaseApplication):
         return FlaskServer(service)
 
 
+class RunServiceConfig(BaseModel):
+    """Run service config."""
+
+    host: str = Field(
+        title="Hostname",
+        description=(
+            "The hostname of the server that runs the service. "
+            "IP address or domain name."
+        ),
+        default="0.0.0.0"
+    )
+    port: int = Field(
+        title="Port number",
+        description="The port that the service listened to.",
+    )
+    num_workers: int = Field(
+        title="Number of workers",
+        description="The number of workers (processes) to run the service.",
+        default=1
+    )
+    num_threads: int = Field(
+        title="Number of threads",
+        description="The number of threads for each worker.",
+        default=20
+    )
+    num_connections: Optional[int] = Field(
+        title="Max number of connections",
+        description="The maximum number of simultaneous clients (or socket connections).",
+        default=1024
+    )
+    backlog: Optional[int] = Field(
+        title="The maximum number of pending connections",
+        description="The maximum number of pending connections.",
+        default=2048
+    )
+    worker_class: Literal["sync", "gthread", "eventlet", "gevent", "tornado"] = Field(
+        title="The type of workers to use",
+        description="The type of workers to use.",
+        default="gthread"
+    )
+    timeout: int = Field(
+        title="Worker timeout",
+        description="Workers silent for more than this many seconds are killed and restarted.",
+        default=60
+    )
+    keyfile: Optional[str] = Field(
+        title="SSL key file",
+        description="SSL key file.",
+        default=None
+    )
+    keyfile_password: Optional[str] = Field(
+        title="SSL key file password",
+        description="SSL key file password.",
+        default=None
+    )
+    certfile: Optional[str] = Field(
+        title="SSL certificate file",
+        description="SSL certificate file.",
+        default=None
+    )
+
+
 def run_service(
         service_type: Union[Type, Callable],
         service_config=None,
-        host: str = "0.0.0.0",
-        port: int = 8888,
-        num_workers: int = 1,
-        num_threads: int = 20,
-        num_connections: Optional[int] = 1000,
-        backlog: Optional[int] = 1000,
-        worker_class: str = "gthread",
-        timeout: int = 60,
+        run_config: Optional[RunServiceConfig] = None,
+        *,
+        host: Optional[str] = None,
+        port: Optional[int] = None,
+        num_workers: Optional[int] = None,
+        num_threads: Optional[int] = None,
+        num_connections: Optional[int] = None,
+        backlog: Optional[int] = None,
+        worker_class: Optional[str] = None,
+        timeout: Optional[int] = None,
         keyfile: Optional[str] = None,
         keyfile_password: Optional[str] = None,
         certfile: Optional[str] = None
-):
+) -> None:
+    kwargs = {}
+    if host is not None:
+        kwargs["host"] = host
+    if port is not None:
+        kwargs["port"] = port
+    if num_workers is not None:
+        kwargs["num_workers"] = num_workers
+    if num_threads is not None:
+        kwargs["num_threads"] = num_threads
+    if num_connections is not None:
+        kwargs["num_connections"] = num_connections
+    if backlog is not None:
+        kwargs["backlog"] = backlog
+    if worker_class is not None:
+        kwargs["worker_class"] = worker_class
+    if timeout is not None:
+        kwargs["timeout"] = timeout
+    if keyfile is not None:
+        kwargs["keyfile"] = keyfile
+    if keyfile_password is not None:
+        kwargs["keyfile_password"] = keyfile_password
+    if certfile is not None:
+        kwargs["certfile"] = certfile
+
+    if run_config is None:
+        run_config = RunServiceConfig(**kwargs)
+    else:
+        for name, value in kwargs.items():
+            setattr(run_config, name, value)
+
     logger.info("Starting gunicorn server.")
-    if num_connections is None or num_connections < num_threads * 2:
-        num_connections = num_threads * 2
-    if backlog is None or backlog < num_threads * 2:
-        backlog = num_threads * 2
 
     def ssl_context(config, _default_ssl_context_factory):
         import ssl
@@ -931,21 +1021,21 @@ def run_service(
         context.load_cert_chain(
             certfile=config.certfile,
             keyfile=config.keyfile,
-            password=keyfile_password
+            password=run_config.keyfile_password
         )
         context.minimum_version = ssl.TLSVersion.TLSv1_3
         return context
 
     options = {
-        "bind": f"{host}:{port}",
-        "workers": num_workers,
-        "threads": num_threads,
-        "timeout": timeout,
-        "worker_connections": num_connections,
-        "backlog": backlog,
-        "keyfile": keyfile,
-        "certfile": certfile,
-        "worker_class": worker_class,
+        "bind": f"{run_config.host}:{run_config.port}",
+        "workers": run_config.num_workers,
+        "threads": run_config.num_threads,
+        "timeout": run_config.timeout,
+        "worker_connections": run_config.num_connections,
+        "backlog": run_config.backlog,
+        "keyfile": run_config.keyfile,
+        "certfile": run_config.certfile,
+        "worker_class": run_config.worker_class,
         "ssl_context": ssl_context
     }
     for name, value in options.items():

{libentry-1.24.8.dist-info → libentry-1.25.dist-info}/METADATA

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

{libentry-1.24.8.dist-info → libentry-1.25.dist-info}/RECORD

@@ -1,6 +1,6 @@
 libentry/__init__.py,sha256=ko2YBIIx5H3dD0tedBkialzJGEDczFaP_PZmT1cIlak,148
 libentry/api.py,sha256=UkXdBv9oqQhaSESRReLWEPj8venBUoCBppIg-FAXqKA,24146
-libentry/argparse.py,sha256=eEjEeLSWnfQNblOPD8wXotvyM6UsP35_NGcR7NN62pU,7993
+libentry/argparse.py,sha256=0_0_EJzLEgE8UFD8fEWCddGnDW1qxyMYMvcD3cuChvY,8882
 libentry/dataclasses.py,sha256=AQV2PuxplJCwGZ5HKX72U-z-POUhTdy3XtpEK9KNIGQ,4541
 libentry/executor.py,sha256=cTV0WxJi0nU1TP-cOwmeodN8DD6L1691M2HIQsJtGrU,6582
 libentry/experiment.py,sha256=ejgAHDXWIe9x4haUzIFuz1WasLY0_aD1z_vyEVGjTu8,4922
@@ -13,7 +13,7 @@ libentry/utils.py,sha256=O7P6GadtUIjq0N2IZH7PhHZDUM3NebzcqyDqytet7CM,683
 libentry/mcp/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 libentry/mcp/api.py,sha256=GDErVCz_hh_ZeMxLS8bTPyBUhCTHw3Mm-nGFMV2W2yo,3669
 libentry/mcp/client.py,sha256=3HmKAsFenPlDk1lfSdXL9GJwfCSU92uQS04zvZGeJfQ,23065
-libentry/mcp/service.py,sha256=qA6I9Mi-eudRYwVGhff1_aAImK424bhASlsWaHP8XNI,34971
+libentry/mcp/service.py,sha256=VTLA9dS85NJ64OCf1zBKLWvvE9l76I-U8WV5crCmBlM,37943
 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.24.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-libentry-1.24.8.dist-info/METADATA,sha256=xTdFiobksHW7BiQG0DgyOd3d46as22zSv3q2uxz5jKM,1135
-libentry-1.24.8.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-libentry-1.24.8.dist-info/entry_points.txt,sha256=1v_nLVDsjvVJp9SWhl4ef2zZrsLTBtFWgrYFgqvQBgc,61
-libentry-1.24.8.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
-libentry-1.24.8.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-libentry-1.24.8.dist-info/RECORD,,
+libentry-1.25.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+libentry-1.25.dist-info/METADATA,sha256=mlgd255AfSGCHpiUuPqHMVipyfRm-Xal71qb5dQAJlg,1133
+libentry-1.25.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+libentry-1.25.dist-info/entry_points.txt,sha256=1v_nLVDsjvVJp9SWhl4ef2zZrsLTBtFWgrYFgqvQBgc,61
+libentry-1.25.dist-info/top_level.txt,sha256=u2uF6-X5fn2Erf9PYXOg_6tntPqTpyT-yzUZrltEd6I,9
+libentry-1.25.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+libentry-1.25.dist-info/RECORD,,