fxn 0.0.36__tar.gz → 0.0.37__tar.gz

{fxn-0.0.36 → fxn-0.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fxn
-Version: 0.0.36
+Version: 0.0.37
 Summary: Run prediction functions locally in Python. Register at https://fxn.ai.
 Author-email: "NatML Inc." <hi@fxn.ai>
 License:                                  Apache License

fxn-0.0.37/fxn/cli/misc.py

@@ -0,0 +1,32 @@
+# 
+#   Function
+#   Copyright © 2024 NatML Inc. All Rights Reserved.
+#
+
+from rich import print
+from typer import Exit, Option
+from webbrowser import open as open_browser
+
+from ..version import __version__
+
+def _explore (value: bool):
+    if value:
+        open_browser("https://fxn.ai/explore")
+        raise Exit()
+
+def _learn (value: bool):
+    if value:
+        open_browser("https://docs.fxn.ai")
+        raise Exit()
+
+def _version (value: bool):
+    if value:
+        print(__version__)
+        raise Exit()
+
+def cli_options (
+    explore: bool = Option(None, "--explore", callback=_explore, help="Explore predictors on Function."),
+    learn: bool = Option(None, "--learn", callback=_learn, help="Learn about Function."),
+    version: bool = Option(None, "--version", callback=_version, help="Get the Function CLI version.")
+):
+    pass

{fxn-0.0.36 → fxn-0.0.37}/fxn/function.py

@@ -6,7 +6,7 @@
 from os import environ
 
 from .api import GraphClient
-from .services import EnvironmentVariableService, PredictionService, PredictorService, StorageService, UserService
+from .services import PredictionService, PredictorService, UserService
 
 class Function:
     """
@@ -26,16 +26,11 @@ class Function:
     users: UserService
     predictors: PredictorService
     predictions: PredictionService
-    #environment_variables: EnvironmentVariableService
-    #storage: StorageService
 
     def __init__ (self, access_key: str=None, api_url: str=None):
         access_key = access_key or environ.get("FXN_ACCESS_KEY", None)
         api_url = api_url or environ.get("FXN_API_URL", "https://api.fxn.ai")
-        client = GraphClient(access_key, api_url)
-        storage = StorageService(client)
-        self.client = client
-        self.users = UserService(client)
-        self.predictors = PredictorService(client, storage)
-        self.predictions = PredictionService(client)
-        #self.environment_variables = EnvironmentVariableService(self.client)
+        self.client = GraphClient(access_key, api_url)
+        self.users = UserService(self.client)
+        self.predictors = PredictorService(self.client)
+        self.predictions = PredictionService(self.client)

{fxn-0.0.36 → fxn-0.0.37}/fxn/services/__init__.py

@@ -5,6 +5,4 @@
 
 from .user import UserService, PROFILE_FIELDS, USER_FIELDS
 from .predictor import PredictorService, PREDICTOR_FIELDS
-from .prediction import PredictionService, PREDICTION_FIELDS
-from .environment import EnvironmentVariableService, ENVIRONMENT_VARIABLE_FIELDS
-from .storage import StorageService
+from .prediction import PredictionService, PREDICTION_FIELDS

{fxn-0.0.36 → fxn-0.0.37}/fxn/services/predictor.py

@@ -7,14 +7,12 @@ from typing import List
 
 from ..api import GraphClient
 from ..types import Predictor, PredictorStatus
-from .storage import StorageService
 from .user import PROFILE_FIELDS
 
 class PredictorService:
 
-    def __init__ (self, client: GraphClient, storage: StorageService) -> None:
+    def __init__ (self, client: GraphClient) -> None:
         self.client = client
-        self.storage = storage
 
     def retrieve (self, tag: str) -> Predictor:
         """

{fxn-0.0.36 → fxn-0.0.37}/fxn/version.py

@@ -3,4 +3,4 @@
 #   Copyright © 2024 NatML Inc. All Rights Reserved.
 #
 
-__version__ = "0.0.36"
+__version__ = "0.0.37"

{fxn-0.0.36 → fxn-0.0.37}/fxn.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fxn
-Version: 0.0.36
+Version: 0.0.37
 Summary: Run prediction functions locally in Python. Register at https://fxn.ai.
 Author-email: "NatML Inc." <hi@fxn.ai>
 License:                                  Apache License

{fxn-0.0.36 → fxn-0.0.37}/fxn.egg-info/SOURCES.txt

@@ -3,7 +3,6 @@ README.md
 pyproject.toml
 fxn/__init__.py
 fxn/function.py
-fxn/magic.py
 fxn/version.py
 fxn.egg-info/PKG-INFO
 fxn.egg-info/SOURCES.txt
@@ -36,10 +35,8 @@ fxn/lib/macos/x86_64/Function.dylib
 fxn/lib/windows/arm64/Function.dll
 fxn/lib/windows/x86_64/Function.dll
 fxn/services/__init__.py
-fxn/services/environment.py
 fxn/services/prediction.py
 fxn/services/predictor.py
-fxn/services/storage.py
 fxn/services/user.py
 fxn/types/__init__.py
 fxn/types/dtype.py

fxn-0.0.36/fxn/cli/misc.py

@@ -1,26 +0,0 @@
-# 
-#   Function
-#   Copyright © 2024 NatML Inc. All Rights Reserved.
-#
-
-from rich import print
-from typer import Exit, Option
-from webbrowser import open as open_browser
-
-from ..version import __version__
-
-def _learn_callback (value: bool):
-    if value:
-        open_browser("https://docs.fxn.ai")
-        raise Exit()
-
-def _version_callback (value: bool):
-    if value:
-        print(__version__)
-        raise Exit()
-
-def cli_options (
-    learn: bool = Option(None, "--learn", callback=_learn_callback, help="Learn about Function."),
-    version: bool = Option(None, "--version", callback=_version_callback, help="Get the Function CLI version.")
-):
-    pass

fxn-0.0.36/fxn/magic.py

@@ -1,35 +0,0 @@
-# 
-#   Function
-#   Copyright © 2024 NatML Inc. All Rights Reserved.
-#
-
-from IPython.core.interactiveshell import InteractiveShell
-from IPython.core.magic import Magics, magics_class, line_magic
-from typing import List
-
-@magics_class
-class FunctionMagics (Magics):
-
-    @line_magic
-    def fxn (self, line: str):
-        COMMANDS = {
-            "python": self.__python,
-            "image": self.__image,
-        }
-        args = line.split(" ")
-        command = COMMANDS.get(args[0], None)
-        if command is not None:
-            command(args[1:])
-        else:
-            raise RuntimeError(f"Unrecognized Function command: {args[0]}")
-
-    def __python (self, args: List[str]):
-        version = args[0]
-        print(f"Predictor will use Python {version} when running on Function")
-
-    def __image (self, args: List[str]):
-        image = args[0]
-        print(f"Predictor will use base image {image} when running on Function")
-
-def load_ipython_extension (ipython: InteractiveShell):
-    ipython.register_magics(FunctionMagics)

fxn-0.0.36/fxn/services/environment.py

@@ -1,111 +0,0 @@
-#
-#   Function
-#   Copyright © 2024 NatML Inc. All Rights Reserved.
-#
-
-from typing import List
-
-from ..api import GraphClient
-from ..types import EnvironmentVariable
-
-class EnvironmentVariableService:
-
-    def __init__ (self, client: GraphClient) -> None:
-        self.client = client
-        self.__value = "xxxxxxxx"
-
-    def list (self, organization: str=None) -> List[EnvironmentVariable]:
-        """
-        List the current user's environment variables.
-
-        Note that the variable values can only viewed at https://fxn.ai.
-
-        Parameters:
-            organization (str): Organization username.
-
-        Returns:
-            list: User environment variables.
-        """
-        # Query
-        response = self.client.query(f"""
-            query ($input: UserInput) {{
-                user (input: $input) {{
-                    ... on User {{
-                        environmentVariables {{
-                            {ENVIRONMENT_VARIABLE_FIELDS}
-                        }}
-                    }}
-                    ... on Organization {{
-                        environmentVariables {{
-                            {ENVIRONMENT_VARIABLE_FIELDS}
-                        }}
-                    }}
-                }}
-            }}
-            """,
-            { "input": { "username": organization } if organization is not None else None }
-        )
-        # Create envs
-        assert response["user"] is not None, "Failed to list environment variables because user could not be found. Check that you are authenticated."
-        environments = response["user"]["environmentVariables"]
-        environments = [EnvironmentVariable(**env, value=self.__value) for env in environments]
-        # Return
-        return environments
-    
-    def create (self, name: str, value: str, organization: str=None) -> EnvironmentVariable:
-        """
-        Create an environment variable.
-
-        This environment variable will apply to all predictors you create.
-
-        Parameters:
-            name (str): Variable name.
-            value (str): Variable value.
-            organization (str): Organization username. Use this for organization environment variables.
-
-        Returns:
-            EnvironmentVariable: Created environment variable.
-        """
-        # Query
-        response = self.client.query(f"""
-            mutation ($input: CreateEnvironmentVariableInput!) {{
-                environment: createEnvironmentVariable (input: $input) {{
-                    {ENVIRONMENT_VARIABLE_FIELDS}
-                }}
-            }}
-            """,
-            { "input": { "name": name, "value": value, "organization": organization } }
-        )
-        # Create env
-        environment = response["environment"]
-        environment = EnvironmentVariable(**environment, value=self.__value)
-        # Return
-        return environment
-    
-    def delete (self, name: str, organization: str=None) -> bool:
-        """
-        Delete an environment variable.
-
-        Parameters:
-            name (str): Variable name.
-            organization (str): Organization username. Use this for organization environment variables.
-            access_key (str): Function access key.
-        
-        Returns:
-            bool: Whether the environment variable was successfully deleted.
-        """
-        # Query
-        response = self.client.query(f"""
-            mutation ($input: DeleteEnvironmentVariableInput!) {{
-                result: deleteEnvironmentVariable (input: $input)
-            }}
-            """,
-            { "input": { "name": name, "organization": organization } }
-        )
-        # Return
-        return response["result"]
-
-    
-ENVIRONMENT_VARIABLE_FIELDS = f"""
-name
-"""

fxn-0.0.36/fxn/services/storage.py

@@ -1,159 +0,0 @@
-# 
-#   Function
-#   Copyright © 2024 NatML Inc. All Rights Reserved.
-#
-
-from base64 import b64encode
-from io import BytesIO
-from magika import Magika
-from pathlib import Path
-from requests import put
-from rich.progress import open as open_progress, wrap_file
-from urllib.parse import urlparse, urlunparse
-
-from ..api import GraphClient
-from ..types import UploadType
-
-class StorageService:
-
-    def __init__ (self, client: GraphClient) -> None:
-        self.client = client
-
-    def create_upload_url (self, name: str, type: UploadType, key: str=None) -> str:
-        """
-        Create an upload URL.
-
-        Parameters:
-            name (str): File name.
-            type (UploadType): Upload type.
-            key (str): File key. This is useful for grouping related files.
-
-        Returns:
-            str: File upload URL.
-        """
-        # Query
-        response = self.client.query(f"""
-            mutation ($input: CreateUploadUrlInput!) {{
-                createUploadUrl (input: $input)
-            }}
-            """,
-            { "input": { "type": type, "name": name, "key": key } }
-        )
-        # Return
-        return response["createUploadUrl"]
-    
-    def upload (
-        self,
-        file: str | Path | BytesIO,
-        *,
-        type: UploadType,
-        name: str=None,
-        data_url_limit: int=None,
-        key: str=None,
-        verbose: bool=False
-    ) -> str:
-        """
-        Upload a file and return the URL.
-
-        Parameters:
-            file (str | Path | BytesIO): Input file.
-            type (UploadType): File type.
-            name (str): File name. This MUST be provided if `file` is not a file path.
-            data_url_limit (int): Return a data URL if the file is smaller than this limit (in bytes).
-            key (str): File key. This is useful for grouping related files.
-            verbose (bool): Print a progress bar for the upload.
-
-        Returns:
-            str: Upload URL.
-        """
-        file = Path(file) if isinstance(file, str) else file
-        if isinstance(file, Path):
-            return self.__upload_file(file, type=type, name=name, key=key, data_url_limit=data_url_limit, verbose=verbose)
-        else:
-            return self.__upload_buffer(file, type=type, name=name, key=key, data_url_limit=data_url_limit, verbose=verbose)
-        
-    def __upload_file (
-        self,
-        file: Path,
-        *,
-        type: UploadType,
-        name: str=None,
-        key: str=None,
-        data_url_limit: int=None,
-        verbose: bool=False
-    ) -> str:
-        # Check file
-        assert file.exists(), f"Cannot upload {file.name} because the file does not exist"
-        assert file.is_file(), f"Cannot upload {file.name} becaause it does not point to a file"   
-        # Create data URL
-        mime = self.__infer_mime(file)
-        if file.stat().st_size < (data_url_limit or 0):
-            with open(file, mode="rb") as f:
-                buffer = BytesIO(f.read())
-            return self.__create_data_url(buffer, mime=mime)
-        # Upload
-        name = name or file.name
-        url = self.create_upload_url(name, type, key=key)
-        with open_progress(file, mode="rb", description=name, disable=not verbose) as f:
-            put(url, data=f, headers={ "Content-Type": mime }).raise_for_status()
-        # Return
-        return self.__simplify_url(url)
-    
-    def __upload_buffer (
-        self,
-        file: BytesIO,
-        *,
-        type: UploadType,
-        name: str=None,
-        key: str=None,
-        data_url_limit: int=None,
-        verbose: bool=False
-    ) -> str:
-        # Check name
-        assert name, "You must specify the file `name` if the `file` is not a path"
-        # Create data URL
-        file.seek(0)
-        mime = self.__infer_mime(file)
-        size = file.getbuffer().nbytes
-        if size < (data_url_limit or 0):
-            return self.__create_data_url(file, mime=mime)
-        # Upload
-        url = self.create_upload_url(name, type, key=key)
-        with wrap_file(file, total=size, description=name, disable=not verbose) as f:
-            put(url, data=f, headers={ "Content-Type": mime }).raise_for_status()
-        # Return
-        return self.__simplify_url(url)
-    
-    def __create_data_url (self, file: BytesIO, *, mime: str) -> str:
-        encoded_data = b64encode(file.getvalue()).decode("ascii")
-        url = f"data:{mime};base64,{encoded_data}"
-        return url
-    
-    def __simplify_url (self, url: str) -> str:
-        if url.startswith("data:"):
-            return url
-        parsed_url = urlparse(url)
-        parsed_url = parsed_url._replace(netloc="cdn.fxn.ai", query="")
-        url = urlunparse(parsed_url)
-        return url
-    
-    def __infer_mime (self, file: str | Path | BytesIO) -> str:
-        MAGIC_TO_MIME = {
-            b"\x00\x61\x73\x6d": "application/wasm"
-        }
-        # Read magic
-        file = Path(file) if isinstance(file, str) else file
-        if isinstance(file, Path):
-            with open(file, "rb") as f:
-                magic = f.read(4)
-        elif isinstance(file, BytesIO):
-            magic = file.getvalue()[:4]
-        # Check known mime
-        mime = MAGIC_TO_MIME.get(magic)
-        # Infer
-        if mime is None:
-            magika = Magika()
-            result = magika.identify_bytes(file.getvalue()) if isinstance(file, BytesIO) else magika.identify_path(file)
-            mime = result.output.mime_type
-        # Return
-        return mime