Rivalcoder commited on
Commit
2497a16
·
verified ·
1 Parent(s): 5ed1779

Update process_aware_rag.py

Browse files
Files changed (1) hide show
  1. 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-1.5-flash')
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"""