- RAG & SPIN Specialist improvements
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
from typing import Dict, Any
|
||||
from typing import Dict, Any, Optional
|
||||
|
||||
from flask import current_app
|
||||
from pydantic import BaseModel, Field, model_validator
|
||||
@@ -11,6 +11,7 @@ class RetrieverMetadata(BaseModel):
|
||||
document_id: int = Field(..., description="ID of the source document")
|
||||
version_id: int = Field(..., description="Version ID of the source document")
|
||||
document_name: str = Field(..., description="Name of the source document")
|
||||
url: Optional[str] = Field(..., description="URL of the source document")
|
||||
user_metadata: Dict[str, Any] = Field(
|
||||
default_factory=dict, # This will use an empty dict if None is provided
|
||||
description="User-defined metadata"
|
||||
|
||||
Reference in New Issue
Block a user