refactor: Refactor for core SDK (#1092)

This commit is contained in:
Fangyin Cheng
2024-01-21 09:57:57 +08:00
committed by GitHub
parent ba7248adbb
commit 2d905191f8
45 changed files with 236 additions and 133 deletions

View File

@@ -28,7 +28,7 @@ from dbgpt.core.interface.llm import ModelMetadata
from dbgpt.serve.agent.team.plan.team_auto_plan import AutoPlanChatManager
if __name__ == "__main__":
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)

View File

@@ -30,7 +30,7 @@ parent_dir = os.path.dirname(current_dir)
test_plugin_dir = os.path.join(parent_dir, "test_files")
if __name__ == "__main__":
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)

View File

@@ -24,7 +24,7 @@ from dbgpt.agent.memory.gpts_memory import GptsMemory
from dbgpt.core.interface.llm import ModelMetadata
if __name__ == "__main__":
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)

View File

@@ -27,7 +27,7 @@ from dbgpt.core.interface.llm import ModelMetadata
def summary_example_with_success():
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(

View File

@@ -24,7 +24,7 @@ from dbgpt.agent.memory.gpts_memory import GptsMemory
from dbgpt.core.interface.llm import ModelMetadata
if __name__ == "__main__":
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)

View File

@@ -25,7 +25,7 @@ from dbgpt.core.interface.llm import ModelMetadata
def summary_example_with_success():
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="summarize", llm_provider=llm_client)
@@ -76,7 +76,7 @@ def summary_example_with_success():
def summary_example_with_faliure():
from dbgpt.model import OpenAILLMClient
from dbgpt.model.proxy import OpenAILLMClient
llm_client = OpenAILLMClient()
context: AgentContext = AgentContext(conv_id="summarize", llm_provider=llm_client)