Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .env.example
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
# Pinecone Vector DB — get at https://app.pinecone.io
PINECONE_API_KEY=your_pinecone_api_key_here
PINECONE_INDEX_NAME=your_pinecone_index_name_here
PINECONE_NAMESPACE=c2si
# HuggingFace Inference API — get at https://huggingface.co/settings/tokens
HUGGINGFACE_TOKEN=hf_your_token_here
HF_TOKEN=hf_your_token_here
Expand Down
2 changes: 1 addition & 1 deletion db_update/Update.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ def update_database(overwrite=(len(sys.argv) > 1 and sys.argv[1] == '--overwrite

# Connect to the index
index = pc.Index(index_name)
namespace = "c2si"
namespace = os.getenv("PINECONE_NAMESPACE", "c2si")
from sentence_transformers import SentenceTransformer, SparseEncoder
import numpy as np

Expand Down
5 changes: 3 additions & 2 deletions models/NewsModel.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
from config.Database import client
from datetime import datetime
import os

class CybernewsDB:
def __init__(self):
from sentence_transformers import SentenceTransformer, SparseEncoder
self.client = client
self.index_name = "cybernews-hybrid-test-2"
self.namespace = "c2si"
self.index_name = os.getenv("PINECONE_INDEX_NAME", "cybernews-index")
self.namespace = os.getenv("PINECONE_NAMESPACE", "c2si")
self.index = self.client.Index(self.index_name)

# Initialize native local embedding models
Expand Down