mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-22 01:05:52 -04:00
Zander Chase
bee136efa4
Add client methods to read / list runs and sessions. Update walkthrough to: - Let the user create a dataset from the runs without going to the UI - Use the new CLI command to start the server Improve the error message when `docker` isn't found
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
from datetime import datetime
|
|
from typing import Any, Dict, List, Optional
|
|
from uuid import UUID
|
|
|
|
from pydantic import BaseModel, Field, root_validator
|
|
|
|
from langchain.callbacks.tracers.schemas import Run, RunTypeEnum
|
|
|
|
|
|
class ExampleBase(BaseModel):
|
|
"""Example base model."""
|
|
|
|
dataset_id: UUID
|
|
inputs: Dict[str, Any]
|
|
outputs: Optional[Dict[str, Any]] = Field(default=None)
|
|
|
|
|
|
class ExampleCreate(ExampleBase):
|
|
"""Example create model."""
|
|
|
|
id: Optional[UUID]
|
|
created_at: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
|
|
class Example(ExampleBase):
|
|
"""Example model."""
|
|
|
|
id: UUID
|
|
created_at: datetime
|
|
modified_at: Optional[datetime] = Field(default=None)
|
|
runs: List[Run] = Field(default_factory=list)
|
|
|
|
|
|
class DatasetBase(BaseModel):
|
|
"""Dataset base model."""
|
|
|
|
tenant_id: UUID
|
|
name: str
|
|
description: str
|
|
|
|
|
|
class DatasetCreate(DatasetBase):
|
|
"""Dataset create model."""
|
|
|
|
id: Optional[UUID]
|
|
created_at: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
|
|
class Dataset(DatasetBase):
|
|
"""Dataset ORM model."""
|
|
|
|
id: UUID
|
|
created_at: datetime
|
|
modified_at: Optional[datetime] = Field(default=None)
|
|
|
|
|
|
class ListRunsQueryParams(BaseModel):
|
|
"""Query params for GET /runs endpoint."""
|
|
|
|
class Config:
|
|
extra = "forbid"
|
|
|
|
id: Optional[List[UUID]]
|
|
"""Filter runs by id."""
|
|
parent_run: Optional[UUID]
|
|
"""Filter runs by parent run."""
|
|
run_type: Optional[RunTypeEnum]
|
|
"""Filter runs by type."""
|
|
session: Optional[UUID] = Field(default=None, alias="session_id")
|
|
"""Only return runs within a session."""
|
|
reference_example: Optional[UUID]
|
|
"""Only return runs that reference the specified dataset example."""
|
|
execution_order: Optional[int]
|
|
"""Filter runs by execution order."""
|
|
error: Optional[bool]
|
|
"""Whether to return only runs that errored."""
|
|
offset: Optional[int]
|
|
"""The offset of the first run to return."""
|
|
limit: Optional[int]
|
|
"""The maximum number of runs to return."""
|
|
start_time: Optional[datetime] = Field(
|
|
default=None,
|
|
alias="start_before",
|
|
description="Query Runs that started <= this time",
|
|
)
|
|
end_time: Optional[datetime] = Field(
|
|
default=None,
|
|
alias="end_after",
|
|
description="Query Runs that ended >= this time",
|
|
)
|
|
|
|
@root_validator
|
|
def validate_time_range(cls, values: Dict[str, Any]) -> Dict[str, Any]:
|
|
"""Validate that start_time <= end_time."""
|
|
start_time = values.get("start_time")
|
|
end_time = values.get("end_time")
|
|
if start_time and end_time and start_time > end_time:
|
|
raise ValueError("start_time must be <= end_time")
|
|
return values
|