48026129Zxc commited on
Commit
1dd7754
·
verified ·
1 Parent(s): e402257

Update my_app/rag_graph/build_graph.py

Browse files
Files changed (1) hide show
  1. my_app/rag_graph/build_graph.py +1 -1
my_app/rag_graph/build_graph.py CHANGED
@@ -17,7 +17,7 @@ def build_knowledge_graph(text: str) -> str:
17
  password="password"
18
  )
19
 
20
- llm = GoogleGenerativeAI(model="gemini-1.5-flash", temperature=0.3)
21
 
22
  transformer = LLMGraphTransformer(llm=llm)
23
  docs = [Document(page_content=text)]
 
17
  password="password"
18
  )
19
 
20
+ llm = GoogleGenerativeAI(model="gemini-2.5-flash", temperature=0.3)
21
 
22
  transformer = LLMGraphTransformer(llm=llm)
23
  docs = [Document(page_content=text)]