Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .release-please-manifest.json
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
{
".": "0.1.0-alpha.3"
".": "0.1.0-alpha.4"
}
2 changes: 1 addition & 1 deletion .stats.yml
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
configured_endpoints: 4
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/kernel%2Fkernel-d168b58fcf39dbd0458d132091793d3e2d0930070b7dda2d5f7f1baff20dd31b.yml
openapi_spec_hash: b7e0fd7ee1656d7dbad57209d1584d92
config_hash: 75c0b894355904e2a91b70445072d4b4
config_hash: c2bc5253d8afd6d67e031f73353c9b22
8 changes: 8 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,13 @@
# Changelog

## 0.1.0-alpha.4 (2025-05-10)

Full Changelog: [v0.1.0-alpha.3...v0.1.0-alpha.4](https://github.com/onkernel/kernel-python-sdk/compare/v0.1.0-alpha.3...v0.1.0-alpha.4)

### Features

* **api:** update via SDK Studio ([d93116e](https://github.com/onkernel/kernel-python-sdk/commit/d93116e633eb9503647acfbe3e9769f33fdd19ed))

## 0.1.0-alpha.3 (2025-05-10)

Full Changelog: [v0.1.0-alpha.2...v0.1.0-alpha.3](https://github.com/onkernel/kernel-python-sdk/compare/v0.1.0-alpha.2...v0.1.0-alpha.3)
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ It is generated with [Stainless](https://www.stainless.com/).

## Documentation

The full API of this library can be found in [api.md](api.md).
The REST API documentation can be found on [docs.onkernel.com](https://docs.onkernel.com). The full API of this library can be found in [api.md](api.md).

## Installation

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[project]
name = "kernel"
version = "0.1.0-alpha.3"
version = "0.1.0-alpha.4"
description = "The official Python library for the kernel API"
dynamic = ["readme"]
license = "Apache-2.0"
Expand Down
22 changes: 22 additions & 0 deletions src/kernel/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,18 @@
)
from ._base_client import DefaultHttpxClient, DefaultAsyncHttpxClient
from ._utils._logs import setup_logging as _setup_logging
from .app_framework import (
App,
KernelApp,
KernelJson,
KernelAction,
KernelAppJson,
KernelContext,
KernelActionJson,
KernelAppRegistry,
app_registry,
export_registry,
)

__all__ = [
"types",
Expand Down Expand Up @@ -68,6 +80,16 @@
"DEFAULT_CONNECTION_LIMITS",
"DefaultHttpxClient",
"DefaultAsyncHttpxClient",
"KernelContext",
"KernelAction",
"KernelActionJson",
"KernelAppJson",
"KernelJson",
"KernelApp",
"KernelAppRegistry",
"App",
"app_registry",
"export_registry",
]

if not _t.TYPE_CHECKING:
Expand Down
2 changes: 1 addition & 1 deletion src/kernel/_version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

__title__ = "kernel"
__version__ = "0.1.0-alpha.3" # x-release-please-version
__version__ = "0.1.0-alpha.4" # x-release-please-version
155 changes: 155 additions & 0 deletions src/kernel/app_framework.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,155 @@
import json
import inspect
import functools
from typing import Any, Dict, List, Union, TypeVar, Callable, Optional
from dataclasses import dataclass

T = TypeVar("T")

# Context definition
@dataclass
class KernelContext:
"""Context object passed to action handlers"""
invocation_id: str

# Action definition
@dataclass
class KernelAction:
"""Action that can be invoked on a Kernel app"""
name: str
handler: Callable[..., Any]

# JSON interfaces
@dataclass
class KernelActionJson:
"""JSON representation of a Kernel action"""
name: str

@dataclass
class KernelAppJson:
"""JSON representation of a Kernel app"""
name: str
actions: List[KernelActionJson]

@dataclass
class KernelJson:
"""JSON representation of Kernel manifest"""
apps: List[KernelAppJson]
entrypoint: str

# App class
class KernelApp:
def __init__(self, name: str):
self.name = name
self.actions: Dict[str, KernelAction] = {}
# Register this app in the global registry
_app_registry.register_app(self)

def action(self, name_or_handler: Optional[Union[str, Callable[..., Any]]] = None) -> Callable[..., Any]:
"""Decorator to register an action with the app"""
if name_or_handler is None:
# This is the @app.action() case, which should return the decorator
def decorator(f: Callable[..., Any]) -> Callable[..., Any]:
return self._register_action(f.__name__, f)
return decorator
elif callable(name_or_handler):
# This is the @app.action case (handler passed directly)
return self._register_action(name_or_handler.__name__, name_or_handler)
else:
# This is the @app.action("name") case (name_or_handler is a string)
def decorator(f: Callable[..., Any]) -> Callable[..., Any]:
return self._register_action(name_or_handler, f) # name_or_handler is the name string here
return decorator

def _register_action(self, name: str, handler: Callable[..., Any]) -> Callable[..., Any]:
"""Internal method to register an action"""

@functools.wraps(handler)
def wrapper(*args: Any, **kwargs: Any) -> Any:
# Determine if the original handler accepts context as first argument
sig = inspect.signature(handler)
param_names = list(sig.parameters.keys())
param_count = len(param_names)

if param_count == 1:
actual_input = None
# The handler only takes input
if len(args) > 0: # Prioritize args if context was implicitly passed
# If context (args[0]) and input (args[1]) were provided, or just input (args[0])
actual_input = args[1] if len(args) > 1 else args[0]
elif kwargs:
# Attempt to find the single expected kwarg
if param_names: # Should always be true if param_count == 1
param_name = param_names[0]
if param_name in kwargs:
actual_input = kwargs[param_name]
elif kwargs: # Fallback if name doesn't match but kwargs exist
actual_input = next(iter(kwargs.values()))
elif kwargs: # param_names is empty but kwargs exist (unlikely for param_count==1)
actual_input = next(iter(kwargs.values()))
# If no args/kwargs, actual_input remains None, handler might raise error or accept None
return handler(actual_input)
else: # param_count == 0 or param_count > 1
# Handler takes context and input (or more), or no args
return handler(*args, **kwargs)

action = KernelAction(name=name, handler=wrapper)
self.actions[name] = action
return wrapper

def get_actions(self) -> List[KernelAction]:
"""Get all actions for this app"""
return list(self.actions.values())

def get_action(self, name: str) -> Optional[KernelAction]:
"""Get an action by name"""
return self.actions.get(name)

def to_dict(self) -> KernelAppJson:
"""Export app information without handlers"""
return KernelAppJson(
name=self.name,
actions=[KernelActionJson(name=action.name) for action in self.get_actions()]
)


# Registry for storing Kernel apps
class KernelAppRegistry:
def __init__(self) -> None:
self.apps: Dict[str, KernelApp] = {}

def register_app(self, app: KernelApp) -> None:
self.apps[app.name] = app

def get_apps(self) -> List[KernelApp]:
return list(self.apps.values())

def get_app_by_name(self, name: str) -> Optional[KernelApp]:
return self.apps.get(name)

def export(self, entrypoint_relpath: str) -> KernelJson:
"""Export the registry as a KernelJson object"""
apps = [app.to_dict() for app in self.get_apps()]
return KernelJson(apps=apps, entrypoint=entrypoint_relpath)

def export_json(self, entrypoint_relpath: str) -> str:
"""Export the registry as JSON"""
kernel_json = self.export(entrypoint_relpath)
return json.dumps(kernel_json.__dict__, indent=2)


# Create singleton registry for apps
_app_registry = KernelAppRegistry()

# Create a simple function for creating apps
def App(name: str) -> KernelApp:
"""Create a new Kernel app"""
return KernelApp(name)

# Export the app registry for boot loader
app_registry = _app_registry

# Function to export registry as JSON
def export_registry(entrypoint_relpath: str) -> str:
"""Export the registry as JSON"""
return _app_registry.export_json(entrypoint_relpath)