Spaces:
Running
Running
Update process_aware_rag.py
Browse files- process_aware_rag.py +1 -1
process_aware_rag.py
CHANGED
|
@@ -59,7 +59,7 @@ class ProcessAwareRAG:
|
|
| 59 |
|
| 60 |
# Initialize LLM
|
| 61 |
genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
|
| 62 |
-
self.llm = genai.GenerativeModel('gemini-
|
| 63 |
|
| 64 |
def retrieve_graph_context(self, classification: Dict) -> Dict:
|
| 65 |
"""Retrieve relevant context from knowledge graph"""
|
|
|
|
| 59 |
|
| 60 |
# Initialize LLM
|
| 61 |
genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
|
| 62 |
+
self.llm = genai.GenerativeModel('gemini-2.5-flash-lite')
|
| 63 |
|
| 64 |
def retrieve_graph_context(self, classification: Dict) -> Dict:
|
| 65 |
"""Retrieve relevant context from knowledge graph"""
|