github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-23 18:31:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(rag): Support RAG SDK (#1322)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-22 15:36:57 +08:00
committed by GitHub
parent e65732d6e4
commit 8a17099dd2
69 changed files with 1332 additions and 558 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/util/chat_util.py
View File

@@ -1,19 +1,21 @@
import asyncio
from typing import Any, Coroutine, List
from dbgpt.app.scene import BaseChat, ChatFactory
chat_factory = ChatFactory()
async def llm_chat_response_nostream(chat_scene: str, **chat_param):
"""llm_chat_response_nostream"""
from dbgpt.app.scene import BaseChat, ChatFactory
chat_factory = ChatFactory()
chat: BaseChat = chat_factory.get_implementation(chat_scene, **chat_param)
res = await chat.get_llm_response()
return res
async def llm_chat_response(chat_scene: str, **chat_param):
from dbgpt.app.scene import BaseChat, ChatFactory
chat_factory = ChatFactory()
chat: BaseChat = chat_factory.get_implementation(chat_scene, **chat_param)
return chat.stream_call()