refactor: Refactor for core SDK (#1092)

examples/agents/auto_plan_agent_dialogue_example.py

@@ -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)

examples/agents/awel_layout_agents_chat_examples.py

@@ -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)

examples/agents/plugin_agent_dialogue_example.py

@@ -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)

examples/agents/retrieve_summary_agent_dialogue_example.py

@@ -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(

examples/agents/single_agent_dialogue_example.py

@@ -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)

examples/agents/single_summary_agent_dialogue_example.py

@@ -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)

examples/awel/simple_rag_retriever_example.py

@@ -6,7 +6,7 @@ from pydantic import BaseModel, Field
 
 from dbgpt.configs.model_config import MODEL_PATH, PILOT_PATH
 from dbgpt.core.awel import DAG, HttpTrigger, JoinOperator, MapOperator
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.chunk import Chunk
 from dbgpt.rag.embedding.embedding_factory import DefaultEmbeddingFactory
 from dbgpt.rag.operator.embedding import EmbeddingRetrieverOperator

examples/awel/simple_rag_rewrite_example.py

@@ -32,7 +32,7 @@ from typing import Dict
 
 from dbgpt._private.pydantic import BaseModel, Field
 from dbgpt.core.awel import DAG, HttpTrigger, MapOperator
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.operator.rewrite import QueryRewriteOperator
 
 

examples/awel/simple_rag_summary_example.py

@@ -30,7 +30,7 @@ from typing import Dict
 
 from dbgpt._private.pydantic import BaseModel, Field
 from dbgpt.core.awel import DAG, HttpTrigger, MapOperator
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.knowledge.base import KnowledgeType
 from dbgpt.rag.operator.knowledge import KnowledgeOperator
 from dbgpt.rag.operator.summary import SummaryAssemblerOperator

examples/rag/rewrite_rag_example.py

@@ -1,7 +1,7 @@
 import asyncio
 import os
 
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.retriever.rewrite import QueryRewrite
 
 """Query rewrite example.

examples/rag/summary_extractor_example.py

@@ -1,6 +1,6 @@
 import asyncio
 
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.chunk_manager import ChunkParameters
 from dbgpt.rag.knowledge.factory import KnowledgeFactory
 from dbgpt.serve.rag.assembler.summary import SummaryAssembler

examples/sdk/simple_sdk_llm_example.py

@@ -7,7 +7,7 @@ from dbgpt.core.operator import (
     PromptBuilderOperator,
     RequestBuilderOperator,
 )
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 
 with DAG("simple_sdk_llm_example_dag") as dag:
     prompt_task = PromptBuilderOperator(
@@ -20,8 +20,6 @@ with DAG("simple_sdk_llm_example_dag") as dag:
 
 if __name__ == "__main__":
     output = asyncio.run(
-        out_parse_task.call(
-            call_data={"data": {"dialect": "mysql", "table_name": "user"}}
-        )
+        out_parse_task.call(call_data={"dialect": "mysql", "table_name": "user"})
     )
     print(f"output: \n\n{output}")

examples/sdk/simple_sdk_llm_sql_example.py

@@ -17,7 +17,7 @@ from dbgpt.core.operator import (
 )
 from dbgpt.datasource.operator.datasource_operator import DatasourceOperator
 from dbgpt.datasource.rdbms.conn_sqlite import SQLiteTempConnect
-from dbgpt.model import OpenAILLMClient
+from dbgpt.model.proxy import OpenAILLMClient
 from dbgpt.rag.operator.datasource import DatasourceRetrieverOperator
 
 
@@ -144,12 +144,10 @@ with DAG("simple_sdk_llm_sql_example") as dag:
 
 if __name__ == "__main__":
     input_data = {
-        "data": {
-            "db_name": "test_db",
-            "dialect": "sqlite",
-            "top_k": 5,
-            "user_input": "What is the name and age of the user with age less than 18",
-        }
+        "db_name": "test_db",
+        "dialect": "sqlite",
+        "top_k": 5,
+        "user_input": "What is the name and age of the user with age less than 18",
     }
     output = asyncio.run(sql_result_task.call(call_data=input_data))
     print(f"\nthoughts: {output.get('thoughts')}\n")