mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-09 12:59:43 +00:00
refactor: Refactor for core SDK (#1092)
This commit is contained in:
@@ -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)
|
||||
|
@@ -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)
|
||||
|
@@ -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)
|
||||
|
@@ -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(
|
||||
|
@@ -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)
|
||||
|
@@ -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)
|
||||
|
Reference in New Issue
Block a user