muhammadsalmanalfaridzi commited on
Commit
94621b3
·
verified ·
1 Parent(s): 50b1af7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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
- model_name="nomic-ai/nomic-embed-text-v2-moe",
91
- trust_remote_code=True,
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()