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