ManTea commited on
Commit
58605d4
·
1 Parent(s): efa6086

fix threshold -> 0.0

Browse files
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.75"))
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.75, description="Threshold for vector similarity (0-1)")
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")