Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,6 +20,11 @@ load_dotenv()
|
|
| 20 |
logging.basicConfig(level=logging.INFO)
|
| 21 |
logger = logging.getLogger(__name__)
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
# Fetch API key for SambaNova
|
| 24 |
SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
|
| 25 |
if not SAMBANOVA_API_KEY:
|
|
@@ -87,9 +92,9 @@ with st.sidebar:
|
|
| 87 |
docs = loader.load_data()
|
| 88 |
|
| 89 |
embed_model = HuggingFaceEmbedding(
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
)
|
| 93 |
Settings.embed_model = embed_model
|
| 94 |
|
| 95 |
llm_predictor = load_llm()
|
|
|
|
| 20 |
logging.basicConfig(level=logging.INFO)
|
| 21 |
logger = logging.getLogger(__name__)
|
| 22 |
|
| 23 |
+
# Custom exception for application errors
|
| 24 |
+
class GitHubRAGError(Exception):
|
| 25 |
+
"""Custom exception for GitHub RAG application errors"""
|
| 26 |
+
pass
|
| 27 |
+
|
| 28 |
# Fetch API key for SambaNova
|
| 29 |
SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
|
| 30 |
if not SAMBANOVA_API_KEY:
|
|
|
|
| 92 |
docs = loader.load_data()
|
| 93 |
|
| 94 |
embed_model = HuggingFaceEmbedding(
|
| 95 |
+
model_name="nomic-ai/nomic-embed-text-v2-moe",
|
| 96 |
+
trust_remote_code=True,
|
| 97 |
+
)
|
| 98 |
Settings.embed_model = embed_model
|
| 99 |
|
| 100 |
llm_predictor = load_llm()
|