argo-proxy 3.1.1__tar.gz → 3.2.0a0__tar.gz

{argo_proxy-3.1.1/src/argo_proxy.egg-info → argo_proxy-3.2.0a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: argo-proxy
-Version: 3.1.1
+Version: 3.2.0a0
 Summary: Proxy server to Argo API, OpenAI format compatible
 Author-email: Peng Ding <oaklight@gmx.com>
 License-Expression: MIT
@@ -15,7 +15,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.12.2
-Requires-Dist: llm-rosetta<0.7.0,>=0.6.8
+Requires-Dist: llm-rosetta>=0.7.0a0
 Requires-Dist: pydantic>=2.11.7
 Requires-Dist: tiktoken>=0.9.0
 Requires-Dist: tqdm>=4.67.1
@@ -23,9 +23,9 @@ Requires-Dist: Pillow>=12.0.0
 Provides-Extra: dev
 Requires-Dist: dotenv>=0.9.9; extra == "dev"
 Requires-Dist: openai>=1.79.0; extra == "dev"
-Requires-Dist: ruff>=0.15.0; extra == "dev"
-Requires-Dist: ty>=0.0.31; extra == "dev"
-Requires-Dist: complexipy>=5.2.0; extra == "dev"
+Requires-Dist: ruff==0.15.20; extra == "dev"
+Requires-Dist: ty==0.0.54; extra == "dev"
+Requires-Dist: complexipy==5.6.1; extra == "dev"
 Requires-Dist: build>=1.2.2.post1; extra == "dev"
 Requires-Dist: twine>=6.1.0; extra == "dev"
 Requires-Dist: httpx>=0.28.1; extra == "dev"

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/pyproject.toml

@@ -20,7 +20,7 @@ classifiers = [
 
 dependencies = [
     "aiohttp>=3.12.2",
-    "llm-rosetta>=0.6.8,<0.7.0",
+    "llm-rosetta>=0.7.0a0",
     "pydantic>=2.11.7",
     "tiktoken>=0.9.0",
     "tqdm>=4.67.1",
@@ -31,9 +31,9 @@ dependencies = [
 dev = [
     "dotenv>=0.9.9",
     "openai>=1.79.0",
-    "ruff>=0.15.0",
-    "ty>=0.0.31",
-    "complexipy>=5.2.0",
+    "ruff==0.15.20",
+    "ty==0.0.54",
+    "complexipy==5.6.1",
     "build>=1.2.2.post1",
     "twine>=6.1.0",
     "httpx>=0.28.1",

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0/src/argo_proxy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: argo-proxy
-Version: 3.1.1
+Version: 3.2.0a0
 Summary: Proxy server to Argo API, OpenAI format compatible
 Author-email: Peng Ding <oaklight@gmx.com>
 License-Expression: MIT
@@ -15,7 +15,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: aiohttp>=3.12.2
-Requires-Dist: llm-rosetta<0.7.0,>=0.6.8
+Requires-Dist: llm-rosetta>=0.7.0a0
 Requires-Dist: pydantic>=2.11.7
 Requires-Dist: tiktoken>=0.9.0
 Requires-Dist: tqdm>=4.67.1
@@ -23,9 +23,9 @@ Requires-Dist: Pillow>=12.0.0
 Provides-Extra: dev
 Requires-Dist: dotenv>=0.9.9; extra == "dev"
 Requires-Dist: openai>=1.79.0; extra == "dev"
-Requires-Dist: ruff>=0.15.0; extra == "dev"
-Requires-Dist: ty>=0.0.31; extra == "dev"
-Requires-Dist: complexipy>=5.2.0; extra == "dev"
+Requires-Dist: ruff==0.15.20; extra == "dev"
+Requires-Dist: ty==0.0.54; extra == "dev"
+Requires-Dist: complexipy==5.6.1; extra == "dev"
 Requires-Dist: build>=1.2.2.post1; extra == "dev"
 Requires-Dist: twine>=6.1.0; extra == "dev"
 Requires-Dist: httpx>=0.28.1; extra == "dev"

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argo_proxy.egg-info/SOURCES.txt

@@ -33,7 +33,6 @@ src/argoproxy/utils/image_processing.py
 src/argoproxy/utils/logging.py
 src/argoproxy/utils/misc.py
 src/argoproxy/utils/models.py
-src/argoproxy/utils/tool_calls.py
 src/argoproxy/utils/transports.py
 test/test_chat_completions.py
 test/test_embeddings.py
@@ -41,4 +40,5 @@ test/test_function_calling_multiple.py
 test/test_function_calling_single.py
 test/test_leaked_tool_parser.py
 test/test_model_resolution.py
-tests/test_config_migrate.py
+tests/test_config_migrate.py
+tests/test_unix_socket.py

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argo_proxy.egg-info/requires.txt

@@ -1,5 +1,5 @@
 aiohttp>=3.12.2
-llm-rosetta<0.7.0,>=0.6.8
+llm-rosetta>=0.7.0a0
 pydantic>=2.11.7
 tiktoken>=0.9.0
 tqdm>=4.67.1
@@ -11,9 +11,9 @@ argo-proxy[dev,test]
 [dev]
 dotenv>=0.9.9
 openai>=1.79.0
-ruff>=0.15.0
-ty>=0.0.31
-complexipy>=5.2.0
+ruff==0.15.20
+ty==0.0.54
+complexipy==5.6.1
 build>=1.2.2.post1
 twine>=6.1.0
 httpx>=0.28.1

argo_proxy-3.2.0a0/src/argoproxy/__init__.py

@@ -0,0 +1 @@
+__version__ = "3.2.0a0"

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argoproxy/app.py

@@ -331,9 +331,10 @@ def create_app():
             context="app",
         )
 
-        # Only register health check, version, and dev proxy routes
+        # Register basic utility routes available in dev mode
         app.router.add_get("/health", health_check)
         app.router.add_get("/version", get_version)
+        app.router.add_post("/refresh", refresh_models)
 
         # Register dev proxy routes using a temporary config for URL resolution
         dev_config = ArgoConfig()
@@ -366,7 +367,7 @@ def create_app():
     return app
 
 
-def run(*, host: str = "0.0.0.0", port: int = 8080):
+def run(*, host: str = "0.0.0.0", port: int = 8080, socket: str = ""):
     app = create_app()
 
     # Add this to ensure signal handlers trigger a full shutdown
@@ -378,7 +379,68 @@ def run(*, host: str = "0.0.0.0", port: int = 8080):
         signal.signal(sig, _force_exit)
 
     try:
-        web.run_app(app, host=host, port=port)
+        if socket:
+            _run_unix_socket(app, socket)
+        else:
+            web.run_app(app, host=host, port=port)
     except Exception as e:
         log_error(f"An error occurred while starting the server: {e}", context="app")
         sys.exit(1)
+
+
+def _run_unix_socket(app: web.Application, socket_path: str):
+    """Start the server listening on a Unix domain socket.
+
+    Removes any stale socket file, starts the app, then restricts
+    permissions to owner-only (0o600) so other users on a shared
+    host cannot connect.
+    """
+    import stat
+
+    path = os.path.realpath(socket_path)
+
+    # Remove stale socket if present
+    if os.path.exists(path):
+        try:
+            st = os.stat(path)
+            if stat.S_ISSOCK(st.st_mode):
+                os.unlink(path)
+                log_info(f"Removed stale socket: {path}", context="app")
+            else:
+                log_error(
+                    f"Socket path exists and is not a socket: {path}", context="app"
+                )
+                sys.exit(1)
+        except OSError as e:
+            log_error(f"Cannot remove stale socket {path}: {e}", context="app")
+            sys.exit(1)
+
+    # Ensure parent directory exists
+    parent = os.path.dirname(path)
+    if not os.path.isdir(parent):
+        log_error(f"Socket parent directory does not exist: {parent}", context="app")
+        sys.exit(1)
+
+    async def _set_socket_permissions(app_inner: web.Application):
+        """Set restrictive permissions on the socket after it is created."""
+        if os.path.exists(path):
+            os.chmod(path, 0o600)
+            log_info(
+                f"Socket permissions set to 0700 (owner-only): {path}",
+                context="app",
+            )
+
+    async def _cleanup_socket(app_inner: web.Application):
+        """Remove socket file on shutdown."""
+        if os.path.exists(path):
+            try:
+                os.unlink(path)
+                log_info(f"Removed socket: {path}", context="app")
+            except OSError:
+                pass
+
+    app.on_startup.append(_set_socket_permissions)
+    app.on_shutdown.append(_cleanup_socket)
+
+    log_info(f"Starting server on unix socket: {path}", context="app")
+    web.run_app(app, path=path)

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argoproxy/cli/handlers.py

@@ -12,7 +12,7 @@ from .._vendor.semver import version_parse
 from ..__init__ import __version__
 from ..config import PATHS_TO_TRY, validate_config
 from ..utils.attack_logger import get_attack_logger
-from ..utils.logging import log_error
+from ..utils.logging import log_error, log_info
 from .display import CHANGELOG_URL, display_startup_banner
 
 
@@ -25,8 +25,12 @@ def set_config_envs(args: argparse.Namespace):
     """Set environment variables from serve CLI arguments."""
     if args.config:
         os.environ["CONFIG_PATH"] = args.config
+    if args.host:
+        os.environ["HOST"] = args.host
     if args.port:
         os.environ["PORT"] = str(args.port)
+    if getattr(args, "socket", None):
+        os.environ["SOCKET"] = args.socket
     if args.verbose:
         os.environ["VERBOSE"] = str(True)
     if args.quiet:
@@ -73,7 +77,22 @@ def handle_serve(args: argparse.Namespace):
         if config_path is not None:
             get_attack_logger().set_config_path(config_path)
 
-        run(host=config_instance.host, port=config_instance.port)
+        if config_instance.socket:
+            log_info(
+                f"🌐 Listening on unix://{config_instance.socket}",
+                context="cli",
+            )
+        else:
+            log_info(
+                f"🌐 Listening on http://{config_instance.host}:{config_instance.port}",
+                context="cli",
+            )
+
+        run(
+            host=config_instance.host,
+            port=config_instance.port,
+            socket=config_instance.socket,
+        )
     except KeyError:
         log_error("Port not specified in configuration file.", context="cli")
         sys.exit(1)

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argoproxy/cli/parser.py

@@ -40,6 +40,17 @@ def _add_serve_arguments(parser: argparse.ArgumentParser) -> None:
         type=int,
         help="Port number to bind the server to",
     )
+    parser.add_argument(
+        "--socket",
+        "-S",
+        type=str,
+        default=None,
+        help=(
+            "Unix socket path to listen on (overrides --host/--port).\n"
+            "Permissions are set to 0700 (owner-only) for security on\n"
+            "shared hosts. Example: /run/user/$(id -u)/argo-proxy.sock"
+        ),
+    )
 
     verbosity = parser.add_mutually_exclusive_group()
     verbosity.add_argument(

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argoproxy/config/io.py

@@ -34,7 +34,15 @@ def _format_config_yaml(data: dict) -> str:
     groups: list[tuple[str, list[str]]] = [
         (
             "# Core settings",
-            ["config_version", "user", "host", "port", "verbose", "log_to_file"],
+            [
+                "config_version",
+                "user",
+                "host",
+                "port",
+                "socket",
+                "verbose",
+                "log_to_file",
+            ],
         ),
         (
             "# Upstream",
@@ -118,6 +126,12 @@ def save_config(
 
 def _apply_env_overrides(config_data: ArgoConfig) -> ArgoConfig:
     """Apply environment variable overrides to the config."""
+    if env_host := os.getenv("HOST"):
+        config_data.host = env_host
+
+    if env_socket := os.getenv("SOCKET"):
+        config_data.socket = env_socket
+
     if env_port := os.getenv("PORT"):
         config_data.port = int(env_port)
 

{argo_proxy-3.1.1 → argo_proxy-3.2.0a0}/src/argoproxy/config/model.py

@@ -30,6 +30,7 @@ class ArgoConfig:
     # Configuration fields with default values
     host: str = "0.0.0.0"  # Default to 0.0.0.0
     port: int = 44497
+    socket: str = ""  # Unix socket path; overrides host:port when set
     user: str = ""
     verbose: bool = True
 
@@ -248,6 +249,10 @@ class ArgoConfig:
         # Drop private fields
         serialized = {k: v for k, v in serialized.items() if not k.startswith("_")}
 
+        # Only persist socket when explicitly set
+        if not self.socket:
+            serialized.pop("socket", None)
+
         # Add the user-configurable base URL (stored as private field)
         if self._argo_base_url:
             serialized["argo_base_url"] = self.argo_base_url