jvserve 2.0.15__py3-none-any.whl → 2.1.1__py3-none-any.whl

jvserve/__init__.py

@@ -4,5 +4,7 @@ jvserve package initialization.
 This package provides the webserver for loading and interacting with JIVAS agents.
 """
 
-__version__ = "2.0.15"
-__supported__jivas__versions__ = ["2.0.0"]
+from importlib.metadata import version
+
+__version__ = version("jvserve")
+__supported__jivas__versions__ = [__version__]

jvserve/cli.py

@@ -1,23 +1,29 @@
 """Module for registering CLI plugins for jaseci."""
 
+import asyncio
 import logging
 import os
-import time
+from concurrent.futures import ThreadPoolExecutor
 from contextlib import asynccontextmanager
+from pickle import load
 from typing import AsyncIterator, Optional
 
+import aiohttp
+import pymongo
+from bson import ObjectId
 from dotenv import load_dotenv
-from fastapi.responses import FileResponse, Response, StreamingResponse
-from jac_cloud.jaseci.security import authenticator
+from fastapi import FastAPI, HTTPException, Response
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse, StreamingResponse
+from jac_cloud.core.context import JaseciContext
+from jac_cloud.jaseci.main import FastAPI as JaseciFastAPI  # type: ignore
 from jac_cloud.plugin.jaseci import NodeAnchor
+from jaclang import JacMachine as Jac
 from jaclang.cli.cmdreg import cmd_registry
-from jaclang.plugin.default import hookimpl
-from jaclang.runtimelib.context import ExecutionContext
-from jaclang.runtimelib.machine import JacMachine
-from uvicorn import run as _run
+from jaclang.runtimelib.machine import hookimpl
+from watchfiles import Change, run_process
 
 from jvserve.lib.agent_interface import AgentInterface
-from jvserve.lib.agent_pulse import AgentPulse
 from jvserve.lib.file_interface import (
     DEFAULT_FILES_ROOT,
     FILE_INTERFACE,
@@ -25,59 +31,207 @@ from jvserve.lib.file_interface import (
 )
 from jvserve.lib.jvlogger import JVLogger
 
+# quiet the jac_cloud logger down to errors only
+# jac cloud dumps payload details to console which makes it hard to debug in JIVAS
+os.environ["LOGGER_LEVEL"] = "ERROR"
 load_dotenv(".env")
-
-
-def serve_proxied_file(file_path: str) -> FileResponse | StreamingResponse:
-    """Serve a proxied file from a remote or local URL."""
-    import mimetypes
-
-    import requests
-    from fastapi import HTTPException
-
+# Set up logging
+JVLogger.setup_logging(level="INFO")
+logger = logging.getLogger(__name__)
+
+# Global for MongoDB collection with thread-safe initialization
+url_proxy_collection = None
+collection_init_lock = asyncio.Lock()
+
+
+async def get_url_proxy_collection() -> pymongo.collection.Collection:
+    """Thread-safe initialization of MongoDB collection"""
+    global url_proxy_collection
+    if url_proxy_collection is None:
+        async with collection_init_lock:
+            if url_proxy_collection is None:  # Double-check locking
+                loop = asyncio.get_running_loop()
+                with ThreadPoolExecutor() as pool:
+                    url_proxy_collection = await loop.run_in_executor(
+                        pool,
+                        lambda: NodeAnchor.Collection.get_collection("url_proxies"),
+                    )
+    return url_proxy_collection
+
+
+async def serve_proxied_file(
+    file_path: str,
+) -> FileResponse | StreamingResponse | Response:
+    """Serve a proxied file from a remote or local URL (async version)"""
     if FILE_INTERFACE == "local":
-        return FileResponse(path=os.path.join(DEFAULT_FILES_ROOT, file_path))
+        root_path = os.environ.get("JIVAS_FILES_ROOT_PATH", DEFAULT_FILES_ROOT)
+        full_path = os.path.join(root_path, file_path)
+        if not os.path.exists(full_path):
+            raise HTTPException(status_code=404, detail="File not found")
+        return FileResponse(full_path)
 
     file_url = file_interface.get_file_url(file_path)
 
+    # Security check to prevent recursive calls
     if file_url and ("localhost" in file_url or "127.0.0.1" in file_url):
-        # prevent recusive calls when env vars are not detected
-        raise HTTPException(status_code=500, detail="Environment not set up correctly")
+        raise HTTPException(
+            status_code=500, detail="Environment misconfiguration detected"
+        )
 
     if not file_url:
         raise HTTPException(status_code=404, detail="File not found")
 
-    file_extension = os.path.splitext(file_path)[1].lower()
+    try:
+        async with aiohttp.ClientSession() as session:
+            async with session.get(file_url) as response:
+                response.raise_for_status()
+
+                return StreamingResponse(
+                    response.content.iter_chunked(8192),
+                    media_type=response.headers.get(
+                        "Content-Type", "application/octet-stream"
+                    ),
+                )
+    except aiohttp.ClientError as e:
+        raise HTTPException(status_code=502, detail=f"File fetch error: {str(e)}")
+
+
+def run_jivas(filename: str, host: str = "localhost", port: int = 8000) -> None:
+    """Starts JIVAS server with integrated file services"""
+
+    # Create agent interface instance with configuration
+    agent_interface = AgentInterface.get_instance(host=host, port=port)
+
+    base, mod = os.path.split(filename)
+    base = base if base else "./"
+    mod = mod[:-4]
+
+    JaseciFastAPI.enable()
+
+    ctx = JaseciContext.create(None)
+    if filename.endswith(".jac"):
+        Jac.jac_import(target=mod, base_path=base, override_name="__main__")
+    elif filename.endswith(".jir"):
+        with open(filename, "rb") as f:
+            Jac.attach_program(load(f))
+            Jac.jac_import(target=mod, base_path=base, override_name="__main__")
+    else:
+        raise ValueError("Not a valid file!\nOnly supports `.jac` and `.jir`")
+
+    # Define post-startup function to run AFTER server is ready
+    async def post_startup() -> None:
+        """Wait for server to be ready before initializing agents"""
+        health_url = f"http://{host}:{port}/healthz"
+        max_retries = 10
+        retry_delay = 1.0
+
+        for attempt in range(max_retries):
+            try:
+                async with aiohttp.ClientSession() as session:
+                    async with session.get(health_url, timeout=1) as response:
+                        if response.status == 200:
+                            logger.info("Server is ready, initializing agents...")
+                            await agent_interface.init_agents()
+                            return
+            except (aiohttp.ClientConnectorError, asyncio.TimeoutError) as e:
+                logger.warning(
+                    f"Server not ready yet (attempt {attempt + 1} / {max_retries}): {e}"
+                )
+                await asyncio.sleep(retry_delay)
+                retry_delay *= 1.5  # Exponential backoff
+
+        logger.error(
+            "Server did not become ready in time. Agent initialization skipped."
+        )
+
+    # set up lifespan events
+    async def on_startup() -> None:
+        logger.info("JIVAS is starting up...")
+        # Start initialization in background without blocking
+        asyncio.create_task(post_startup())
+
+    async def on_shutdown() -> None:
+        logger.info("JIVAS is shutting down...")
+
+    app = JaseciFastAPI.get()
+    app_lifespan = app.router.lifespan_context
+
+    @asynccontextmanager
+    async def lifespan_wrapper(app: FastAPI) -> AsyncIterator[Optional[str]]:
+        await on_startup()
+        async with app_lifespan(app) as maybe_state:
+            yield maybe_state
+        await on_shutdown()
+
+    app.router.lifespan_context = lifespan_wrapper
+
+    # Add CORS middleware to main app
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+    # Ensure the local file directory exists if that's the interface
+    if FILE_INTERFACE == "local":
+        directory = os.environ.get("JIVAS_FILES_ROOT_PATH", DEFAULT_FILES_ROOT)
+        if not os.path.exists(directory):
+            os.makedirs(directory, exist_ok=True)
+
+    # Setup file serving endpoint for both local and S3
+    @app.get("/files/{file_path:path}", response_model=None)
+    async def serve_file(
+        file_path: str,
+    ) -> FileResponse | StreamingResponse | Response:
+        # The serve_proxied_file function already handles both local and S3 cases
+        return await serve_proxied_file(file_path)
+
+    # Setup URL proxy endpoint
+    @app.get("/f/{file_id:path}", response_model=None)
+    async def get_proxied_file(
+        file_id: str,
+    ) -> FileResponse | StreamingResponse | Response:
+        params = file_id.split("/")
+        object_id = params[0]
+
+        try:
+            # Get MongoDB collection (thread-safe initialization)
+            collection = await get_url_proxy_collection()
+
+            # Run blocking MongoDB operation in thread pool
+            loop = asyncio.get_running_loop()
+            file_details = await loop.run_in_executor(
+                None, lambda: collection.find_one({"_id": ObjectId(object_id)})
+            )
 
-    # List of extensions to serve directly
-    direct_serve_extensions = [
-        ".pdf",
-        ".html",
-        ".txt",
-        ".js",
-        ".css",
-        ".json",
-        ".xml",
-        ".svg",
-        ".csv",
-        ".ico",
-    ]
+            descriptor_path = os.environ.get("JIVAS_DESCRIPTOR_ROOT_PATH")
 
-    if file_extension in direct_serve_extensions:
-        file_response = requests.get(file_url)
-        file_response.raise_for_status()  # Raise HTTPError for bad responses (4XX or 5XX)
+            if file_details:
+                if descriptor_path and descriptor_path in file_details["path"]:
+                    return Response(status_code=403)
+                return await serve_proxied_file(file_details["path"])
 
-        mime_type = mimetypes.guess_type(file_path)[0] or "application/octet-stream"
+            raise HTTPException(status_code=404, detail="File not found")
+        except Exception as e:
+            logger.error(f"Proxy error: {str(e)}")
+            raise HTTPException(status_code=500, detail="Internal server error")
 
-        return StreamingResponse(iter([file_response.content]), media_type=mime_type)
+    ctx.close()
+    # Run the app
+    JaseciFastAPI.start(host=host, port=port)
 
-    file_response = requests.get(file_url, stream=True)
-    file_response.raise_for_status()
 
-    return StreamingResponse(
-        file_response.iter_content(chunk_size=1024),
-        media_type="application/octet-stream",
+def log_reload(changes: set[tuple[Change, str]]) -> None:
+    """Log changes."""
+    num_of_changes = len(changes)
+    logger.warning(
+        f'Detected {num_of_changes} change{"s" if num_of_changes > 1 else ""}'
     )
+    for change in changes:
+        logger.warning(f"{change[1]} ({change[0].name})")
+    logger.warning("Reloading ...")
 
 
 class JacCmd:
@@ -89,182 +243,14 @@ class JacCmd:
         """Create Jac CLI cmds."""
 
         @cmd_registry.register
-        def jvserve(
-            filename: str,
-            host: str = "0.0.0.0",
-            port: int = 8000,
-            loglevel: str = "INFO",
-            workers: Optional[int] = None,
-        ) -> None:
-            """Launch the jac application."""
-            from jaclang import jac_import
-
-            # set up logging
-            JVLogger.setup_logging(level=loglevel)
-            logger = logging.getLogger(__name__)
-
-            # load FastAPI
-            from jac_cloud import FastAPI
-
-            FastAPI.enable()
-
-            # load the JAC application
-            jctx = ExecutionContext.create()
-
-            base, mod = os.path.split(filename)
-            base = base if base else "./"
-            mod = mod[:-4]
-
-            if filename.endswith(".jac"):
-                start_time = time.time()
-                jac_import(
-                    target=mod,
-                    base_path=base,
-                    cachable=True,
-                    override_name="__main__",
-                )
-                logger.info(f"Loading took {time.time() - start_time} seconds")
-
-            AgentInterface.HOST = host
-            AgentInterface.PORT = port
-
-            # set up lifespan events
-            async def on_startup() -> None:
-                # Perform initialization actions here
-                logger.info("JIVAS is starting up...")
-
-            async def on_shutdown() -> None:
-                # Perform initialization actions here
-                logger.info("JIVAS is shutting down...")
-                AgentPulse.stop()
-                # await AgentRTC.on_shutdown()
-                jctx.close()
-                JacMachine.detach()
-
-            app_lifespan = FastAPI.get().router.lifespan_context
-
-            @asynccontextmanager
-            async def lifespan_wrapper(app: FastAPI) -> AsyncIterator[Optional[str]]:
-                await on_startup()
-                async with app_lifespan(app) as maybe_state:
-                    yield maybe_state
-                await on_shutdown()
-
-            FastAPI.get().router.lifespan_context = lifespan_wrapper
-
-            # Setup custom routes
-            FastAPI.get().add_api_route(
-                "/interact", endpoint=AgentInterface.interact, methods=["POST"]
+        def jvserve(filename: str, host: str = "localhost", port: int = 8000) -> None:
+            """Launch unified JIVAS server with file services"""
+            watchdir = os.path.join(
+                os.path.abspath(os.path.dirname(filename)), "actions", ""
             )
-            FastAPI.get().add_api_route(
-                "/webhook/{key}",
-                endpoint=AgentInterface.webhook_exec,
-                methods=["GET", "POST"],
+            run_process(
+                watchdir,
+                target=run_jivas,
+                args=(filename, host, port),
+                callback=log_reload,
             )
-            FastAPI.get().add_api_route(
-                "/action/walker",
-                endpoint=AgentInterface.action_walker_exec,
-                methods=["POST"],
-                dependencies=authenticator,
-            )
-
-            # run the app
-            _run(FastAPI.get(), host=host, port=port, lifespan="on", workers=workers)
-
-        @cmd_registry.register
-        def jvfileserve(
-            directory: str, host: str = "0.0.0.0", port: int = 9000
-        ) -> None:
-            """Launch the file server for local files."""
-            # load FastAPI
-            from fastapi import FastAPI
-            from fastapi.middleware.cors import CORSMiddleware
-            from fastapi.staticfiles import StaticFiles
-
-            # Setup custom routes
-            app = FastAPI()
-
-            # Add CORS middleware
-            app.add_middleware(
-                CORSMiddleware,
-                allow_origins=["*"],
-                allow_credentials=True,
-                allow_methods=["*"],
-                allow_headers=["*"],
-            )
-
-            if not os.path.exists(directory):
-                os.makedirs(directory)
-
-            # Set the environment variable for the file root path
-            os.environ["JIVAS_FILES_ROOT_PATH"] = directory
-
-            # Mount the static files directory
-            app.mount(
-                "/files",
-                StaticFiles(directory=directory),
-                name="files",
-            )
-
-            # run the app
-            _run(app, host=host, port=port)
-
-        @cmd_registry.register
-        def jvproxyserve(
-            directory: str, host: str = "0.0.0.0", port: int = 9000
-        ) -> None:
-            """Launch the file proxy server for remote files."""
-            # load FastAPI
-            from fastapi import FastAPI
-            from fastapi.middleware.cors import CORSMiddleware
-
-            # Setup custom routes
-            app = FastAPI()
-
-            # Add CORS middleware
-            app.add_middleware(
-                CORSMiddleware,
-                allow_origins=["*"],
-                allow_credentials=True,
-                allow_methods=["*"],
-                allow_headers=["*"],
-            )
-
-            # Add proxy routes only if using S3
-            if FILE_INTERFACE == "s3":
-
-                @app.get("/files/{file_path:path}", response_model=None)
-                async def serve_file(
-                    file_path: str,
-                ) -> FileResponse | StreamingResponse | Response:
-                    descriptor_path = os.environ["JIVAS_DESCRIPTOR_ROOT_PATH"]
-                    if descriptor_path and descriptor_path in file_path:
-                        return Response(status_code=403)
-
-                    return serve_proxied_file(file_path)
-
-            @app.get("/f/{file_id:path}", response_model=None)
-            async def get_proxied_file(
-                file_id: str,
-            ) -> FileResponse | StreamingResponse | Response:
-                from bson import ObjectId
-                from fastapi import HTTPException
-
-                params = file_id.split("/")
-                object_id = params[0]
-
-                # mongo db collection
-                collection = NodeAnchor.Collection.get_collection("url_proxies")
-                file_details = collection.find_one({"_id": ObjectId(object_id)})
-                descriptor_path = os.environ["JIVAS_DESCRIPTOR_ROOT_PATH"]
-
-                if file_details:
-                    if descriptor_path and descriptor_path in file_details["path"]:
-                        return Response(status_code=403)
-
-                    return serve_proxied_file(file_details["path"])
-
-                raise HTTPException(status_code=404, detail="File not found")
-
-            # run the app
-            _run(app, host=host, port=port)