diff --git a/dbgpt/agent/agents/agent.py b/dbgpt/agent/agents/agent.py
index 01dcbf344..432e351db 100644
--- a/dbgpt/agent/agents/agent.py
+++ b/dbgpt/agent/agents/agent.py
@@ -172,7 +172,7 @@ class AgentContext:
     resource_db: Optional[AgentResource] = None
     resource_knowledge: Optional[AgentResource] = None
     resource_internet: Optional[AgentResource] = None
-    llm_models: Optional[List[ModelMetadata]] = None
+    llm_models: Optional[List[Union[ModelMetadata, str]]] = None
     model_priority: Optional[dict] = None
     agents: Optional[List[str]] = None
 
diff --git a/examples/agents/auto_plan_agent_dialogue_example.py b/examples/agents/auto_plan_agent_dialogue_example.py
index 1cc322f0f..aab7a7477 100644
--- a/examples/agents/auto_plan_agent_dialogue_example.py
+++ b/examples/agents/auto_plan_agent_dialogue_example.py
@@ -25,6 +25,7 @@
 
 from dbgpt.agent.agents.agent import AgentContext
 from dbgpt.agent.memory.gpts_memory import GptsMemory
+from dbgpt.core.interface.llm import ModelMetadata
 
 import asyncio
 import os
@@ -34,7 +35,7 @@
 
     llm_client = OpenAILLMClient()
     context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)
-    context.llm_models = ["gpt-3.5-turbo"]
+    context.llm_models =  [ModelMetadata(model="gpt-3.5-turbo")]
 
     default_memory = GptsMemory()
     coder = CodeAssistantAgent(memory=default_memory, agent_context=context)
diff --git a/examples/agents/single_agent_dialogue_example.py b/examples/agents/single_agent_dialogue_example.py
index 50b0da71b..71418c351 100644
--- a/examples/agents/single_agent_dialogue_example.py
+++ b/examples/agents/single_agent_dialogue_example.py
@@ -18,6 +18,7 @@
 from dbgpt.agent.agents.user_proxy_agent import UserProxyAgent
 from dbgpt.agent.memory.gpts_memory import GptsMemory
 from dbgpt.agent.agents.agent import AgentContext
+from dbgpt.core.interface.llm import ModelMetadata
 import asyncio
 import os
 
@@ -28,7 +29,7 @@
 
     llm_client = OpenAILLMClient()
     context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)
-    context.llm_models = ["gpt-3.5-turbo"]
+    context.llm_models = [ModelMetadata(model="gpt-3.5-turbo")]
 
     default_memory = GptsMemory()
     coder = CodeAssistantAgent(memory=default_memory, agent_context=context)