src/ctp_slack_bot/models/vector_query.py

from pydantic import BaseModel, Field, validator
from typing import Optional, List, Dict, Any

class VectorQuery(BaseModel):
    """Model for vector database similarity search queries.
    
    Attributes:
        query_text: The text to be vectorized and used for similarity search
        k: Number of similar documents to retrieve
        score_threshold: Minimum similarity score threshold for inclusion in results
        filter_metadata: Optional filters for metadata fields
    """
    query_text: str
    k: int
    score_threshold: float = Field(default=0.7)
    filter_metadata: Optional[Dict[str, Any]] = None