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

refactor(agent): Agent modular refactoring (#1487)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-05-07 09:45:26 +08:00
committed by GitHub
parent 2a418f91e8
commit 863b5404dd
86 changed files with 4513 additions and 967 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
examples/agents/retrieve_summary_agent_dialogue_example.py
View File

@@ -17,7 +17,7 @@
import asyncio
import os
from dbgpt.agent import AgentContext, GptsMemory, LLMConfig, UserProxyAgent
from dbgpt.agent import AgentContext, AgentMemory, LLMConfig, UserProxyAgent
from dbgpt.agent.expand.retrieve_summary_assistant_agent import (
RetrieveSummaryAssistantAgent,
)
@@ -29,17 +29,16 @@ async def summary_example_with_success():
llm_client = OpenAILLMClient(model_alias="gpt-3.5-turbo-16k")
context: AgentContext = AgentContext(conv_id="retrieve_summarize")
default_memory = GptsMemory()
agent_memory = AgentMemory()
summarizer = (
await RetrieveSummaryAssistantAgent()
.bind(context)
.bind(LLMConfig(llm_client=llm_client))
.bind(default_memory)
.bind(agent_memory)
.build()
)
user_proxy = UserProxyAgent(memory=default_memory, agent_context=context)
user_proxy = UserProxyAgent(memory=agent_memory, agent_context=context)
paths_urls = [
os.path.join(ROOT_PATH, "examples/agents/example_files/Nuclear_power.pdf"),
@@ -56,7 +55,7 @@ async def summary_example_with_success():
)
# dbgpt-vis message infos
print(await default_memory.one_chat_completions("retrieve_summarize"))
print(await agent_memory.gpts_memory.one_chat_completions("retrieve_summarize"))
if __name__ == "__main__":