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

feat:add rag awel operator view metadata. (#1174)

Browse Source
This commit is contained in:
Aries-ckt
2024-02-21 10:24:12 +08:00
committed by GitHub
parent c78bd22fda
commit 32e1554282
10 changed files with 527 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/rag/simple_rag_embedding_example.py
View File

@@ -76,7 +76,7 @@ with DAG("simple_sdk_rag_embedding_example") as dag:
"/examples/rag/embedding", methods="POST", request_body=TriggerReqBody
)
request_handle_task = RequestHandleOperator()
knowledge_operator = KnowledgeOperator(knowledge_type=KnowledgeType.URL)
knowledge_operator = KnowledgeOperator(knowledge_type=KnowledgeType.URL.name)
vector_connector = _create_vector_connector()
url_parser_operator = MapOperator(map_function=lambda x: x["url"])
embedding_operator = EmbeddingAssemblerOperator(

2
examples/rag/simple_rag_retriever_example.py
View File

@@ -39,7 +39,7 @@ from dbgpt.storage.vector_store.connector import VectorStoreConnector
..code-block:: shell
DBGPT_SERVER="http://127.0.0.1:5555"
curl -X POST $DBGPT_SERVER/api/v1/awel/trigger/examples/rag/retrieve \
-H "Content-Type: application/json" -d '{
-H "Content-Type: application/json" -d '{ \
"query": "what is awel talk about?"
}'
"""