mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-29 21:40:44 +00:00
feat(agent): Release agent SDK (#1396)
This commit is contained in:
@@ -17,14 +17,17 @@
|
||||
import asyncio
|
||||
import os
|
||||
|
||||
from dbgpt.agent.agents.agent import AgentContext
|
||||
from dbgpt.agent.agents.expand.data_scientist_agent import DataScientistAgent
|
||||
from dbgpt.agent.agents.llm.llm import LLMConfig
|
||||
from dbgpt.agent.agents.user_proxy_agent import UserProxyAgent
|
||||
from dbgpt.agent.memory.gpts_memory import GptsMemory
|
||||
from dbgpt.agent.resource.resource_api import AgentResource, ResourceType
|
||||
from dbgpt.agent.resource.resource_db_api import SqliteLoadClient
|
||||
from dbgpt.agent.resource.resource_loader import ResourceLoader
|
||||
from dbgpt.agent import (
|
||||
AgentContext,
|
||||
AgentResource,
|
||||
GptsMemory,
|
||||
LLMConfig,
|
||||
ResourceLoader,
|
||||
ResourceType,
|
||||
UserProxyAgent,
|
||||
)
|
||||
from dbgpt.agent.expand.data_scientist_agent import DataScientistAgent
|
||||
from dbgpt.agent.resource import SqliteLoadClient
|
||||
|
||||
current_dir = os.getcwd()
|
||||
parent_dir = os.path.dirname(current_dir)
|
||||
@@ -47,7 +50,7 @@ async def main():
|
||||
|
||||
resource_loader = ResourceLoader()
|
||||
sqlite_file_loader = SqliteLoadClient()
|
||||
resource_loader.register_resesource_api(sqlite_file_loader)
|
||||
resource_loader.register_resource_api(sqlite_file_loader)
|
||||
|
||||
user_proxy = await UserProxyAgent().bind(default_memory).bind(context).build()
|
||||
|
||||
@@ -61,14 +64,14 @@ async def main():
|
||||
.build()
|
||||
)
|
||||
|
||||
await user_proxy.a_initiate_chat(
|
||||
await user_proxy.initiate_chat(
|
||||
recipient=sql_boy,
|
||||
reviewer=user_proxy,
|
||||
message="当前库有那些表",
|
||||
)
|
||||
|
||||
## dbgpt-vis message infos
|
||||
print(await default_memory.one_chat_competions("test456"))
|
||||
print(await default_memory.one_chat_completions("test456"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
Reference in New Issue
Block a user