arbor-ai 0.1.1__tar.gz → 0.1.3__tar.gz

{arbor_ai-0.1.1 → arbor_ai-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: arbor-ai
-Version: 0.1.1
+Version: 0.1.3
 Summary: A framework for fine-tuning and managing language models
 License: MIT
 Keywords: machine learning,fine-tuning,language models

{arbor_ai-0.1.1 → arbor_ai-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arbor-ai"
-version = "0.1.1"
+version = "0.1.3"
 description = "A framework for fine-tuning and managing language models"
 authors = [
     {name = "Noah Ziems",email = "nziems2@nd.edu"}
@@ -19,16 +19,18 @@ dependencies = [
     "peft (>=0.14.0,<0.15.0)",
 ]
 
-[tool.poetry.scripts]
+[project.scripts]
 arbor = "arbor.cli:cli"
 
 [tool.poetry]
 name = "arbor"
-version = "0.1.1"
+version = "0.1.3"
 description = "A framework for fine-tuning and managing language models"
 authors = ["Noah Ziems <nziems2@nd.edu>"]
 readme = "README.md"
-packages = [{include = "arbor", from = "src"}]
+packages = [
+    { include = "arbor", from = "src" }
+]
 repository = "https://github.com/arbor-ai/arbor"
 keywords = ["machine learning", "fine-tuning", "language models"]
 license = "MIT"
@@ -42,6 +44,9 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
 ]
 
+[tool.poetry.scripts]
+arbor = "arbor.cli:cli"
+
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.5"
 pytest-asyncio = "^0.25.3"

arbor_ai-0.1.3/src/arbor/server/api/models/schemas.py

@@ -0,0 +1,18 @@
+from pydantic import BaseModel
+
+class FileResponse(BaseModel):
+    id: str
+    object: str = "file"
+    bytes: int
+    created_at: int
+    filename: str
+    purpose: str
+
+class FineTuneRequest(BaseModel):
+    model_name: str
+    training_file: str  # id of uploaded jsonl file
+
+class JobStatusResponse(BaseModel):
+    job_id: str
+    status: str
+    details: str = ""