github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-09 04:49:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: GRAPH_COMMUNITY_SUMMARY_ENABLED = False error. (#2086)

Browse Source
This commit is contained in:
Aries-ckt
2024-10-22 11:16:01 +08:00
committed by GitHub
parent d9e20426fe
commit 6c682befa8
3 changed files with 4 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/rag/graph_rag_example.py
View File

@@ -20,11 +20,9 @@ from dbgpt.storage.knowledge_graph.knowledge_graph import (
)
"""GraphRAG example.
pre-requirements:
* Set LLM config (url/sk) in `.env`.
* Install pytest utils: `pip install pytest pytest-asyncio`
* Config TuGraph following the format below.
```
# Set LLM config (url/sk) in `.env`.
# Install pytest utils: `pip install pytest pytest-asyncio`
GRAPH_STORE_TYPE=TuGraph
TUGRAPH_HOST=127.0.0.1
TUGRAPH_PORT=7687
@@ -88,8 +86,7 @@ def __create_community_kg_connector():
async def ask_chunk(chunk: Chunk, question) -> str:
rag_template = (
"Based on the following [Context] {context}, "
"answer [Question] {question}."
"Based on the following [Context] {context}, " "answer [Question] {question}."
)
template = HumanPromptTemplate.from_template(rag_template)
messages = template.format_messages(context=chunk.content, question=question)