47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any, List
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class RagItemPayload(BaseModel):
|
|
"""Payload for creating or updating a single RAG item."""
|
|
|
|
model_config = ConfigDict(populate_by_name=True, extra="ignore")
|
|
|
|
id: int = Field(..., description="Unique identifier for the RAG item.")
|
|
workspace_id: int = Field(..., alias="workspaceId", description="Workspace identifier.")
|
|
name: str = Field(..., description="Readable name of the item.")
|
|
embedding_data: str = Field(..., alias="embeddingData", description="Serialized embedding payload.")
|
|
type: str = Field(..., description='Item type, e.g. "METRIC".')
|
|
|
|
|
|
class RagDeleteRequest(BaseModel):
|
|
"""Payload for deleting a single RAG item."""
|
|
|
|
model_config = ConfigDict(populate_by_name=True, extra="ignore")
|
|
|
|
id: int = Field(..., description="Identifier of the item to delete.")
|
|
type: str = Field(..., description="Item type matching the stored record.")
|
|
|
|
|
|
class RagRetrieveRequest(BaseModel):
|
|
"""Payload for retrieving RAG items by semantic query."""
|
|
|
|
model_config = ConfigDict(populate_by_name=True, extra="ignore")
|
|
|
|
query: str = Field(..., description="Search query text.")
|
|
num: int = Field(..., description="Number of items to return.")
|
|
workspace_id: int = Field(..., alias="workspaceId", description="Workspace scope for the search.")
|
|
type: str = Field(..., description="Item type to search, e.g. METRIC.")
|
|
|
|
|
|
class RagRetrieveResponse(BaseModel):
|
|
"""Generic RAG retrieval response wrapper."""
|
|
|
|
model_config = ConfigDict(extra="allow")
|
|
|
|
data: List[Any] = Field(default_factory=list, description="Retrieved items.")
|
|
|