28fe56cfc1
MyPy compliance for the whole codebase - systematic way of executing tools (`executor.py`) - support for using pydantic models in tool inputs and outputs - mypy compliance (most of the changes) - removal of unused code (from previous iterations) Co-authored-by: Nate Barbettini <nate@arcade-ai.com>
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
import os
|
|
from pathlib import Path
|
|
from typing import Optional, Union
|
|
|
|
import toml
|
|
import tomlkit
|
|
from pydantic import BaseModel, EmailStr
|
|
|
|
|
|
class PackInfo(BaseModel):
|
|
"""Package Manager-esk info about a pack of tools."""
|
|
|
|
name: str
|
|
description: str
|
|
version: str
|
|
author: Optional[str]
|
|
email: Optional[EmailStr]
|
|
|
|
|
|
class ToolPack(BaseModel):
|
|
"""A package of tools and their dependencies."""
|
|
|
|
pack: PackInfo
|
|
depends: Optional[dict[str, str]] = None
|
|
tools: dict[str, str] = {}
|
|
|
|
def write_lock_file(self, pack_dir: Union[str, os.PathLike]) -> None:
|
|
"""Write the pack definition to a lock file."""
|
|
|
|
lock_file = Path(pack_dir) / "pack.lock.toml"
|
|
pack_dict = self.dict(by_alias=True, exclude_none=True)
|
|
pack_ordered_dict = {
|
|
"pack": pack_dict.get("pack"),
|
|
"depends": pack_dict.get("depends"),
|
|
"tools": pack_dict.get("tools"),
|
|
}
|
|
|
|
# Create a tomlkit document from the ordered dictionary
|
|
doc = tomlkit.document()
|
|
for key, value in pack_ordered_dict.items():
|
|
doc[key] = value
|
|
|
|
# Write the tomlkit document to file
|
|
with open(lock_file, "w") as f:
|
|
f.write(tomlkit.dumps(doc))
|
|
|
|
@classmethod
|
|
def from_lock_file(cls, pack_dir: Union[str, os.PathLike]) -> "ToolPack":
|
|
"""Create a ToolPack object from a lock file."""
|
|
|
|
pack_dir = Path(pack_dir).resolve()
|
|
lock_file = pack_dir / "pack.lock.toml"
|
|
with open(lock_file) as f:
|
|
data = toml.load(f)
|
|
return cls(**data)
|