fix threshold -> 0.0
Browse files- app/database/pinecone.py +1 -1
- app/models/rag_models.py +1 -1
app/database/pinecone.py
CHANGED
|
@@ -26,7 +26,7 @@ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
|
|
| 26 |
DEFAULT_LIMIT_K = int(os.getenv("PINECONE_DEFAULT_LIMIT_K", "10"))
|
| 27 |
DEFAULT_TOP_K = int(os.getenv("PINECONE_DEFAULT_TOP_K", "6"))
|
| 28 |
DEFAULT_SIMILARITY_METRIC = os.getenv("PINECONE_DEFAULT_SIMILARITY_METRIC", "cosine")
|
| 29 |
-
DEFAULT_SIMILARITY_THRESHOLD = float(os.getenv("PINECONE_DEFAULT_SIMILARITY_THRESHOLD", "0.
|
| 30 |
ALLOWED_METRICS = os.getenv("PINECONE_ALLOWED_METRICS", "cosine,dotproduct,euclidean").split(",")
|
| 31 |
|
| 32 |
# Export constants for importing elsewhere
|
|
|
|
| 26 |
DEFAULT_LIMIT_K = int(os.getenv("PINECONE_DEFAULT_LIMIT_K", "10"))
|
| 27 |
DEFAULT_TOP_K = int(os.getenv("PINECONE_DEFAULT_TOP_K", "6"))
|
| 28 |
DEFAULT_SIMILARITY_METRIC = os.getenv("PINECONE_DEFAULT_SIMILARITY_METRIC", "cosine")
|
| 29 |
+
DEFAULT_SIMILARITY_THRESHOLD = float(os.getenv("PINECONE_DEFAULT_SIMILARITY_THRESHOLD", "0.0"))
|
| 30 |
ALLOWED_METRICS = os.getenv("PINECONE_ALLOWED_METRICS", "cosine,dotproduct,euclidean").split(",")
|
| 31 |
|
| 32 |
# Export constants for importing elsewhere
|
app/models/rag_models.py
CHANGED
|
@@ -12,7 +12,7 @@ class ChatRequest(BaseModel):
|
|
| 12 |
similarity_top_k: int = Field(6, description="Number of top similar documents to return (after filtering)")
|
| 13 |
limit_k: int = Field(10, description="Maximum number of documents to retrieve from vector store")
|
| 14 |
similarity_metric: str = Field("cosine", description="Similarity metric to use (cosine, dotproduct, euclidean)")
|
| 15 |
-
similarity_threshold: float = Field(0.
|
| 16 |
|
| 17 |
# User information
|
| 18 |
session_id: Optional[str] = Field(None, description="Session ID for tracking conversations")
|
|
|
|
| 12 |
similarity_top_k: int = Field(6, description="Number of top similar documents to return (after filtering)")
|
| 13 |
limit_k: int = Field(10, description="Maximum number of documents to retrieve from vector store")
|
| 14 |
similarity_metric: str = Field("cosine", description="Similarity metric to use (cosine, dotproduct, euclidean)")
|
| 15 |
+
similarity_threshold: float = Field(0.0, description="Threshold for vector similarity (0-1)")
|
| 16 |
|
| 17 |
# User information
|
| 18 |
session_id: Optional[str] = Field(None, description="Session ID for tracking conversations")
|