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")
|